44.06.09 · optimization-control / 06-first-order-large-scale

The Chambolle-Pock Primal-Dual Hybrid Gradient Method

shipped3 tiersLean: none

Anchor (Master): Chambolle & Pock 2011 'A first-order primal-dual algorithm for convex problems with applications to imaging' (J. Math. Imaging Vision 40:1, 120-145) full development: Theorem 1 (O(1/N) ergodic gap), Theorem 2 (O(1/N^2) under partial strong convexity), Theorem 3 (linear rate under bilateral strong convexity), and the §6 imaging applications; Condat 2013 'A primal-dual splitting method for convex optimization involving Lipschitzian, proximable and linear composite terms' (J. Optim. Theory Appl. 158:2, 460-479); He & Yuan 2012 'Convergence analysis of primal-dual algorithms for a saddle-point problem: from contraction perspective' (SIAM J. Imaging Sci. 5:1) the proximal-point / variational-inequality reading

Intuition Beginner

Many problems pull in two directions at once. You want a picture that stays close to the noisy photo you measured, and you want it to be smooth, with sharp edges but no speckle. The two wishes fight: matching the photo exactly keeps the noise; over-smoothing erases the edges. A good answer is a truce between them.

A clean way to find that truce is to picture a tug-of-war. One player, the "primal" player, proposes a picture. The other player, the "dual" player, is a critic who probes where the picture is too rough and pushes back on exactly those spots. They take turns: the critic sharpens its complaints given the current picture, then the picture-maker adjusts to answer the latest complaints. Round by round, the picture and the complaints settle into balance.

The Chambolle-Pock method runs this tug-of-war with two friendly moves you already know. Each round the critic does one proximal step (its update is a simple, often closed-form, adjustment), and the picture-maker does one proximal step too. The only thing coupling them is a linear operator — for images, the operator that measures how much neighboring pixels differ. There is one extra trick: the picture-maker looks slightly ahead, using not just the latest picture but a small over-shoot of it, which steadies the back-and-forth and makes it converge.

What makes this method shine on images is that neither move needs solving a big system of equations. Each step is just a cheap formula plus multiplying by the difference operator. There is a single rule to keep it stable: the two step sizes, multiplied together and scaled by the operator's strength, must stay below one.

Visual Beginner

Picture one round as three moves: the critic (dual ) sharpens its complaints from the current picture, the picture-maker (primal ) answers, and then a small over-shoot of the new picture is passed forward to steady the next round.

   round n                      what each step does

   dual y-update  ------->   prox-ascent: sharpen the critic given
   (the critic)              the look-ahead picture (a clip / prox step)
        |
        v
   primal x-update  ----->   prox-descent: adjust the picture to
   (the picture-maker)       answer the new critic (a prox step)
        |
        v
   extrapolation  ------->   look slightly ahead:
   (over-shoot)             bar x = x_new + (x_new - x_old)
        |
        v
   repeat until x and y settle   (the truce holds)

   stability rule:  step_x  x  step_y  x  (operator strength)^2  <  1
the piece its update does
critic (dual) prox-ascent: a cheap step (often a clip) given the look-ahead picture
picture (primal) prox-descent: a cheap step answering the latest critic
over-shoot pass forward plus its change, to steady the rounds

Worked example Beginner

We run one round on a tiny two-pixel denoising problem. The noisy data is . We want a picture that stays near but whose two pixels are not too different. The roughness is measured by the single difference , so . The fit cost is and the roughness cost is the absolute value with . We use step sizes , , and start at , , critic .

Step 1. The critic's update. The critic for an absolute value lives between and . Its update is: take , then clip into . Here , so , and clipping to gives .

Step 2. The picture's update. The picture answers by a fit-step. The fit prox for is . Here , so . Add and divide by : .

Step 3. The over-shoot. Pass forward .

What this tells us. The gap between the two pixels started at and after one round the picture's gap is — the roughness is already shrinking, while each pixel stays near its data value. Every step was a clip or a one-line formula; no system of equations was solved.

Check your understanding Beginner

Formal definition Intermediate+

Throughout, and are Euclidean spaces, is a bounded linear operator with adjoint and operator norm , and , are proper closed convex. The proximal operator and its resolvent identity are from 44.06.02; the Fenchel conjugate and Fenchel-Rockafellar duality are from 44.02.03.

Definition (the saddle-point model). Consider the composite primal problem

Conjugating via of 44.02.03 turns it into the saddle-point (convex-concave) problem

convex in for fixed and concave in for fixed . The variable is the dual variable dual to the constraint inside .

Definition (the Chambolle-Pock / PDHG iteration). Fix step sizes and an extrapolation parameter . Starting from and , the primal-dual hybrid gradient (PDHG) / Chambolle-Pock iteration produces by

The first step is a proximal ascent in (an explicit step followed by the prox of ); the second is a proximal descent in (an explicit step followed by the prox of ). The third is the over-relaxation / extrapolation step. The standard choice is .

Definition (step-size condition). The convergence of the unaccelerated method () holds under

This is the only restriction relating the two step sizes; either may otherwise be chosen freely, balancing the conditioning of the primal and dual blocks.

Definition (partial primal-dual gap). For bounded sets , the partial primal-dual gap is

nonnegative, vanishing iff is a saddle point inside ; it is the convergence criterion the rate theorem controls.

Counterexamples to common slips Intermediate+

  • "Chambolle-Pock is ADMM." Both solve the saddle model of 44.02.03, but ADMM 44.06.04 minimizes an augmented Lagrangian blockwise, and its -update couples with the quadratic — when this is a generalized prox requiring a linear solve. Chambolle-Pock uses only , , and matrix-vector products with : never a solve.

  • "The extrapolation is optional cosmetics." Dropping it () gives the plain Arrow-Hurwicz primal-dual method, which is not guaranteed to converge for general convex under ; it needs strong convexity or smaller steps. The over-relaxation is what delivers the guarantee for the general convex case.

  • "Smaller steps always converge, like a gradient method." The binding condition is the product , not each step individually. Shrinking only while is large can still violate it; both must be controlled jointly through their product against .

Key theorem with proof Intermediate+

The convergence of Chambolle-Pock is established not by descent on either variable alone but by a single estimate on the joint primal-dual iterate, telescoped and averaged: the ergodic gap decays as . The estimate uses only the firm nonexpansiveness of the two proximal maps and the bilinear coupling through .

Theorem (ergodic convergence of the primal-dual gap). Let be proper closed convex, linear, and let be a saddle point of . Run Chambolle-Pock with and . Then the ergodic averages , satisfy, for every bounded containing ,

Proof. The prox-ascent step has, by the resolvent characterization of 44.06.02, the optimality , i.e. . The subgradient inequality for the convex then gives, for every ,

Symmetrically, gives , hence for every ,

Add the two inequalities and regroup the bilinear terms into . Using the polarization identity on the - and -weighted inner products produces, after collecting,

where the residual coupling term collects the extrapolation contribution (using ) together with the squared-distance slack. The condition is exactly what makes the accumulated terms, after a Cauchy-Schwarz / Young split , telescope into a nonnegative quantity that can be dropped. Summing the displayed inequality over telescopes the squared distances:

By convexity of and concavity of (Jensen), . Taking the supremum over on both sides gives the stated bound on the partial gap.

Bridge. The single telescoping estimate on the joint iterate is the foundational reason Chambolle-Pock converges, and the role of — absorbing the bilinear coupling so the squared distances telescope cleanly — builds toward the accelerated and linear rates of the Advanced results, where strong convexity supplies an extra negative term that sharpens the same estimate. This is exactly the saddle-point reading of 44.02.03 driven by the two proximal primitives of 44.06.02, and it is dual to the operator-splitting analysis of 44.06.04: where ADMM is Douglas-Rachford on the Fenchel dual, Chambolle-Pock is the proximal point algorithm on the primal-dual monotone operator in a preconditioned metric, so the method generalises forward-backward splitting to a coupled primal-dual pair without ever inverting . Putting these together, the central insight is that the gap functional, not either variable's value, is the right convergence certificate, and the bridge is the resolvent of 44.06.02 — the same primitive that ADMM, forward-backward, and this method all share, here paired with across the operator .

Exercises Intermediate+

Advanced results Master

The Chambolle-Pock iteration , , admits a self-contained development: its placement as a preconditioned proximal point method, the base rate, the two accelerated regimes, the no-linear-solve property that separates it from ADMM, and the Condat-Vu extension to a smooth term.

Theorem 1 (preconditioned proximal point reading). For , Chambolle-Pock is the proximal point algorithm of Rockafellar applied to the maximal monotone primal-dual operator in the metric of the symmetric positive-definite preconditioner M = \begin{psmallmatrix} I/\tau & -K^\top \\ -K & I/\sigma\end{psmallmatrix}, which is positive definite exactly when [He & Yuan 2012]. The resulting -resolvent is firmly nonexpansive in the -norm, so the iterates form a Fejér-monotone sequence converging to a zero of — a saddle point of . This is the structural reason convergence holds for any fixed steps satisfying the product condition, paralleling the averaged-operator reading of ADMM in 44.06.04, but with the operator entering through the metric rather than through an augmented-Lagrangian quadratic.

Theorem 2 (base ergodic rate). For proper closed convex with a saddle point, , and , the ergodic averages satisfy over , the partial-primal-dual-gap bound [Chambolle & Pock 2011]. The rate is on the ergodic average; the last iterate also converges (by Fejér monotonicity from Theorem 1) but without a uniform worst-case rate in the general nonsmooth case.

Theorem 3 (acceleration under partial strong convexity, ). If is -strongly convex (equivalently , hence , is smooth on the dual side), the adaptive schedule

keeps invariant and yields , the optimal rate for this smoothness class [Chambolle & Pock 2011]. The mechanism is that strong convexity contributes a term that lets grow linearly, so and the primal distance, scaled by , decays quadratically — the saddle-point analogue of Nesterov acceleration for the smooth class.

Theorem 4 (linear rate under bilateral strong convexity). If both and are strongly convex (with moduli — equivalently strongly convex and smooth, and strongly convex and smooth, so both primal and dual problems are smooth strongly convex), fixed accelerated steps give linear convergence for some determined by [Chambolle & Pock 2011]. The contraction factor improves as the problem's two-sided conditioning improves, and the optimal fixed steps balance the primal and dual moduli against — the primal-dual counterpart of the linear rate of gradient descent on smooth strongly convex objectives.

Theorem 5 (no linear solve; Condat-Vu extension). Each Chambolle-Pock step evaluates , , and the actions of — never the inverse of , , or any quadratically-coupled prox; this is the decisive contrast with ADMM 44.06.04, whose -update is a generalized prox in the -metric requiring a linear solve (or a linearizing inner approximation) when is not diagonal, exactly the case for the difference operator of imaging [Chambolle & Pock 2011]. The Condat-Vu generalization handles a third, smooth term: for with -Lipschitz, the iteration adds a forward gradient step , , convergent under and recovering Chambolle-Pock when [Condat 2013].

Synthesis. Chambolle-Pock is the proximal point algorithm on the primal-dual monotone operator in the preconditioned -metric, and this single identification is the foundational reason the method coheres: the alternating prox-ascent and prox-descent are the two block rows of one -resolvent, the extrapolation is the off-diagonal coupling in , and the step condition is exactly the positive-definiteness of that makes the resolvent firmly nonexpansive. This is exactly the saddle-point form of 44.02.03 solved by the two proximal primitives of 44.06.02, and it is dual to the Douglas-Rachford reading of ADMM in 44.06.04: where ADMM splits the Fenchel dual through reflected resolvents and pays a linear solve when is a genuine coupling operator, Chambolle-Pock keeps in the metric and never inverts it, which is the central insight that makes it the method of choice for total-variation and imaging problems where is a difference operator with no usable factorization.

The base estimate generalises to the accelerated regimes by a uniform mechanism — strong convexity adds a damping term, -strong-convexity alone buys through the adaptive schedule, and bilateral strong convexity buys a linear rate — so one gap-descent argument, sharpened by curvature, covers , , and linear convergence at once. Putting these together, the Condat-Vu extension folds a smooth term in by a forward gradient step, placing Chambolle-Pock inside the same three-operator splitting family as forward-backward 44.06.02 and ADMM 44.06.04, all of which are resolvent iterations on read through different reformulations of one saddle model.

Full proof set Master

Proposition 1 (prox optimality as a subgradient inclusion). The Chambolle-Pock dual and primal updates satisfy and .

Proof. The update with is, by definition, the minimizer of . Fermat's rule 44.06.02 gives , i.e. . Substituting gives . The primal update is handled identically: , i.e. .

Proposition 2 (one-step primal-dual estimate). For every ,

Proof. The convex subgradient inequality for at with the inclusion of Proposition 1 gives ; for at , . Recall . Insert the two inequalities (as upper bounds on and ), giving

where collects the bilinear terms that combine to . Apply the polarization identity to each inner product and likewise for the -term. Collecting yields the stated inequality.

Proposition 3 (the extrapolation makes the coupling telescope). With , , so , and the coupling term telescopes: summing over leaves only boundary terms plus a quantity controlled by .

Proof. With , write , so . Then . Abel summation regroups into plus boundary terms . Each interior term is bounded by Young's inequality, ; choosing makes these absorbable into the slack of Proposition 2 precisely when , i.e. . The leftover is nonnegative and is dropped.

Proposition 4 (ergodic gap bound). Under , , summing Propositions 2-3 over and averaging gives for all .

Proof. Sum the inequality of Proposition 2 over . The squared-distance differences telescope to ; drop the nonpositive . The coupling sum is handled by Proposition 3, contributing only the absorbed nonnegative quantity (dropped) and boundary terms that vanish since and the final term is again absorbed. Hence . By convexity of and concavity of , Jensen gives , and division by yields the claim.

Proposition 5 (the partial gap is the right certificate). Taking the supremum of Proposition 4 over with gives , and iff is a saddle point of on .

Proof. The left side of Proposition 4 is ; its supremum over is exactly by the definition of the partial gap. The right side's supremum is , finite since is bounded, so . The gap is a difference of a max and a min of the same function, hence ; it is iff for all admissible , the saddle-point inequalities.

Proposition 6 (-positive-definiteness equals the step condition). The block matrix M = \begin{psmallmatrix} I/\tau & -K^\top \\ -K & I/\sigma\end{psmallmatrix} is positive definite iff .

Proof. iff the leading block (always, ) and the Schur complement . The latter holds iff , i.e. since . Positive-definiteness of makes an inner product, in which the proximal point step of Theorem 1 (Advanced results) is firmly nonexpansive, giving Fejér monotonicity of the iterates toward .

Connections Master

  • Chambolle-Pock solves the Fenchel saddle model of 44.02.03: the primal becomes by conjugating , and the dual variable ranges over . The conjugate calculus of that unit — that of a norm is a dual-ball indicator, that the prox of an indicator is a projection — is exactly what makes the dual prox-ascent step a closed-form clip in the imaging applications.

  • The two updates are the proximal operators of 44.06.02: and , and the firm nonexpansiveness of the resolvent proved there is what makes the preconditioned proximal point operator of this unit contractive in the -metric. Forward-backward of that unit is the single-operator special case; Chambolle-Pock is its primal-dual extension across the coupling operator .

  • Chambolle-Pock is the no-linear-solve companion of ADMM and Douglas-Rachford 44.06.04: both attack the saddle model of 44.02.03 with proximal primitives, but ADMM is Douglas-Rachford on the Fenchel dual whose subproblems can carry a -coupled quadratic (a linear solve when ), while Chambolle-Pock keeps in the metric and uses only actions — the distinction that decides which method suits total-variation imaging.

  • Chambolle-Pock is the saddle-point solver that consumes the structured min-max model whose smoothing was studied in 44.06.08: Nesterov smoothing of regularizes the inner and runs an accelerated method on the smoothed primal, while Chambolle-Pock attacks the same saddle function directly without smoothing, trading the smoothed-primal rate for an gap rate that needs no smoothing parameter, and matching the accelerated rate when a block is strongly convex.

Historical & philosophical context Master

The primal-dual hybrid gradient method has roots in the Arrow-Hurwicz scheme of 1958 for finding saddle points of a Lagrangian by alternating gradient ascent in the dual and gradient descent in the primal, developed in the study of constrained optimization and economic equilibrium [Arrow, Hurwicz & Uzawa 1958]. The modern proximal form for imaging was introduced by Antonin Chambolle and Thomas Pock in 2011 in A first-order primal-dual algorithm for convex problems with applications to imaging (Journal of Mathematical Imaging and Vision 40(1), 120-145), which gave the over-relaxed iteration, the step condition , the ergodic-gap theorem, and the accelerated and linear variants under strong convexity, together with the total-variation denoising and deconvolution applications that made the method standard in computational imaging [Chambolle & Pock 2011]. The closely related primal-dual splitting that adds a smooth Lipschitz-gradient term was given independently by Laurent Condat in 2013 and by Bằng Công Vũ in the same period, yielding the three-operator scheme now called the Condat-Vu method [Condat 2013]. The proximal-point and variational-inequality interpretation, which placed the method inside maximal-monotone-operator theory and explained its convergence as Fejér monotonicity in a preconditioned metric, was developed by Bingsheng He and Xiaoming Yuan in 2012 [He & Yuan 2012]. The earlier PDHG iteration without the convergence-securing extrapolation was studied by Mingqiang Zhu and Tony Chan in 2008.

Bibliography Master

@article{chambollepock2011,
  author  = {Chambolle, Antonin and Pock, Thomas},
  title   = {A first-order primal-dual algorithm for convex problems with applications to imaging},
  journal = {Journal of Mathematical Imaging and Vision},
  volume  = {40},
  number  = {1},
  pages   = {120--145},
  year    = {2011}
}

@book{beck2017fom,
  author    = {Beck, Amir},
  title     = {First-Order Methods in Optimization},
  series    = {MOS-SIAM Series on Optimization},
  number    = {25},
  publisher = {SIAM},
  year      = {2017}
}

@article{condat2013,
  author  = {Condat, Laurent},
  title   = {A primal-dual splitting method for convex optimization involving {Lipschitzian}, proximable and linear composite terms},
  journal = {Journal of Optimization Theory and Applications},
  volume  = {158},
  number  = {2},
  pages   = {460--479},
  year    = {2013}
}

@article{vu2013,
  author  = {V\~{u}, B\`{a}ng C\^{o}ng},
  title   = {A splitting algorithm for dual monotone inclusions involving cocoercive operators},
  journal = {Advances in Computational Mathematics},
  volume  = {38},
  number  = {3},
  pages   = {667--681},
  year    = {2013}
}

@article{heyuan2012,
  author  = {He, Bingsheng and Yuan, Xiaoming},
  title   = {Convergence analysis of primal-dual algorithms for a saddle-point problem: from contraction perspective},
  journal = {SIAM Journal on Imaging Sciences},
  volume  = {5},
  number  = {1},
  pages   = {119--149},
  year    = {2012}
}

@book{arrowhurwiczuzawa1958,
  author    = {Arrow, Kenneth J. and Hurwicz, Leonid and Uzawa, Hirofumi},
  title     = {Studies in Linear and Non-Linear Programming},
  publisher = {Stanford University Press},
  year      = {1958}
}

@article{zhuchan2008,
  author  = {Zhu, Mingqiang and Chan, Tony F.},
  title   = {An efficient primal-dual hybrid gradient algorithm for total variation image restoration},
  journal = {UCLA CAM Report 08-34},
  year    = {2008}
}