Reinforcement Learning for AMR Charging Decisions: The Impact of Reward and Action Space Design
Janik Bischoff, Alexandru Rinciog, Anne Meyer
TL;DR
This work tackles the problem of optimizing AMR charging in large-scale block-stacking warehouses using reinforcement learning. It extends the SLAPStack framework with battery management, formulates a Markov decision process for joint charging decisions and durations, and evaluates multiple reward and action-space configurations using PPO on the WEPAStacks benchmark. The results reveal a trade-off: flexible, unconstrained RL designs can uncover high-performing strategies but may converge slowly and be unstable, whereas domain-guided reward shaping and reduced action spaces improve learning stability but may limit exploration and generalization. Overall, RL-based charging strategies can outperform strong heuristics on realistic, large-scale instances, with the choice of reward design and action masking playing a pivotal role in practical performance and reproducibility.
Abstract
We propose a novel reinforcement learning (RL) design to optimize the charging strategy for autonomous mobile robots in large-scale block stacking warehouses. RL design involves a wide array of choices that can mostly only be evaluated through lengthy experimentation. Our study focuses on how different reward and action space configurations, ranging from flexible setups to more guided, domain-informed design configurations, affect the agent performance. Using heuristic charging strategies as a baseline, we demonstrate the superiority of flexible, RL-based approaches in terms of service times. Furthermore, our findings highlight a trade-off: While more open-ended designs are able to discover well-performing strategies on their own, they may require longer convergence times and are less stable, whereas guided configurations lead to a more stable learning process but display a more limited generalization potential. Our contributions are threefold. First, we extend SLAPStack, an open-source, RL-compatible simulation-framework to accommodate charging strategies. Second, we introduce a novel RL design for tackling the charging strategy problem. Finally, we introduce several novel adaptive baseline heuristics and reproducibly evaluate the design using a Proximal Policy Optimization agent and varying different design configurations, with a focus on reward.
