跳到正文
原文
arXiv · Machine Learning Theory· Ranajoy Sadhukhan, Euijin Hong, Sheng Cao, Zechun Liu, Shuowei Jin, Sha Li, Changlong Yu, Xin Liu, Qin Lu, Zhenyu Shi, Priyanka Nigam, Beidi Chen·· 3 小时前AI 评分33

STAPLE:通过嵌入模块对预训练检查点做静态稀疏升级

STAPLE: Static Sparse Upcycling of Pretrained Checkpoints via Embedding Modules

AI 导读

STAPLE 在 STEM 基础上把 token 索引嵌入作为残差路径接入预训练模型,保留原 FFN 上投影并学习互补的 token 特定信息。在 OLMo-2 1B 上较继续稠密训练在 HumanEval、MBPP、GSM8K、DROP 分别提升 4.27、2.80、2.27、1.65 分,OLMo-2 7B 上同样有效。

来源:arXiv · Machine Learning Theory · arxiv.org