Dual Optimistic Ascent (PI Control) is the Augmented Lagrangian Method in Disguise
Juan Ramirez, Simon Lacoste-Julien
Abstract
Constrained optimization is a powerful framework for enforcing requirements on neural networks. These constrained deep learning problems are typically solved using first-order methods on their min-max Lagrangian formulation, but such approaches often suffer from oscillations and can fail to find all local solutions. While the Augmented Lagrangian method (ALM) addresses these issues, practitioners often favor dual optimistic ascent schemes (PI control) on the standard Lagrangian, which perform well empirically but lack formal guarantees. In this paper, we establish a previously unknown equivalence between these approaches: dual optimistic ascent on the Lagrangian is equivalent to gradient descent-ascent on the Augmented Lagrangian. This finding allows us to transfer the robust theoretical guarantees of the ALM to the dual optimistic setting, proving it converges linearly to all local solutions. Furthermore, the equivalence provides principled guidance for tuning the optimism hyper-parameter. Our work closes a critical gap between the empirical success of dual optimistic methods and their theoretical foundation in the single-step, first-order regime commonly used in constrained deep learning.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext feabb584-d370-4a55-ae97-f9bf75ab81a2Builds on14
- Safe RLHF: Safe Reinforcement Learning from Human FeedbackJosef Dai, Xuehai Pan, Ruiyang Sun, Jiaming Ji et al.ICLR 2024 · 656 citations
- Responsive Safety in Reinforcement Learning by PID Lagrangian MethodsAdam Stooke, Joshua Achiam, Pieter AbbeelICML 2020 · 403 citations
- Differentiable Causal Discovery from Interventional DataPhilippe Brouillard, Sébastien Lachapelle, Alexandre Lacoste, Simon Lacoste-Julien et al.NeurIPS 2020 · 295 citations
- Last-Iterate Convergence of Optimistic Gradient Method for Monotone Variational InequalitiesEduard Gorbunov, Adrien B. Taylor, Gauthier GidelNeurIPS 2022 · 65 citations
- Adversarial Robustness with Semi-Infinite Constrained LearningAlexander Robey, Luiz F. O. Chamon, George J. Pappas, Hamed Hassani et al.NeurIPS 2021 · 51 citations
Related papers
- On PI Controllers for Updating Lagrange Multipliers in Constrained OptimizationMotahareh Sohrabi, Juan Ramirez, Tianyue H. Zhang, Simon Lacoste-Julien et al.ICML 2024 · 7 citations
- Near-Optimal Solutions of Constrained Learning ProblemsJuan Elenter, Luiz F. O. Chamon, Alejandro RibeiroICLR 2024 · 10 citations
- Self-Supervised Primal-Dual Learning for Constrained OptimizationSeonho Park, Pascal Van HentenryckAAAI 2023 · 81 citations
- ReLOAD: Reinforcement Learning with Optimistic Ascent-Descent for Last-Iterate Convergence in Constrained MDPsTed Moskovitz, Brendan O'Donoghue, Vivek Veeriah, Sebastian Flennerhag et al.ICML 2023 · 24 citations
- Constrained Optimization via Exact Augmented Lagrangian and Randomized Iterative SketchingIlgee Hong, Sen Na, Michael W. Mahoney, Mladen KolarICML 2023 · 7 citations
