Beats tuned baseline
2026
Jacobian-Free Short-Trace Backpropagation
Use a full primal-dual optimization solve in the forward pass, but backpropagate only through the last r iterations starting from a detached warm-start iterate. This avoids storing the full solver trajectory while preserving the forward solution, and provides a tunable bias-versus-memory tradeoff: r=0 is a cheap surrogate gradient, while increasing r should converge toward the implicit equilibrium gradient.
Paper: Truncated Differentiation Through Primal-Dual Solvers for Inverse Potential Mean-Field Games
arXiv:2608.00217