← 返回列表

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.

GitHub原文发布 未知首次收录 2026-09-22上架 2026-09-22 07:01

☆ 收藏查看项目 ↗

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

登录后参与讨论,阅读资讯无需注册。

还没有评论,来分享第一个看法。