Truncated Differentiation Through Primal-Dual Solvers for Inverse Potential Mean-Field Games

Siting Liu, Yat Tin Chow, Samy Wu Fung

Abstract

We study inverse potential mean-field games (MFGs), in which an unknown spatial inverse-cost (mobility) map is inferred from observed population densities. We solve the forward MFG with a preconditioned primal-dual hybrid gradient (PDHG) method and develop Jacobian-free backpropagation (JFB-$r$), which records only the final $r$ iterations from a detached warm start while retaining the full forward solve. To analyze this truncated differentiation method, we show that the exact-proximal dual-extrapolated PDHG map is a metric resolvent of the maximal monotone KKT operator. This resolvent view shows that JFB-$r$ exactly differentiates a finite-trajectory surrogate and, under a locally fixed active set at an exact equilibrium detach point, converges to the implicit gradient as the tracked depth increases. Across several inverse-MFG settings, numerical experiments show that JFB-r at moderate tracked depths can achieve recovery accuracy comparable to full unrolling while reducing memory and runtime.

Disclosure

“ve depth, generalized kink sensitivity, and extending the framework to higher-dimensional inverse problems arising in control and dynamics [35, 17], as well as inverse transport and flow reconstruction [59, 48, 24, 5]. Acknowledgments. ChatGPT (OpenAI) was used to polish the authors’ written text for spelling, grammar, and style. Claude (Anthropic) was used to assist with JAX-based accel- eration of portions of the numerical implementation, specifically translating and vectorizi”

PDF page 19
Classification
Proofreading, grammar, or spelling
Multiplier
1
Verified

Structural counts

Pages 36 pdf
Theorems 3 source
Lemmas 2 source
Propositions 5 source
Corollaries 2 source
Definitions 0 source
Displayed equations 117 source
Bibliography entries 69 source
Appendix pages 24 estimated

Count notes

  • Source counts use the expanded primary TeX file arxiv_main.tex.
  • Appendix pages include the first PDF page with an explicit Appendix heading through the final page.