mlsanityData to 7 Oct 2026

Regime-Conditional Stabilisation of LLM-Augmented Cooperative Multi-Agent Reinforcement Learning

arXiv 2607.04470 ↗First posted 5 Jul 2026cs.LG cs.AI math.OCcited by 0

On the map: LLM agents and tool use → Reinforcement learning for LLM agents

Abstract (arXiv)

Large Language Models (LLMs) offer a natural interface for translating human objectives into reward signals for cooperative multi-agent reinforcement learning (MARL), yet the training-time dynamics of this integration remain poorly understood. We show that dynamically updating LLM-generated reward weights during off-policy MARL violates the stationarity assumption of Potential-Based Reward Shaping (PBRS) and contaminates the experience replay buffer, whose stored transitions carry reward labels computed under stale shaping weights. We characterise the result as a regime-dependent failure whose severity depends on how competent the unshaped baseline already is. To control it we propose two stabilisation strategies: a Phase-Based Freeze Schedule that enforces strict stationarity within training phases, and Exponential Moving Average (EMA) smoothing that bounds per-episode weight drift. We evaluate across three cooperative environments and five random seeds with QMIX, complemented by an exploratory VDN extension, yielding a three-regime taxonomy. In the augmentative regime (Simple Spread), where the baseline is functional (74.4 %), EMA significantly improves success to 86.7 % (+12.3 pp, p<0.01) while naive dynamic updates collapse it to 15.2 %. In the essential regime (Level-Based Foraging), where the baseline is broken (0.1 %), any shaping unlocks the task (95.9 % under EMA). In the supplementary regime (SMAC 3m), where the baseline is near-saturated (98.8 %), stabilised shaping preserves performance (99.9 %) while unstabilised shaping adds variance without gain. These findings establish reward-signal stationarity as a necessary design constraint and indicate that regime placement is a practical predictor of whether dynamic LLM shaping helps or harms.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 17 of its 25 references that are in the map. First posted on arXiv 5 Jul 2026.

Builds on

17 of 25 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025LAMARL: LLM-Aided Multi-Agent Reinforcement Learning for Cooperative Policy Generation2 Jun

Other references · 16

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
  3. 2017Deep reinforcement learning from human preferencescited by 3,846
  4. 2017Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environmentscited by 1,519
  5. 2021The Surprising Effectiveness of PPO in Cooperative, Multi-Agent Gamescited by 903
  6. 2018QMIX: Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement Learningcited by 697
  7. 2019The StarCraft Multi-Agent Challengecited by 589
  8. 2023Eureka: Human-Level Reward Design via Coding Large Language Modelscited by 542
  9. 2019QTRAN: Learning to Factorize with Transformation for Cooperative Multi-Agent Reinforcement Learningcited by 372
  10. 2023Language to Rewards for Robotic Skill Synthesiscited by 307
  11. 2020QPLEX: Duplex Dueling Multi-Agent Q-Learningcited by 267
  12. 2023Reward Design with Language Modelscited by 257
4 more
  1. 2020Benchmarking Multi-Agent Deep Reinforcement Learning Algorithms in Cooperative Taskscited by 207
  2. 2023Text2Reward: Reward Shaping with Language Models for Reinforcement Learningcited by 153
  3. 2017Stabilising Experience Replay for Deep Multi-Agent Reinforcement Learningcited by 114
  4. 2017A Survey of Learning in Multiagent Environments: Dealing with Non-Stationaritycited by 93

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practices
  2. 2026LLM-ALSO: LLM-Driven Adaptive Learning-Signal Optimization for Multi-Agent Reinforcement Learning
  3. 2026Policy-Invariant Reward Shaping from LLM Feedback: A Framework for Hybrid RL Agents
  4. 2024Coevolving with the Other You: Fine-Tuning LLM with Sequential Cooperative Multi-Agent Reinforcement Learning
  5. 2023Shattering the Agent-Environment Interface for Fine-Tuning Inclusive Language Models
  6. 2026Large Language Model Guided Incentive Aware Reward Design for Cooperative Multi-Agent Reinforcement Learning
  7. 2025Stabilizing Off-Policy Training for Long-Horizon LLM Agent via Turn-Level Importance Sampling and Clipping-Triggered Normalization
  8. 2025Zero-Shot LLMs in Human-in-the-Loop RL: Replacing Human Feedback for Reward Shaping