Learning Structured Reasoning via Tractable Trajectory Control

LLM Latent Reasoning CoT ML RL PGPO
大语言模型可展现出涌现式的推理行为,这类行为通常体现为反复出现的词汇模式(例如,“等等”一词即常表示验证环节)。然而,在无约束采样下,复杂的推理路径仍十分稀疏;而标准的强化学习方法往往难以确保模型习得多样化的推理行为。为此,我们提出一种通过结构化推理实现多样化推理模式的系统性发现与强化方法——该范式要求在强化学习过程中有针对性地探索特定的推理模式。为此,我们提出了 Ctrl-R 框架:一种基于可处理轨迹控制的结构化推理学习框架,它能主动引导 rollout 过程,激励模型探索对解决复杂问题至关重要的多样化推理模式。由此获得的行为策略支持精确的重要性采样估计,从而保障无偏的策略内优化(on-policy optimization)。此外,我们在重要性采样权重上引入了一个幂次缩放因子(power-scaling factor),使策略能够有选择地从探索性、分布外(out-of-distribution)的轨迹中学习,同时保持优化过程的稳定性。实验表明,Ctrl-R 能够有效促进模型探索并内化以往无法达成的推理模式,在数学推理任务上,显著且一致地提升了语言模型与视觉-语言模型的性能。
Large language models can exhibit emergent reasoning behaviors, often manifested as recurring lexical patterns (e.g., "wait," indicating verification). However, complex reasoning trajectories remain sparse in unconstrained sampling, and standard RL often fails to guarantee the acquisition of diverse reasoning behaviors. We propose a systematic discovery and reinforcement of diverse reasoning patterns through structured reasoning, a paradigm that requires targeted exploration of specific reasoning patterns during the RL process. To this end, we propose Ctrl-R, a framework for learning structured reasoning via tractable trajectory control that actively guides the rollout process, incentivizing the exploration of diverse reasoning patterns that are critical for complex problem-solving. The resulting behavior policy enables accurate importance-sampling estimation, supporting unbiased on-policy optimization. We further introduce a power-scaling factor on the importance-sampling weights, allowing the policy to selectively learn from exploratory, out-of-distribution trajectories while maintaining stable optimization. Experiments demonstrate that Ctrl-R enables effective exploration and internalization of previously unattainable reasoning patterns, yielding consistent improvements across language and vision-language models on mathematical reasoning tasks.
许愿