RAST-MoE-RL: A Regime-Aware Spatio-Temporal MoE Framework for Deep Reinforcement Learning in Ride-Hailing
On the map: Mixture-of-experts → MoE in reinforcement learning and robotics
Abstract (arXiv)
Ride-hailing platforms face the challenge of balancing passenger waiting times with overall system efficiency under highly uncertain supply-demand conditions. Adaptive delayed matching, which controls the holding intervals for batched sets of requests and vehicles, reveals an inherent trade-off between matching and pickup delays. The resulting environment with temporally varying request arrival patterns and dynamic congestion calls for more expressive networks with sufficient capacity to capture their non-stationarity. To address the limitations of existing methods that rely on shallow encoders that cannot capture dynamic supply-demand patterns and congestion effects, we introduce the Regime-Aware Spatio-Temporal Mixture-of-Experts (RAST-MoE) framework, which formalizes adaptive delayed matching as a regime-aware Markov Decision Process and equips RL agents with a self-attention MoE encoder. Instead of relying on a single monolithic network, our design allows different experts to specialize automatically in varying operational conditions, improving representation capacity while maintaining per-sample computation efficiency. Despite its modest size of only 12M parameters, our framework consistently outperforms strong baselines. On real-world Uber trajectory data from San Francisco, it reduces average matching delay by 10%, and pickup delay by 15%. In addition, it demonstrates robustness to unseen demand regimes, stable training behavior without reward hacking, and expert specialization to different regimes. This study shows the strength of MoE-enhanced RL for large-scale decision-making tasks with complex spatiotemporal dynamics.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in MoE in reinforcement learning and robotics, one of 12 directions in Mixture-of-experts. Builds on 24 of its 49 references that are in the map. First posted on arXiv 13 Dec 2025.
Builds on
24 of 49 references.
Closest ancestors · 3
Same direction, newest first
- 2024Mixture of Experts in a Mixture of RL settings26 Jun
- 2024Mixtures of Experts Unlock Parameter Scaling for Deep RL13 Feb
- 2021Probabilistic Mixture-of-Experts for Efficient Deep Reinforcement Learning19 Apr
Other references · 21
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,258
- 2016Concrete Problems in AI Safetycited by 1,642
- 2022DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scalecited by 404
- 2018T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Predictioncited by 401
- 2023From Sparse to Soft Mixtures of Expertscited by 248
- 2020Reinforcement Learning for Combinatorial Optimization: A Surveycited by 236
- 2016Sample Efficient Actor-Critic with Experience Replaycited by 188
- 2018A Study on Overfitting in Deep Reinforcement Learningcited by 174
- 2024Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Modelscited by 59
9 more
- 2024JetMoE: Reaching Llama2 Performance with 0.1M Dollarscited by 57
- 2021Reinforcement Learning for Ridesharing: An Extended Surveycited by 35
- 2024No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPOcited by 27
- 2024ST-MoE-BERT: A Spatial-Temporal Mixture-of-Experts Framework for Long-Term Cross-City Mobility Predictioncited by 7
- 2020Real-Time Dispatching of Large-Scale Ride-Sharing Systems: Integrating Optimization, Machine Learning, and Model Predictive Controlcited by 4
- 2022A dynamic multi-region MFD model for ride-sourcing with ridesplittingcited by 2
- 2025Timing the Match: A Deep Reinforcement Learning Approach for Ride-Hailing and Ride-Pooling Servicescited by 2
- 2023Simulating the Integration of Urban Air Mobility into Existing Transportation Systems: A Surveycited by 1
- 2024Large Scale Multi-GPU Based Parallel Traffic Simulation for Accelerated Traffic Assignment and Propagationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026ALIGN-HOLD: Experience Alignment for Real-Time Hold Control in Large-Scale Ride-Hailing Matching at DiDi
- 2022Spatio-temporal Incentives Optimization for Ride-hailing Services with Offline Deep Reinforcement Learning
- 2026RideSkill: A Hierarchical Algorithm for Generalized Ride Sharing with LLM-Driven Automatic Evolution
- 2024An End-to-End Reinforcement Learning Based Approach for Micro-View Order-Dispatching in Ride-Hailing
- 2025LLM-ODDR: A Large Language Model Framework for Joint Order Dispatching and Driver Repositioning
- 2026EXHOLD: Experience-Aware Real-Time Hold Control for Large-Scale Ride-Hailing Matching at DiDi
- 2024Assessing How Ride-hailing Rebalancing Strategies Improve the Resilience of Multi-modal Transportation Systems
- 2025Non-myopic Matching and Rebalancing in Large-Scale On-Demand Ride-Pooling Systems Using Simulation-Informed Reinforcement Learning