TensorX
返回文献探索

Paper · arXiv 2502.12143

Small Models Struggle to Learn from Strong Reasoners

Yuetai Li, Xiang Yue, Zhangchen Xu, Fengqing Jiang, Luyao Niu, Bill Yuchen Lin, Bhaskar Ramasubramanian, Radha Poovendran

40 upvotesFebruary 17, 2025arXiv 预印本
AI 摘要

Mix Distillation improves small model reasoning by balancing long and short chain-of-thought examples, addressing the Small Model Learnability Gap.

large language models (LLMs)chain-of-thought (CoT)distillationsmall modelsparameter-efficient fine-tuningMix Distillation

Abstract

Large language models (LLMs) excel in complex reasoning tasks, and distilling their reasoning capabilities into smaller models has shown promise. However, we uncover an interesting phenomenon, which we term the Small Model Learnability Gap: small models (leq3B parameters) do not consistently benefit from long chain-of-thought (CoT) reasoning or distillation from larger models. Instead, they perform better when fine-tuned on shorter, simpler reasoning chains that better align with their intrinsic learning capacity. To address this, we propose Mix Distillation, a simple yet effective strategy that balances reasoning complexity by combining long and short CoT examples or reasoning from both larger and smaller models. Our experiments demonstrate that Mix Distillation significantly improves small model reasoning performance compared to training on either data alone. These findings highlight the limitations of direct strong model distillation and underscore the importance of adapting reasoning complexity for effective reasoning capability transfer.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Small Models Struggle to Learn from Strong Reasoners | TensorX