学习与研究AI 资源
TinyCeNN-LM:用细胞神经网络循环状态核替代 Transformer 层的小模型研究
vtavakkoli/TinyCeNN-LM · Tiny language-model research lab exploring Cellular Neural Network (CeNN) recurrent state cores as alternatives to Transformer layers, with distillation, shared-weight iteration, and efficiency benchmarks.
ELI5:它是什么,解决什么问题
TinyCeNN-LM 是一个小型语言模型研究项目,探索用细胞神经网络(CeNN)循环状态核作为 Transformer 注意力层的替代方案,尝试在保留预训练语言模型能力的前提下减少部分注意力层的开销。仓库覆盖 SmolLM2、Qwen3.5、FunctionGemma 和 Gemma 4 等模型,包含蒸馏、共享权重迭代、缓存解码、质量/速度/内存对比评测,并保留负面结果与消融实验;当前主张先做选择性替换,通过表示相似度与模型级损失门控逐个接受替换层。
核心能力与快速开始
★ 1 · Jupyter Notebook
- 能做什么
- 提供细胞神经网络(CeNN)循环状态核替代 Transformer 注意力层的研究实现,涵盖 Integrated Memory(保守选择性替换)、PDelta3-GDN2-CLVR(循环/增量记忆替换)、MemoryFusion、AMCeNN/Top-2 与原始 TinyCeNN 等架构族。包含蒸馏、共享权重迭代、缓存解码、门控接受流程,以及质量、速度、内存和可训练参数规模的对比基准。仓库保留负面结果与消融记录,并用 MODEL_STATUS.md 标注各模型/笔记本的保留、归档与发布状态。
- 适用场景
- 适合研究者在 SmolLM2、Qwen3.5、FunctionGemma、Gemma 4 等小模型上实验注意力层替换与循环记忆架构;用于复现和对比 CeNN/循环记忆方案的效率与质量权衡;用于分析负面结果、消融实验与选择性替换策略;也可作为在 Colab 中开展小模型蒸馏与参数效率实验的起点。
- 项目形态
- 研究实现
- 部署方式
- 未说明;README 提供本地 git clone 与 pip install -e . 的安装方式,主要实验以 Colab notebook 形式运行。
- 主要依赖
- 未说明具体硬件与显存要求;README 提到 GPU 结果尚未确立,实验环境涉及 Jupyter Notebook、Python 虚拟环境与 PyTorch。
- 使用限制
- GPU 结果尚未建立;Gemma 4 等目标仍属实验性,未提升为已验证的胜出方案;未完成严格门控的运行不能作为适配模型优于基座模型的证明;未通过门控时保留原预训练机制;部分笔记本为 legacy/ablation 轨迹,需先查阅 MODEL_STATUS.md 再引用。
- 许可证
- MIT
- 最近推送
- 2026-09-22 05:50
资料检查:2026-09-22 07:00 · 卡片整理:2026-09-22 07:00
卡片由 AI 根据仓库资料整理,未进行安装或运行验证 · 本次更新未完成,保留上次成功内容
讨论 0
登录后参与讨论,阅读资讯无需注册。
还没有评论,来分享第一个看法。