LLM-Based Hierarchical Coordinated Control with Continuation-Aware Policy Learning
Organizations: Beihang University, Beijing, China · Baidu, Inc., Beijing, China
Abstract
Coordinating multiple interacting units in complex engineering systems is challenging when system interactions are difficult to model, operational information is heterogeneous, and low-level actions must satisfy strict constraints. We propose an LLM-based hierarchical framework in which the LLM coordinates interacting units based on heterogeneous operational context, while task-specific controllers or optimizers generate executable and constraint-aware actions. We further introduce Continuation-Aware GRPO to capture the consequences of coordination decisions over subsequent control intervals. Rather than judging a decision only by its immediate outcome, the method also evaluates how the system evolves afterward under the current policy. We validate the framework on multi-ramp traffic control and virtual power plant (VPP) energy management, using simplified system models for training and more realistic simulators for evaluation. Across both tasks, the proposed method consistently outperforms direct task-specific control and optimization, end-to-end reinforcement learning, rule-based and RL-based hierarchical coordination, and prompting-only LLM coordinators, demonstrating the value of heterogeneous-context reasoning, hierarchical execution, and continuation-aware policy learning.
Figures & tables
| Method | Seen | Unseen | High-U | Overall | Gain |
| Direct Control | |||||
| Feedback Control | 16283.4 41.8 | 15576.8 50.6 | 14917.3 68.3 | 15592.5 49.1 | – |
| End-to-End RL | 16537.6 163.7 | 15684.9 191.5 | 14813.2 228.4 | 15678.6 181.2 | +0.55% |
| Hierarchical Coordination | |||||
| Fixed-Mode Hierarchy | 16502.8 36.2 | 15792.4 44.9 | 15158.7 60.5 | 15818.0 43.7 | +1.45% |
| Hierarchical RL | 16843.1 98.6 | 16117.6 121.4 | 15434.8 151.8 | 16131.8 110.2 | +3.46% |
| Method | Seen | Unseen | High-U | Overall | Gain |
| Direct Control | |||||
| MPC-MILP | 32783.5 47.2 | 33887.9 65.8 | 35843.3 111.4 | 34171.6 64.2 | – |
| End-to-End RL | 35076.9 512.4 | 37423.6 647.3 | 40947.2 923.8 | 37815.9 608.5 | |
| Hierarchical Coordination | |||||
| Fixed-Mode Hierarchy | 32984.7 52.6 | 34052.2 71.9 | 35796.4 119.8 | 34277.8 70.6 | |
| Hierarchical RL | 32183.7 267.5 | 33237.2 351.6 | 35124.6 505.2 | 33515.2 317.4 | +1.92% |
| Multi-Ramp Traffic Control | Virtual Power Plant | |||||
| Method | LTM | SUMO | Gap | Fast Simulator | OpenDSS | Gap |
| Qwen3-8B | 16312.6 33.4 | 15270.7 55.1 | 6.39 0.21% | 34025.2 87.9 | 36876.2 144.7 | 8.38 0.26% |
| End-to-End RL | 17438.7 147.8 | 15678.6 181.2 | 10.09 1.07% | 33785.3 438.6 | 37815.9 608.5 | 11.94 2.03% |
| Hierarchical RL | 17043.9 91.5 | 16131.8 110.2 | 5.35 0.54% | 30684.7 229.4 | 33515.2 317.4 | 9.23 0.95% |
| Ours | 17286.4 66.7 | 16625.4 84.5 | 3.82 0.33% | 30941.8 142.7 | 32679.2 223.6 | 5.62 0.54% |
| Variant | Seen | Unseen | High-U | Overall | Gain |
| GRPO ( ) | 16861.4 112.8 | 16327.6 151.3 | 15767.1 211.6 | 16318.7 136.4 | – |
| Ours ( ) | 17043.6 75.9 | 16617.9 92.7 | 16214.7 118.6 | 16625.4 84.5 | +1.88% |
| Parameter | Value |
| m/s | |
| m/s | |
| per-lane jam density | veh/m |
| per-lane capacity | veh/s |
| lane efficiency | |
| LTM internal step | s |
| Mode | |
| Sensitive | |
| Standard | |
| Sluggish | |
| Always-Open | — |
| Parameter | Value | Parameter | Value |
| kWh | kWh/ ∘ C | ||
| kW/ ∘ C | |||
| kW | |||
| SoC range | band | C | |
| kW | C | ||
| kW | kW |
| Resource | Cons. | Bal. | Pro. | |
| PV | ||||
| Battery | ||||
| Battery | ||||
| EV | ||||
| HVAC |
| Category | Multi-ramp traffic control | Virtual power plant |
| Numerical measurements | per-ramp flow , speed , occupancy ; gate state | , PV power, base load, indoor and outdoor temperature, setpoint, import cap, time-of-use price |
| Recent trends | -step series of , , over the last min | -step series of PV, base load, outdoor temperature over the last min |
| Forecasts | mainline inflow and per-ramp arrivals over the next interval, with stated reliability | PV, base load, and outdoor temperature over the MPC window, with stated reliability |
| Uncertainty estimates | corridor demand-uncertainty level | PV uncertainty level |
| Resource urgency | minutes since last closure, per ramp | residual EV energy, urgent share , required charging pace |
| Operational priorities | per-ramp service priority tier | HVAC comfort priority |
| Setting | Value |
| group size | |
| horizon | ( min / min) |
| clip | |
| KL coefficient | |
| importance ratio | sequence level |
| Setting | Value |
| trunk | MLP , |
| clip | |
| discount | |
| GAE | |
| learning rate | |
| entropy coefficient |
| Ramp | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
| SUMO | ||||||||
| LTM |
| Aspect | Training model | Evaluation model |
| Multi-ramp traffic control | ||
| Representation | link-level cumulative counts, triangular fundamental diagram | individual vehicles with car-following and lane-change behaviour |
| Omitted in training | lane changing, weaving in the merge influence area, driver heterogeneity, stochastic gap acceptance | — |
| Observables | link flow, speed, density from the state update | detector counts and occupancies over s windows |
| Virtual power plant | ||
| Representation | single-bus power balance | three-phase feeder power flow on EPRI Ckt5 |
| Series | Source | Coverage | Transformation |
| Multi-ramp traffic control | |||
| Corridor demand | expressway detector records | northbound Western Expressway, Changchun | per-origin episode totals shaped by the sampled temporal profile and scaled by the level multipliers of Table 17 |
| Virtual power plant | |||
| PV generation | NASA POWER, all-sky surface shortwave irradiance | Suzhou, 2020, hourly | scaled to the kW array, interpolated to min |
| Outdoor temperature | NASA POWER, 2 m air temperature | Suzhou, 2020, hourly | used directly as , interpolated to min |
| Park base load | Building Data Genome 2, cleaned electricity meters | office buildings, 2016–2017, hourly | season-conditional hour-of-day profile, normalised then mapped to kW |
| Tier | Hours | Price (CNY/kWh) |
| Valley | – | |
| Shoulder | – , – | |
| Peak | – , – | |
| Critical peak | – |
| Family | Binding pressure |
| Multi-ramp traffic control | |
| Upstream-dominant | strong mainline, weak ramps |
| Ramp-dominant | moderate mainline, several strong ramps |
| Balanced moderate | no single binding location |
| Balanced heavy | mainline and ramps both near capacity |
| Virtual power plant | |
| Weak | Medium | Strong | |
| Mainline multiplier | |||
| Ramp multiplier | |||
| Ramp rate cap (veh/h) |
| Component | MPC-MILP | Ours |
| Energy purchased at the tariff | ||
| Import above the contracted cap | ||
| EV energy undelivered at deadline | ||
| Comfort-band excursion | ||
| Total |
| Method | Bottleneck speed (km/h) | Congestion duration (min) | Mean ramp queue (veh) | Spillover events |
| Feedback Control | ||||
| End-to-End RL | ||||
| Fixed-Mode Hierarchy | ||||
| Hierarchical RL | ||||
| Qwen3-8B | ||||
| Gemini 3.1 Pro |
| Method | Violation steps | Curtailed (kWh) | Unmet EV (kWh) | Excursion ( ∘ C h) |
| MPC-MILP | ||||
| End-to-End RL | ||||
| Fixed-Mode Hierarchy | ||||
| Hierarchical RL | ||||
| Qwen3-8B | ||||
| Gemini 3.1 Pro |
| Method | Corridor | VPP |
| Qwen3-8B | ||
| Gemini 3.1 Pro | ||
| Claude Sonnet 4.5 | ||
| Ours |
| Multi-ramp traffic control | Virtual power plant | |||||
| Method | LTM | SUMO | Gap | Fast model | OpenDSS | Gap |
| Direct control | ||||||
| End-to-End RL | ||||||
| Fixed-Mode Hierarchy | ||||||
| Hierarchical RL | ||||||
| Qwen3-8B | ||||||
| Multi-ramp traffic control: throughput | Virtual power plant: cost | |||||
| Horizon | High-U | Overall | Gain | High-U | Overall Gain | Train time |
| — | — | |||||
| , data | ||||||
| Withheld category | Throughput | VPP cost |
| None (ours) | ||
| Recent trends | ||
| Forecasts | ||
| Uncertainty estimates | ||
| Priorities and events |
| Context supplied | Throughput | VPP cost |
| Full | ||
| forecasts | ||
| uncertainty | ||
| Measurements only |
| Comparison | Difference | interval |
| Throughput, vehicles per episode | ||
| vs. Feedback Control | ||
| vs. Fixed-Mode Hierarchy | ||
| vs. End-to-End RL | ||
| vs. Hierarchical RL | ||
| vs. Gemini 3.1 Pro | ||
| Corridor roles, share of all ramp-steps | ||||
| Sensitive | Standard | Sluggish | Always-Open | |
| Ours | ||||
| Qwen3-8B | ||||
| VPP modes (ours), share of all resource-steps | ||||
| Resource | Conservative | Balanced | Proactive | |
| PV | ||||
| Corridor: Sensitive / Sluggish share by occupancy trend | |||
| Rising | Flat | Falling | |
| Sensitive | |||
| Sluggish | |||
| VPP: proactive-battery share by tariff tier | |||
| Shoulder Peak Critical | |||
| Phase | Bottleneck occ. | Sens. | Slug. | Open |
| Warm-up ( – min) | ||||
| Rise ( – min) | ||||
| Plateau ( – min) | ||||
| Decline ( – min) | ||||
| Recovery ( – min) |
| Pattern family | Baseline | Gain |
| Corridor, throughput vs. Feedback Control | ||
| Upstream-dominant | ||
| Ramp-dominant | ||
| Balanced moderate | ||
| Balanced heavy | ||
| VPP, cost vs. MPC-MILP | ||
Language Model-Based Control
Language model-based control uses language models to interpret goals, choose actions, and steer AI systems or physical processes.
See all 180 papers