跳到正文
原文
arXiv 机器学习· Tianzhe Chu, Druv Pai, Yi Ma·· 8 小时前AI 评分43

面向可扩展预训练与持续学习的嵌套稀疏架构 Nestor

Nested Sparsity for Scalable Pretraining and Continual Learning

AI 导读

研究团队提出基于嵌套稀疏机制的 Transformer 架构 Nestor,将 attention 算子与 MoE 算子的稀疏性分层组合,在同等计算量和稀疏度下预训练表现优于传统 MoE。Nestor 能更有效适配长上下文和新领域并减少旧领域遗忘,优势随模型与数据规模扩大持续保持,其代码与 checkpoints 已在 GitHub 开源。

来源:arXiv 机器学习 · arxiv.org