Folia
← 返回头版

Mini-AGI:8GB显存下的连续学习模型

Mini-AGI是一个开源字节级连续学习语言模型,可在单块8GB显存的GPU上从零开始训练1。该项目通过将权重存储在磁盘上并按需调入显存的方式突破了显存限制,使个人PC或笔记本用户能够进行无限继续学习1。

项目采用了自适应深度和动态专家路由等创新架构设计1。目前训练进度已达318.1M字符,包含169个专家,总参数规模540.1M1。其中32个专家常驻显存(约109M参数),整体专家池由磁盘空间而非显存容量决定1。模型使用字节级词汇表(256字节值加9个结构标记),上下文长度为4,0961。

在连续学习能力方面,当干预学习率设置为专家学习率的0.1倍时,模型的连续学习遗忘率从+2.23降至+0.0067nat,知识保留率达99.84%1。项目由Alexey Borsky主导,并在Claude Opus 5模型的协助下完成了大部分代码实现1。


评论