Home Science Learning to Optimize by Differentiable Programming
Science

Learning to Optimize by Differentiable Programming

Key Points

arXiv:2601.16510v3 Announce Type: replace Abstract: Solving massive-scale optimization problems requires scalable first-order methods with low per-iteration cost. This tutorial highlights a shift in optimization: using differentiable programming not only to execute algorithms but to learn how to design them. Modern frameworks such as PyTorch, TensorFlow, and JAX enable this paradigm through efficient automatic differentiation.

arXiv:2601.16510v3 Announce Type: replace Abstract: Solving massive-scale optimization problems requires scalable first-order methods with low per-iteration cost. This tutorial highlights a shift in optimization: using differentiable programming not only to execute algorithms but to learn how to design them. Modern frameworks such as PyTorch, TensorFlow, and JAX enable this paradigm through efficient automatic differentiation. Embedding first-order methods within these systems allows end-to-end training that improves convergence and solution quality. Guided by Fenchel-Rockafellar duality, the tutorial demonstrates how duality-informed iterative schemes such as ADMM and PDHG can be learned and adapted. Case studies across LP, NNV, Sum-Rate maximization, OPF, and LRMP illustrate these gains.
Differentiable Programming (ORG) PyTorch (ORG) TensorFlow (ORG) JAX (ORG) Fenchel-Rockafellar (ORG) LP (ORG) NNV (ORG) Sum-Rate (ORG) OPF (ORG) LRMP (ORG)
Originally published by arXiv CS Read original →