TRAPO融合精调与强化学习,提升大模型推理能力。
Trust-Region Adaptive Policy Optimization
- 交替优化专家指令与自生成内容,统一外部监督与自主探索。
- 在5个数学推理任务上超越传统方法与最新技术。
- 适合需要强推理能力的LLM优化场景。
后训练方法,尤其是监督微调(SFT)和强化学习(RL),在提升大语言模型(LLMs)复杂推理能力方面发挥关键作用。然而,主流的两阶段流程(先SFT再RL)存在核心矛盾:SFT强制模仿抑制了探索并导致遗忘,限制了RL的改进潜力。为此,我们提出TRAPO(信任区域自适应策略优化),一种混合框架,通过在每个训练实例中交替优化专家前缀上的SFT损失和模型自生成补全上的RL损失,实现外部监督与自我探索的统一。为稳定训练,引入信任区域SFT(TrSFT),在信任区域内最小化前向KL散度但弱化区域外优化,有效转向反向KL,实现稳定且有利于RL的模式聚焦更新。自适应前缀选择机制根据评估效用分配专家指导。在五个数学推理基准上的实验表明,TRAPO持续优于标准SFT、RL、SFT-then-RL流程以及近期最先进方法,确立了增强推理能力的大模型新范式。
原文摘要 · Abstract (English)
Post-training methods, especially Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL), play an important role in improving large language models' (LLMs) complex reasoning abilities. However, the dominant two-stage pipeline (SFT then RL) suffers from a key inconsistency: SFT enforces rigid imitation that suppresses exploration and induces forgetting, limiting RL's potential for improvements. We address this inefficiency with TRAPO (\textbf{T}rust-\textbf{R}egion \textbf{A}daptive \textbf{P}olicy \textbf{O}ptimization), a hybrid framework that interleaves SFT and RL within each training instance by optimizing SFT loss on expert prefixes and RL loss on the model's own completions, unifying external supervision and self-exploration. To stabilize training, we introduce Trust-Region SFT (TrSFT), which minimizes forward KL divergence inside a trust region but attenuates optimization outside, effectively shifting toward reverse KL and yielding stable, mode-seeking updates favorable for RL. An adaptive prefix-selection mechanism further allocates expert guidance based on measured utility. Experiments on five mathematical reasoning benchmarks show that TRAPO consistently surpasses standard SFT, RL, and SFT-then-RL pipelines, as well as recent state-of-the-art approaches, establishing a strong new paradigm for reasoning-enhanced LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。