Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning Structured Reasoning via Tractable Trajectory Control

About

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.

Po-Nien Kung, Zhen Yang, Jeffrey Luo, Cheng-Fu Yang, Haikang Deng, Zi-Yi Dou, Yinfei Yang, Nanyun Peng, Zhe Gan, Kai-Wei Chang• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVision
Accuracy27.14
144
Math ReasoningMATH500
Accuracy89.2
51
Math ReasoningMathVista
Pass@170.9
24
Math ReasoningAIME 2024
Accuracy @1640
10
Math ReasoningAMC 23
Acc@1682.18
10
Math ReasoningOlympiad
Accuracy@158.6
10
Math ReasoningMinerva
Acc@136.39
10
Math ReasoningMathVerse
Pass@142.72
8
Math ReasoningMMMU-Pro
Accuracy@144.72
5
Math ReasoningEMMA
Accuracy@129.93
5
Showing 10 of 12 rows

Other info

Follow for update