Improving Reasoning Capabilities in Small Models through Mixture-of-Layers Distillation with Stepwise Attention on Key Information
针对 CoT 蒸馏中教师模型动态注意力迁移不足的问题,提出了一种结合 Mixture-of-Layers 模块的新型蒸馏框架。该方法将教师模型在推理过程中对关键信息的渐进注意力转移(stepwise attention shift)作为结构化指导信号传递给学生模型,使小模型能够学习到推理过程中信息聚焦的动态模式。
INSIGHT核心创新在于发现并利用了 LLM 推理时注意力向关键信息渐进聚焦的规律,将其转化为可迁移的蒸馏信号。Mixture-of-Layers 模块允许学生模型动态选择和组合不同层的表示。对搜推系统部署而言,该技术可将大模型的推理能力压缩到可在线服务的小模型中,显著降低推理成本同时保持复杂查询理解能力。