Skip to yearly menu bar Skip to main content


Poster

On PI Controllers for Updating Lagrange Multipliers in Constrained Optimization

Jose Gallego-Posada · Motahareh Sohrabi · Juan Ramirez · Tianyue Zhang · Simon Lacoste-Julien


Abstract: Constrained optimization offers a powerful framework to prescribe desired behaviors in neural network models. Typically, the constrained problems are solved via their min-max Lagrangian formulations, which exhibit unstable oscillatory dynamics when optimized using gradient descent-ascent. The adoption of constrained optimization techniques in the machine learning community is currently limited by the lack of reliable, general-purpose update schemes for the Lagrange multipliers. This paper provides an optimization perspective on Lagrange multiplier updates based on PI controllers, extending the work of Stooke, Achiam and Abbeel (2020). We provide theoretical and empirical insights explaining the inability of (negative) momentum methods to address the shortcomings of gradient descent-ascent, and contrast this with the success of our proposed $\nu$PI controller. Our experiments demonstrate that $\nu$PI reliably stabilizes the multiplier dynamics and its hyperparameters enjoy robust and predictable behavior.

Live content is unavailable. Log in and register to view live content