跳到正文
arXiv cs.CL· Piyush Dhoka, Vedant Pandhare, Varun Inamdar, Aadarsh Pathre, Pavitha Nooji·· 10 小时前AI 评分34

深度剪枝小语言模型的能力损伤与恢复:哪些能力先崩、微调能救回什么

What Breaks and What Comes Back: Capability Damage and Recovery in Depth-Pruned Small Language Models

AI 导读

研究对 Llama-3.2-1B、Qwen2.5-1.5B、SmolLM2-1.7B 和 OLMo-2-1B 四款 1.2-1.7B 指令微调模型做深度剪枝与 healing 微调,发现函数调用和数学推理在所有模型中崩塌,保留率仅 0.00-0.07,且能力失效顺序高度一致(Kendall's W = 0.92)。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:深度剪枝会移除整个 Transformer 块,使语言模型更小、更快,但损害在模型各项能力上的分布并不均匀,而一次简短的微调(称为 healing)决定了哪些能力能够恢复。我们针对用作智能体的小型模型测量了这一点。四个参数量为 1.2-1.7B 的指令微调模型(Llama-3.2-1B、Qwen2.5-1.5B、SmolLM2-1.7B 和 OLMo-2-1B)经过剪枝和 healing,并追踪六项能力:结构化输出、函数调用、指令遵循、数学推理、安全拒绝和知识。三项结果在各模型家族中一致成立。第一,函数调用和数学能力在每个模型中都会崩溃(保留率为未剪枝得分的 0.00-0.07),且四项能力的崩溃顺序在四个模型间高度一致(Kendall's W = 0.92;校正偶然下限后为 0.62)。第二,训练哪些参数与使用哪些数据同样重要:在三个模型家族中,完全微调最后三个块而非向所有块添加低秩适配器,会使上限平均保留率最多损失 0.18,且每个范围一次 300k token 的试点即可检测到这一点。第三,我们测试了一个规划器,它根据 21 次短试点 healing 来分配 1M token 的 healing 预算。在固定预算下,它在两个模型家族上提升了最弱能力,尽管这些增益中只有一个在同时重采样训练种子的检验中得以保留。一旦计入其 3.48M 试点 token,使用相同总训练 token 数的均匀 healing 在每个模型家族上都达到或超过所有规划变体,并在其中三个家族的逐项检验中显著更好地保留了数学能力(在同时重采样种子时为一个)。更长时间的 healing 是一个强大而简单的基线,规划式恢复必须超越它。代码、逐项预测和所有结果均已公开。
评论: 17 页,6 张图
主题: 计算与语言(cs.CL);机器学习(cs.LG)
引用为: arXiv:2611.00877 [cs.CL]
  (或 arXiv:2611.00877v1 [cs.CL] 用于此版本)
  https://doi.org/10.48550/arXiv.2611.00877

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Vedant Pandhare [查看电子邮件]
[v1] 2026 年 10 月 31 日星期六 12:25:46 UTC(122 KB)

来源:arXiv cs.CL · arxiv.org