Do Modules Stay in Their Lane? Role Drift in Compound LLM Systems
Not on the map. Why?
Abstract (arXiv)
End-to-end reinforcement learning can improve the accuracy of compound LLM systems, but it does not constrain how modules divide labor internally. We identify Role Drift, a failure mode in which modules preserve or improve end-task performance while deviating from their assigned roles through role-violating shortcuts that remain invisible to system-level evaluation. To make role drift observable and controllable, we propose Role Anchor, a regularizer that modulates how much each module deviates from its assigned role during end-to-end training. The key idea is to preserve how the role prompt shifts the module's next-token predictions relative to a neutral prompt, which serves as a proxy for the role's intended effect during training. Experiments on two compound LLM pipelines reveal role drift that accuracy alone fails to detect: a decomposer meant to split a question into sub-questions for a separate solver instead plants the answer in them, and a reader meant to answer from retrieved passages instead falls back on parametric memory. In fact, on the decomposer pipeline this shortcut drives most of the apparent RL gain: 86% of it vanishes once the decomposer is held to its role, indicating that terminal accuracy alone can badly overstate how much a compound system has genuinely learned. Across both pipelines, Role Anchor mitigates role drift at a tunable accuracy cost that varies by pipeline and anchor strength. Additional gradient analysis suggests that the regularizer reduces alignment with the role-drift direction rather than simply suppressing learning.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 30 of its 36 references that are in the map. First posted on arXiv 7 Jul 2026.
Builds on
30 of 36 references.
References · 30
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2023Let's Verify Step by Stepcited by 3,789
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
- 2021A General Language Assistant as a Laboratory for Alignmentcited by 1,055
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
18 more
- 2023Towards Understanding Sycophancy in Language Modelscited by 1,040
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
- 2022Solving math word problems with process- and outcome-based feedbackcited by 901
- 2023DSPy: Compiling Declarative Language Model Calls into Self-Improving Pipelinescited by 832
- 2020Underspecification Presents Challenges for Credibility in Modern Machine Learningcited by 496
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 482
- 2024AFlow: Automating Agentic Workflow Generationcited by 432
- 2025LLMs Get Lost In Multi-Turn Conversationcited by 429
- 2024TextGrad: Automatic "Differentiation" via Textcited by 273
- 2024KTO: Model Alignment as Prospect Theoretic Optimizationcited by 264
- 2022Defining and Characterizing Reward Hackingcited by 203
- 2022Learning by Distilling Contextcited by 138
- 2024Trace is the Next AutoDiff: Generative Optimization with Rich Feedback, Execution Traces, and LLMscited by 69
- 2025Consistently Simulating Human Personas with Multi-Turn Reinforcement Learningcited by 39
- 2022Prompt Injection: Parameterization of Fixed Inputscited by 36
- 2025Optimas: Optimizing Compound AI Systems with Globally Aligned Local Rewardscited by 24
- 2024Generative Prompt Internalizationcited by 8
- 2025Aligning Compound AI Systems via System-level DPOcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025The Illusion of Role Separation: Hidden Shortcuts in LLM Role Learning (and How to Fix Them)
- 2026OPSRD: On-Policy Self-Role Distillation
- 2026Roles with Rails: Contract-Preserving Role Evolution in Multi-Agent Structured Reasoning
- 2026Role-Agent: Bootstrapping LLM Agents via Dual-Role Evolution
- 2026When Does Multi-Agent RL Improve LLM Workflows? Workflow, Scale, and Policy-Sharing Tradeoffs
- 2026Does RL Expand the Capability Boundary of LLM Agents? A PASS@(k,T) Analysis
- 2026Maestro: Reinforcement Learning to Orchestrate Hierarchical Model-Skill Ensembles
- 2026Set-shifting Behavioral Test for Harnessed Agents