RAST-MoE-RL: A Regime-Aware Spatio-Temporal MoE Framework for Deep Reinforcement Learning in Ride-Hailing
Yuhan Tang, Kangxin Cui, Jung Ho Park, Yibo Zhao, Xuan Jiang, Haoze He, Jiangbo Yu, Haris Koutsopoulos, Jinhua Zhao
Abstract
Ride-hailing platforms must balance 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, exposes an inherent trade-off between matching and pickup delays. The non-stationary nature of request arrivals and dynamic congestion exposes a key limitation of existing methods, which rely on shallow encoders that cannot capture these spatio-temporal regime shifts. 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 Mixture-of-Experts encoder, letting different experts specialize automatically across operational regimes while keeping per-sample computation modest. On real-world TNC trajectory data from San Francisco, our 12M-parameter model reduces average matching delay by 10% and pickup delay by 15%, trains stably without reward hacking, generalizes zero-shot to two additional cities, and exhibits clear expert specialization across demand regimes. These results demonstrate the value of MoE-enhanced RL for large-scale decision-making tasks with complex spatio-temporal dynamics.