SRL-MPC: Shape-Aware Reinforcement Learned Model Predictive Control

TL;DR: SRL-MPC combines shape-aware high-order control barrier functions (HOCBFs) with reinforcement learning for online MPC parameter adaptation, enabling safe and efficient navigation of heterogeneous robot shapes in dense crowds without geometry simplification or policy retraining.

Abstract

Safe and efficient shape-aware navigation in heterogeneous crowds and robot fleets remains challenging. Traditional approaches often assume homogeneous robots, sparse workspaces, simplified geometry, offline computation, or handcrafted parameters to make the problem tractable, which limits their deployment in dense crowd scenarios. Toward this end, we propose Shape-Aware Reinforcement Learned Model Predictive Control (SRL-MPC), a method for safe, efficient, and adaptive navigation in crowds with heterogeneous shapes without geometry simplification.

SRL-MPC formulates high-order control barrier function residuals from geometric separation features based on support-function geometry. A reinforcement learning policy reads these local features and produces real-time MPC parameter updates, while the executed controls remain the solution of an explicit model-predictive optimization problem. Randomized crowd experiments with arbitrarily shaped robot fleets demonstrate strong effectiveness, scalability, and robustness, especially as crowd density increases.

Architecture

SRL-MPC architecture from crowd geometry through geometric separation features, reinforcement learned parameter adaptation, HOCBF-MPC, and robot navigation

Key Advantages and Features

1. Shape-aware geometry: compact geometric separation features represent circular and convex-polygon bodies without reducing every robot to a disc.

2. Adaptive planning: each learned policy updates path tracking, control effort, and safety-distance parameters from local crowd geometry.

3. Model-based execution: reinforcement learning tunes the planner; the action is still produced by an explicit local MPC problem with bounded controls.

4. Dense-scene robustness: independently trained policies use the same 15-robot training distribution and are evaluated in dense 20- and 25-robot scenes.

SRL-MPC reinforcement learning weight adapter architecture

Demonstrations

All demonstrations come from held-out, randomly generated scenes rather than hand-designed layouts. Each scenario shows two independent episodes at 2× speed.

Density-Sweep Demonstrations

Each episode independently samples random robot start and goal positions together with random convex robot shapes; the policy is deployed unchanged from sparse to dense scenes.

Random episodeA
Random episodeB
10 robotsSparse crowdeval.100%
Random episodeA
Random episodeB
15 robotsTraining densityeval.98%
Random episodeA
Random episodeB
20 robotsDense crowdavg.91.0 ± 2.6%
Random episodeA
Random episodeB
25 robotsHighest tested densityavg.86.7 ± 0.6%

Out-of-Distribution Demonstrations

Each OOD family randomly resamples its configuration-specific starts, goals, and footprints for every episode, without policy retraining.

Random episodeA
Random episodeB
Nonconvex union15 differential-drive robots · random nonconvex unions · exchangeavg.96.7 ± 0.6%
Random episodeA
Random episodeB
Cross geometry15 robots · 4 rectangles + circle + convex polygonsavg.59.3 ± 10.8%
Random episodeA
Random episodeB
Circular polygon exchange15 convex polygons · radius 4.5 m · circular swapavg.46.0 ± 0.0%
Random episodeA
Random episodeB
Through traffic5 controlled robots + 5 wandering polygon obstaclesavg.90.7 ± 1.5%
Random episodeA
Random episodeB
Heterogeneous dynamics5 differential-drive polygons + 5 omnidirectional circlesavg.95.0 ± 1.7%

Multi-seed evaluation · 300 episodes per setting

Dense-scene Evaluation

Three independently trained policies, each evaluated for 100 episodes per dense setting. Values report mean ± sample SD across training seeds.

25-robot success 86.7% ±0.6 percentage points across three policies*
Key comparison +65.7 pp

above SARL at 21%, the strongest external baseline in the 25-robot setting.

N = 20 91.0% ±2.6 pp

20-robot success

N = 25 86.7% ±0.6 pp

25-robot success*

Dense macro 88.8% ±1.5 pp

mean over N = 20 and N = 25

Directly labeled slope chart comparing success rates for SRL-MPC and five baselines as robot count increases from 20 to 25

25-robot baseline comparison

Success rate
MethodSuccess
SRL-MPC86.7 ± 0.6%*
SARL21%
AVOCADO11%
VO-polytope8%
ORCA7%
RL-RVO1%

Rebuttal evaluation · no retraining

Out-of-Distribution Evaluation

Three policies trained independently in the 15-robot random-convex-polygon setting are evaluated without retraining for 100 episodes each in five distribution shifts.

77.5%±2.5 pp · OOD macro across five shifts

Heatmap comparing six methods across five out-of-distribution scenarios using SRL-MPC averages over three training seeds

Preliminary deployment

Real-World Demonstration

A controlled indoor platform comparison of SRL-MPC with ORCA, distributed MPC, and manually tuned MPC.

Side-by-side platform comparisonORCA · Dist-MPC · Manual-MPC · SRL-MPC · 2× playback Preliminary demo