用多个教师模型按需指导,让大模型推理更多样、更高效。
More Than One Teacher: Adaptive Multi-Guidance Policy Optimization for Diverse Exploration
- 多教师按需指导:只在学生出错时调用最合适的教师
- 数学推理提升4.3%,分布外任务提升12.2%,探索更丰富
- 适合追求推理多样性与效率的LLM研究者
基于可验证奖励的强化学习(RLVR)是提升大语言模型推理能力的有前景范式。然而,现有方法主要依赖自探索或单一离策略教师,易引入模型偏差并限制探索,最终制约推理多样性与性能。受知识蒸馏中多教师策略启发,本文提出自适应多引导策略优化(AMPO),在策略模型无法正确求解时,动态选择多个优秀教师模型提供引导,实现‘按需指导’,拓展探索空间的同时保留自发现价值。此外,AMPO采用基于理解度的选择机制,使学生仅学习其最可能理解的推理路径,平衡广泛探索与有效利用。大量实验表明,相比强基线GRPO,AMPO在数学推理任务上提升4.3%,分布外任务上提升12.2%,显著改善Pass@k表现并促进多样化探索。值得注意的是,使用四个同规模教师即可达到单个更强教师(如DeepSeek-R1)配合更多数据的效果。这为实现更优推理与泛化提供了更高效、可扩展的路径。代码已开源。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) is a promising paradigm for enhancing the reasoning ability in Large Language Models (LLMs). However, prevailing methods primarily rely on self-exploration or a single off-policy teacher to elicit long chain-of-thought (LongCoT) reasoning, which may introduce intrinsic model biases and restrict exploration, ultimately limiting reasoning diversity and performance. Drawing inspiration from multi-teacher strategies in knowledge distillation, we introduce Adaptive Multi-Guidance Policy Optimization (AMPO), a novel framework that adaptively leverages guidance from multiple proficient teacher models, but only when the on-policy model fails to generate correct solutions. This "guidance-on-demand" approach expands exploration while preserving the value of self-discovery. Moreover, AMPO incorporates a comprehension-based selection mechanism, prompting the student to learn from the reasoning paths that it is most likely to comprehend, thus balancing broad exploration with effective exploitation. Extensive experiments show AMPO substantially outperforms a strong baseline (GRPO), with a 4.3% improvement on mathematical reasoning tasks and 12.2% on out-of-distribution tasks, while significantly boosting Pass@k performance and enabling more diverse exploration. Notably, using four peer-sized teachers, our method achieves comparable results to approaches that leverage a single, more powerful teacher (e.g., DeepSeek-R1) with more data. These results demonstrate a more efficient and scalable path to superior reasoning and generalizability. Our code is available at https://github.com/SII-Enigma/AMPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。