Taming the Adversary: A Cost-to-Disturbance Ratio Approach to Adversarial Reinforcement Learning
Organizations: School of Electrical Engineering, Korea Advanced Institute of Science and Technology (KAIST), Daejeon 34141, Republic of Korea
Abstract
Reinforcement learning (RL) policies trained in simulation often degrade once deployed on real systems, where the controller must reject external disturbances that were never encountered in simulation. Robust RL addresses this by exposing the controller to perturbations while it learns, through domain randomization, adversarial minimax formulations, or probabilistic mixtures of protagonist and adversarial behavior. However, an unregulated disturbance mechanism destabilizes training and often collapses nominal performance relative to standard, non-robust methods. We propose cost-to-disturbance ratio adversarial training (CoDRA), a framework that expresses the controller--adversary trade-off as a ratio of accumulated cost to accumulated squared disturbance norm, and optimizes it through a self-normalized actor--critic update. In this algorithm, each value term is scaled by a stop-gradient normalization constant computed from the current batch. This moderates the adversary's incentive without altering the controller's own update, and requires neither an explicit disturbance penalty nor an auxiliary trade-off parameter. We evaluate CoDRA on two MuJoCo pendulum environments under force and mass sweeps. On InvertedDoublePendulum, CoDRA attains the lowest cost at every force level, including a force outside the range seen during training, and in all but one cell of the mass grid, whereas its advantage is less pronounced on the milder InvertedPendulum.
Figures & tables
| InvertedPendulum | InvertedDoublePendulum | |
| State dimension | ||
| Action range | ||
| Motor gear | ||
| Integrator, timestep | RK4, s | RK4, s |
| Frame skip | ||
| Control period | s | s |
| Hidden layers | , ReLU |
|---|---|
| Output layer | , scaled to the action/force bound |
| Optimizer | Adam |
| Actor / adversary learning rate | |
| Critic learning rate | |
| Discount factor | |
| Mini-batch size |
| Algorithm | Nominal | Force | Worst mass cell |
|---|---|---|---|
| DDPG [ 3 ] | 0.61 | 28.76 | 19.40 |
| DR-DDPG | 0.78 | 53.59 | 11.78 |
| RARL [ 6 ] | 0.59 | 28.48 | 21.85 |
| PR-DDPG [ 13 ] | 4.24 | 43.33 | 20.28 |
| NR-DDPG [ 13 ] | 8.17 | 34.66 | 28.73 |
| CoDRA (ours) | 1.91 | 34.14 | 16.67 |
| Algorithm | Nominal | Force | Worst mass cell |
|---|---|---|---|
| DDPG [ 3 ] | |||
| DR-DDPG | |||
| RARL [ 6 ] | |||
| PR-DDPG [ 13 ] | |||
| NR-DDPG [ 13 ] | |||
| DDPG-RF |
| Environment | Quantity | CoDRA | Lowest-mean baseline | ||
|---|---|---|---|---|---|
| InvertedDoublePendulum | DDPG-RF, | ||||
| DDPG-RF, | |||||
| RARL, | |||||
| RARL, | |||||
| Mass-grid mean | DDPG-RF, | ||||
| Cell (pole , pole2 ) | PR-DDPG, |