ADV-0: Closed-Loop Min-Max Adversarial Training for Long-Tail Robustness in Autonomous Driving
Organizations: The Hong Kong Polytechnic University, Hong Kong SAR, China · Tongji University, Shanghai, China · McGill University, Montreal, QC, Canada
Abstract
Deploying autonomous driving systems requires robustness against long-tail scenarios that are rare but safety-critical. While adversarial training offers a promising solution, existing methods typically decouple scenario generation from policy optimization and rely on heuristic surrogates. This leads to objective misalignment and fails to capture the shifting failure modes of evolving policies. This paper presents ADV-0, a closed-loop min-max optimization framework that treats the interaction between driving policy (defender) and adversarial agent (attacker) as a zero-sum Markov game. By aligning the attacker's utility directly with the defender's objective, we reveal the optimal adversary distribution. To make this tractable, we cast dynamic adversary evolution as iterative preference learning, efficiently approximating this optimum and offering an algorithm-agnostic solution to the game. Theoretically, ADV-0 converges to a Nash Equilibrium and maximizes a certified lower bound on real-world performance. Experiments indicate that it effectively exposes diverse safety-critical failures and greatly enhances the generalizability of both learned policies and motion planners against unseen long-tail risks.
Figures & tables
| Adversary | Replay | IDM | RL Agent | |||
| CR | ER | CR | ER | CR | ER | |
| Replay | - | - | 19.03% | 51.75 | 16.80% | 51.65 |
| Heuristic | 100.00% | 0.00 | 74.70% | 32.12 | 69.64% | 24.90 |
| CAT | 90.08% | 1.03 | 43.13% | 43.47 | 36.84% | 42.26 |
| KING | 23.28% | 47.67 | 24.49% | 49.41 | 21.26% | 49.32 |
| AdvTrajOpt | 69.64% | 3.12 | 26.92% | 45.36 | 28.95% | 45.10 |
| Val. Env. | RC | Crash | Reward | Cost | |
| ADV-0 (w/ IPL) | Replay | 0.742 0.011 | 0.159 0.019 | 49.56 1.05 | 0.480 0.013 |
| ADV-0 | 0.695 0.011 | 0.289 0.029 | 44.60 0.83 | 0.598 0.022 | |
| CAT | 0.704 0.011 | 0.271 0.025 | 45.68 1.16 | 0.585 0.025 | |
| SAGE | 0.699 0.013 | 0.263 0.025 | 45.19 1.54 | 0.567 0.028 | |
| Heuristic | 0.710 0.016 | 0.217 0.021 | 45.41 2.08 | 0.552 0.035 | |
| Avg. | 0.710 0.012 | 0.240 0.024 | 46.09 1.33 | 0.556 0.027 |
| Agent/Adversary | Reward ( ) | ||
| Adversary (w/o IPL) | Adversary (w/ IPL) | Decrease | |
| Agent w/o IPL | |||
| Agent w/ IPL | |||
| Improvement | |||
| Agent/Adversary | Cost ( ) | ||
| Adversary (w/o IPL) | Adversary (w/ IPL) | Decrease | |
| Reactive Traffic | Safety Margin | Stability & Comfort | Defensive Driving | ||||
| Avg Min-TTC ( ) | Avg Min-PET ( ) | Mean Abs Jerk ( ) | 95% Jerk ( ) | Near-Miss Rate ( ) | RDP Violation Rate ( ) | ||
| ADV-0 (w/ IPL) | |||||||
| CAT | |||||||
| Heuristic | |||||||
| Model | RC | Crash | Reward | Cost |
| PlanTF | 0.628 0.025 | 0.357 0.031 | 35.85 2.51 | 1.04 0.04 |
| + ADV-0 | 0.674 0.016 | 0.263 0.025 | 41.98 1.62 | 0.77 0.02 |
| Rel. Change | +7.46% | -26.23% | +17.11% | -25.68% |
| SMART | 0.587 0.029 | 0.396 0.034 | 32.66 2.85 | 1.15 0.05 |
| + ADV-0 | 0.631 0.015 | 0.305 0.023 | 37.85 1.70 | 0.92 0.02 |
| Rel. Change | +7.57% | -22.88% | +15.86% | -20.31% |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| ADV-0 Variation | Replay | IDM | RL Agent | |||
| CR | Reward | CR | Reward | CR | Reward | |
| Pretrained (logit-based sampling) | ||||||
| Pretrained (energy-based sampling) | ||||||
| GRPO Finetuned | ||||||
| PPO Finetuned | ||||||
| SAC Finetuned | ||||||
| Model Phase | Val. Env. | RC | Crash | Reward | Cost |
| PlanTF (Pretrained) | Replay | 0.727 0.021 | 0.220 0.027 | 41.53 3.10 | 0.76 0.03 |
| ADV-0 | 0.581 0.030 | 0.420 0.035 | 32.11 2.54 | 1.24 0.05 | |
| CAT | 0.615 0.025 | 0.385 0.030 | 35.26 2.16 | 1.09 0.03 | |
| SAGE | 0.595 0.028 | 0.407 0.032 | 33.80 2.80 | 1.15 0.04 | |
| Heuristic | 0.620 0.020 | 0.352 0.032 | 36.53 1.97 | 0.94 0.03 | |
| Avg. | 0.628 0.025 | 0.357 0.031 | 35.85 2.51 | 1.04 0.04 |
| Method | Progress | Collision | Off-road | Comfort | Speed Eff. | Total Score |
| ( + ) | ( - ) | ( - ) | ( - ) | ( - ) | ||
| PlanTF (Pretrained) | 14.17 1.53 | -2.85 0.92 | -0.12 0.05 | -0.34 0.08 | -0.54 0.12 | 10.32 1.79 |
| PlanTF (Fine-tuned) | 16.45 0.85 | -0.89 0.45 | -0.05 0.04 | -0.24 0.05 | -0.64 0.13 | 14.63 0.97 |
| Improvement | +16.09% | +68.77% | +58.33% | +29.41% | -18.52% | +41.76% |
| SMART (Pretrained) | 13.20 1.80 | -3.61 1.11 | -0.25 0.08 | -1.26 0.25 | -0.60 0.15 | 7.48 2.14 |
| SMART (Fine-tuned) | 15.10 1.10 | -1.75 0.37 | -0.17 0.05 | -0.92 0.19 | -0.65 0.11 | 11.61 1.18 |
| Scenario Category | Percentage | Reactive Traffic | Safety Margin | Stability & Comfort | Defensive Driving | ||||
| Avg Min-TTC ( ) | Avg Min-PET ( ) | Mean Abs Jerk ( ) | 95% Jerk ( ) | Near-Miss Rate ( ) | RDP Violation Rate ( ) | ||||
| ADV-O (w/ IPL) | Critical TTC | 7.40% | |||||||
| Critical PET | 3.40% | ||||||||
| Hard Dynamics | 3.20% | N/A | |||||||
| Rare Cluster | 5.20% | ||||||||
| Avg | |||||||||
| Module | Parameter | Value |
| Backbone (DenseTNT) | Hidden Size | 128 |
| Sub-graph Depth | 3 | |
| Global-graph Depth | 1 | |
| Trajectory Modes ( ) | 32 | |
| NMS Threshold | 7.2 | |
| IPL Fine-tuning | Learning Rate | 5e-6 |