Method of multipliersΒΆ

The method of multipliers is an algorithm for solving convex optimization problems. Suppose we have a problem of the form

\[\begin{split}\begin{array}{ll} \mbox{minimize} & f(x)\\ \mbox{subject to} & Ax = b, \end{array}\end{split}\]

where \(f\) is convex, \(x \in \mathbf{R}^n\) is the optimization variable, and \(A \in \mathbf{R}^{m \times n}\) and \(b \in \mathbf{R}^m\) are problem data.

To apply the method of multipliers, we first form the augmented Lagrangian

\[L_{\rho}(x,y) = f(x) + y^T(Ax - b) + (\rho/2)\|Ax-b\|^2_2.\]

The dual function associated with the augmented Lagrangian is \(g_{\rho}(y) = \inf_x L_{\rho}(x,y)\). The dual function \(g_{\rho}(y)\) is concave and its maximal value is the same as the optimal value of the original problem.

We maximize the dual function using gradient ascent. Each step of gradient ascent reduces to the \(x\) and \(y\) updates

\[\begin{split}\begin{array}{lll} x^{k+1} & := & \mathop{\rm argmin}_{x}\left(f(x) + (y^k)^T(Ax - b) + (\rho/2)\left\|Ax-b\right\|^2_2 \right) \\ y^{k+1} & := & y^{k} + \rho(Ax^{k+1}-b) \end{array}\end{split}\]

The following CVXPY script implements the method of multipliers and uses it to solve an optimization problem.

import cvxpy as cp
import numpy as np

# Initialize data.
rho = 1.0
n = 20
m = 10
A = np.random.randn(m,n)
b = np.random.randn(m)

# Initialize problem.
x = cp.Variable(shape=n)
f = cp.norm(x, 1)

# Solve with CVXPY.
cp.Problem(cp.Minimize(f), [A*x == b]).solve()
print("Optimal value from CVXPY: {}".format(f.value))

# Solve with method of multipliers.
resid = A*x - b
y = cp.Parameter(shape=(m)); y.value = np.zeros(m)
aug_lagr = f + y.T*resid + (rho/2)*cp.sum_squares(resid)
for t in range(MAX_ITERS):
    y.value += rho*resid.value

print("Optimal value from method of multipliers: {}".format(f.value))
Optimal value from CVXPY: 5.5905035557463005
Optimal value from method of multipliers: 5.572761551213633