arXiv:2501.09804cs.LGcs.AI2025-01被引 7

小模型也能做靠谱的思维链推理,靠的是对抗性训练新方法。

Enhancing Generalization in Chain of Thought Reasoning for Smaller Models

  • 用对抗性微调让小模型学会跨领域通用的思考路径。
  • 在多个任务上超越现有最优方法,提升推理泛化能力。
  • 适合需要可解释性推理的小模型落地场景。

小规模语言模型的思维链(CoT)推理是极具挑战性的自然语言处理问题,但在实际应用中极具价值。现有知识蒸馏方法常导致小模型过度记忆,泛化能力差。由于无法完全保留教师模型的CoT能力,我们假设对抗性CoT微调对提升小模型的鲁棒性至关重要。为此,提出PRADA框架——一种融合多领域CoT的原理性微调方法。该方法首次实现两项改进:(1) 通过领域对抗微调恢复蒸馏中丢失的领域无关特征;(2) 通过对抗策略增强提示工程在不同领域的适应性。理论上证明了方法有效性,实验证明其在广泛任务上显著优于当前最优方法。此外,实验表明,使用PRADA的小模型能更贴近领域知识,提升可解释性。

原文摘要 · Abstract (English)

Chain-of-Thought (CoT) reasoning in smaller language models is a challenging natural language process problem yet highly desirable in many real-life applications. Existing CoT knowledge distillation methods often suffer from overly conservative memorization in smaller LLMs, leading to low generalization confidence. As fully preserving the CoT ability of teacher model is impossible, we hypothesize that adversarial CoT fine-tuning is crucial for developing smaller LLM with robust CoT generalization. To this end, we propose \textit{PRompt-Assisted Domain-Adversarial fine-tuning} (PRADA), a principled fine-tuning framework that integrates diverse CoT domains. Specifically, PRADA pioneers two CoT improvements in smaller LLM: (1) Recovering the domain-invariant feature insight which typically lost during distillation with domain adversarial fine-tuning; (2) Enhancing the domain adaptability of CoT prompt engineering by employing domain-adversarial approaches. We theoretically demonstrate the effectiveness of our approach and empirically show that it significantly outperforms the state of the arts in a wide range of tasks. Moreover, our empirical findings reveal that the smaller LLM, when leveraging PRADA, aligns closely with domain knowledge, thereby improving the explainability of our approach.

思维链小模型对抗训练泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。