SOLAR for Offline MARL: Plateau-Triggered Potential Shaping under World-Model Uncertainty
Jusheng Zhang, Yijia Fan, Ruiqi Chen, Jing Yang, Ziliang Chen, Yongsen Zheng, Yanxi Chen, Jian Wang, Kwok Yan Lam, Liang Lin, Keze Wang
Abstract
Reward shaping can accelerate reinforcement learning, but in sparse-reward offline multi-agent RL it is often brittle: dense intrinsic rewards may alter the underlying Markov game, while world-model guidance can amplify model bias. We find that shaping becomes reliable when it is (i) activated only after statistically validated learning plateaus and (ii) constrained to potential-based shaping, which preserves the task optimum. Motivated by this, we propose SOLAR, a simulate--evaluate--shape framework. A learned world model enables low-cost rollouts to test plateaus; once a plateau is detected, we inject shaping in the form with adaptively updated potentials; and we attenuate shaping using uncertainty-aware throttling in unreliable regions. We provide theoretical analysis on policy invariance and on the deviation of plateau decisions under model error, and establish stability for the resulting two-timescale adaptation. Experiments on sparse-reward offline MARL benchmarks show consistent gains in stability and final performance across dataset qualities.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c3622a76-e3f0-45e8-ad2e-487d27e97ff8Builds on9
- Conservative Q-Learning for Offline Reinforcement LearningAviral Kumar, Aurick Zhou, George Tucker, Sergey LevineNeurIPS 2020 · 2,881 citations
- Offline Reinforcement Learning with Implicit Q-LearningIlya Kostrikov, Ashvin Nair, Sergey LevineICLR 2022 · 1,402 citations
- Believe What You See: Implicit Constraint Approach for Offline Multi-Agent Reinforcement LearningYiqin Yang, Xiaoteng Ma, Chenghao Li, Zewu Zheng et al.NeurIPS 2021 · 133 citations
- MADiff: Offline Multi-agent Learning with Diffusion ModelsZhengbang Zhu, Minghuan Liu, Liyuan Mao, Bingyi Kang et al.NeurIPS 2024 · 116 citations
- Plan Better Amid Conservatism: Offline Multi-Agent Reinforcement Learning with Actor RectificationLing Pan, Longbo Huang, Tengyu Ma, Huazhe XuICML 2022 · 80 citations
Related papers
- Exploration-Guided Reward Shaping for Reinforcement Learning under Sparse RewardsRati Devidze, Parameswaran Kamalaruban, Adish SinglaNeurIPS 2022 · 122 citations
- Learning to Shape Rewards Using a Game of Two PartnersDavid Mguni, Taher Jafferjee, Jianhong Wang, Nicolas Perez Nieves et al.AAAI 2023 · 17 citations
- Bootstrapped Reward ShapingJacob Adamczyk, Volodymyr Makarenko, Stas Tiomkin, Rahul V. KulkarniAAAI 2025 · 7 citations
- Reward Shaping Control Variates for Off-Policy Evaluation Under Sparse RewardsRitam Majumdar, Finale Doshi-Velez, Sonali ParbhooICML 2026 · 3 citations
- Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement LearningHaozhe Ma, Zhengding Luo, Thanh Vinh Vo, Kuankuan Sima et al.NeurIPS 2025 · 9 citations
