DiffNash: Finding Nash Equilibrium in High-Dimensional Continuous Action Spaces with Diffusion Policies
Organizations: Harvard University · Massachusetts Institute of Technology · Georgia Institute of Technology
Abstract
Computing equilibria in high-dimensional continuous games requires learning and representing rich, multi-modal strategy distributions, something conventional neural policies struggle to achieve. Diffusion models offer a natural solution, but standard diffusion training requires direct samples from a fixed target distribution, which are generally unavailable under iterative game-solving updates. We propose a diffusion-based framework for equilibrium computation that enables diffusion-based variants of Online Mirror Descent (OMD) and Fictitious Play (FP). Our approach derives a reweighted score-matching objective that trains diffusion policies without requiring direct samples from each updated policy. Our main method, DiffNash, combines the fast convergence of OMD with the sample efficiency of FP by reusing past actions through a replay buffer. To correct the resulting distribution shift from sample reuse, we derive an unbiased estimator of the difference between current and past policies' Evidence Lower Bounds, whose exponential provides a tractable proxy for the corresponding importance weight without evaluating diffusion likelihoods. We evaluate our approach on forest protection games and continuous Colonel Blotto games, reducing the Nash gap over the strongest baseline by 35.6% and 24.7%, respectively, with minimal hyperparameter tuning.
Figures & tables
| DeepFP | Neural OMD | DiffFP | DiffNash (Ours) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| Forest | Gap | Gap | Gap | Gap | |||||
| F1 | 1 | ||||||||
| F1 | 3 | ||||||||
| F2 | 1 | ||||||||
| F2 | 3 | ||||||||
| F3 | 5 | ||||||||
| Equilibrium | DeepFP | Neural OMD | DiffFP | DiffNash (Ours) | |
|---|---|---|---|---|---|
| 6 | Symmetric | ||||
| 6 | Radial band | ||||
| 10 | Symmetric | ||||
| 10 | Radial band |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| F2 | F3 | ||||
| Setting | Value | ||||
| Default | – | 110.1 | 126.4 | 76.4 | 142.7 |
| Own batch | 128 | 388.9 62.0 | 358.7 88.2 | 250.5 | 510.5 271.8 |
| 512 | 441.5 53.0 | 431.5 66.7 | 669.2 87.8 | 631.4 75.8 | |
| Opponent batch | 64 | 442.0 179.8 | 419.5 200.5 | 679.8 289.7 | 667.2 292.1 |
| 256 | 397.9 129.4 | 385.5 141.3 | 576.7 209.6 | 485.0 246.6 | |
Nash Equilibrium Computation
Nash equilibrium computation develops algorithms for finding stable strategies in strategic, stochastic, and imperfect-information games.
See all 69 papers