InjectRBP: Steering Large Language Model Reasoning Behavior via Pattern Injection
Xiuping Wu, Zhao Yu, Yuxin Cheng, Ngai Wong, Liangjun Ke, Tapas Mishra, Konstantinos V. Katsikopoulos
TL;DR
InjectRBP demonstrates that LLM reasoning can be steered by injecting learned behavior patterns without updating model parameters. By modeling reasoning as sequences over a finite behavior alphabet and employing 3-gram injections, the authors design InjectCorrect (self-imitation from correct traces) and InjectRLOpt (MDP-based optimization with a Reliability-Aware Softmax Policy). Across GPQA, MATH, AIME25, and MBPP, these methods yield measurable gains, with InjectRLOpt often outperforming InjectCorrect and a recommended default discount factor $\gamma$ of $0.98$ for larger models. The approach highlights the significance of domain-aligned reasoning motifs and opens avenues for post-hoc reasoning enhancement without architectural changes.
Abstract
Reasoning can significantly enhance the performance of Large Language Models. While recent studies have exploited behavior-related prompts adjustment to enhance reasoning, these designs remain largely intuitive and lack a systematic analysis of the underlying behavioral patterns. Motivated by this, we investigate how models' reasoning behaviors shape reasoning from the perspective of behavioral patterns. We observe that models exhibit adaptive distributions of reasoning behaviors when responding to specific types of questions, and that structurally injecting these patterns can substantially influence the quality of the models' reasoning processes and outcomes. Building on these findings, we propose two optimization methods that require no parameter updates: InjectCorrect and InjectRLOpt. InjectCorrect guides the model by imitating behavioral patterns derived from its own past correct answers. InjectRLOpt learns a value function from historical behavior-pattern data and, via our proposed Reliability-Aware Softmax Policy, generates behavioral injectant during inference to steer the reasoning process. Our experiments demonstrate that both methods can improve model performance across various reasoning tasks without requiring any modifications to model parameters, achieving gains of up to 5.34% and 8.67%, respectively.
