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

The Proximal Operator, Moreau Envelope, and Proximal-Gradient Method

shipped3 tiersLean: none

Anchor (Master): Beck 2017 First-Order Methods in Optimization (SIAM) Ch. 6 (the Moreau envelope, firm nonexpansiveness, the Moreau decomposition) and Ch. 10 (the sufficient-decrease lemma, the O(1/k) and linear rates, FISTA); Bauschke & Combettes 2017 Convex Analysis and Monotone Operator Theory in Hilbert Spaces (Springer, 2nd ed.) Ch. 12, 23-28 on resolvents of the subdifferential and forward-backward splitting

Intuition Beginner

A plain gradient step asks one question: which way is downhill, and how far should I walk? It works when the landscape is smooth. But many real cost functions are part smooth and part sharp — a smooth data-fitting bowl plus a sharp penalty that pushes answers toward zero. On the sharp part there is no single downhill direction, so a pure gradient step stalls. We need a move that respects both pieces at once.

The proximal operator is that move. Instead of taking a fixed step and hoping, you solve a small, friendly problem at every iteration: "find the point that makes the sharp penalty small while staying close to where I am now." The phrase "staying close" is a spring that pulls you back toward your current spot; the strength of the spring is the step size. A strong spring barely moves you; a weak spring lets the penalty drag you far. The answer to that small problem is the proximal step.

Two everyday pictures make it concrete. If the sharp penalty is "you must land inside this fenced field," then the friendly problem is "find the nearest point inside the fence" — that is just projection, walking to the closest legal spot. If the penalty is "keep your answer small, and prefer exact zeros," the friendly problem has a clean rule: shrink every number toward zero by a fixed amount, and snap anything small all the way to zero. That shrink-and-snap rule is called soft thresholding, and it is why these methods produce sparse, mostly-zero answers.

The full method alternates. Take an ordinary gradient step on the smooth part — a forward step. Then apply the proximal step for the sharp part — a backward step that cleans up. Repeat. Because the smooth part is genuinely smooth, this forward-backward rhythm makes steady progress, faster than treating the whole thing as one big sharp problem.

Visual Beginner

Picture one iteration as two moves. First a gradient step slides you downhill on the smooth bowl. Then the proximal step pulls you toward the point that balances the penalty against staying close, landing you on a cleaner spot — often with some coordinates snapped exactly to zero.

   smooth part f            sharp part g (penalty)        one prox-grad step
   (a bowl)                 (a sharp V at 0)              forward then backward

      \    /                     \      /                  x_k
       \  /                       \    /                    |  gradient step (forward)
        \/                         \  /                     v
   gradient points                 \/                  x_k - t*grad f(x_k)
   straight downhill          no single                     |  prox of g (backward)
                              downhill at 0                  v
                                                        x_{k+1}  (snapped toward 0)
sharp penalty g the proximal step does
indicator of a set (must stay inside) project onto the set: walk to the nearest legal point
the size penalty soft threshold: shrink each number toward 0 by , snap small ones to 0
nothing (g = 0) leave the point where it is: prox is the identity

Worked example Beginner

We soft-threshold a single number. Take the penalty with , and find the proximal step starting from . The small friendly problem is: pick to make as small as possible.

Step 1. Guess the answer is positive. If the penalty is , so we minimise . Its slope in is . Setting the slope to zero gives , so . Since , the guess holds.

Step 2. Read the rule. The starting value got pulled down to — it shrank by exactly . That is the shrink rule: a positive number loses .

Step 3. Try a small starting value. Now start from with the same . Shrinking by would overshoot past zero. The honest answer is to stop at zero: the proximal step returns . You can check that any positive makes larger than the value at , which is .

Step 4. State the full rule. Soft thresholding sends to if , to if , and to if . So , , and .

What this tells us. The proximal step for the size penalty is not magic: it is shrink-toward-zero-and-snap. Small numbers become exactly zero, which is how these methods select a few important variables and discard the rest.

Check your understanding Beginner

Formal definition Intermediate+

Throughout, are proper closed (lower semicontinuous) convex, with the Euclidean inner product and its norm. The subdifferential and the Fermat rule are as in 44.02.05.

Definition (proximal operator). For a proper closed convex and parameter , the proximal operator is the map

The objective is the sum of a convex function and a -strongly convex quadratic, hence -strongly convex; together with closedness and coercivity this makes the minimiser exist and be unique, so is a single-valued map (first prox theorem) [Beck, A. — First-Order Methods in Optimization (SIAM, 2017)]. Writing , the general case is .

Definition (Moreau envelope). The Moreau envelope (or Moreau-Yosida regularisation) of with parameter is the value of that minimisation:

It is a finite, convex, continuously differentiable function with , the same minimisers and minimum value as , and gradient

which is -Lipschitz. The envelope is thus a smooth surrogate for a possibly nonsmooth , and a gradient step on is a proximal step on .

Definition (resolvent / second prox theorem). The proximal operator is the resolvent of the maximal monotone operator :

Definition (firm nonexpansiveness). is firmly nonexpansive:

which implies -Lipschitz (nonexpansive) behaviour, .

Definition (composite problem and proximal-gradient method). Consider the composite objective

with convex and -smooth ( is -Lipschitz) and proper closed convex but possibly nonsmooth and "simple" (its prox is cheap). The proximal-gradient method (forward-backward splitting), with step , iterates

The forward (explicit) step uses the gradient of ; the backward (implicit) step uses the resolvent of . Two limiting cases anchor the picture: gives ordinary gradient descent, and gives the proximal point method .

Counterexamples to common slips Intermediate+

  • "The prox is a gradient step on ." It is a gradient step on the envelope , not on . For nonsmooth such as , has no gradient at , yet is perfectly defined (soft thresholding) and equals the resolvent .

  • "Any step size converges." The rate needs . With and , plain gradient descent diverges on a quadratic; the proximal-gradient method inherits this restriction through the descent lemma, which uses -smoothness of .

  • " depends only on , not ." The parameter sets the spring strength: soft-thresholds at level , and as the prox tends to the identity, while as it tends to the minimiser of .

Key theorem with proof Intermediate+

The composite theory rests on one estimate per step: the proximal-gradient map decreases a quadratic model of enough that the suboptimality telescopes to . The fixed-point equivalence first certifies that the limit is a minimiser.

Theorem (fixed points and the rate). Let with convex and -smooth and proper closed convex, and let minimise . Then:

(i) (Fixed-point characterisation.) For any , if and only if , i.e. minimises .

(ii) (Sufficient decrease.) For and any , writing , for every ,

(iii) ( rate.) With and ,

Proof. (i) By the second prox theorem, iff . Put , : the fixed-point equation holds iff , i.e. . Since is differentiable convex and convex, the sum rule 44.02.05 gives , so this is exactly Fermat's rule , equivalent to .

(ii) Two facts combine. First, the descent lemma for an -smooth convex : for all ,

Second, the prox step minimises a regularised model. By the second prox theorem applied to ,

Call this subgradient . The subgradient inequality for at gives, for every ,

Now bound . Using (1) with and so that ,

From (2), . Adding, and using convexity of at in the form :

The terms cancel: . What remains is

Apply the polarisation identity , which holds because . This is .

(iii) Take in at step (, ):

The right side is nonnegative, so (Fejér monotonicity). Summing telescopes to . Take in at step to see , so the sequence is nonincreasing; hence . Dividing by and using gives the claim.

Bridge. The single estimate is the foundational reason the method converges, and it builds toward the accelerated and splitting methods of the rest of the chapter, where the same prox inequality is reused under a momentum extrapolation. This is exactly the composite analogue of the descent lemma behind smooth gradient descent: where that argument decreases by a full , here the prox step folds the nonsmooth into the same quadratic comparison through its subgradient, so the proof generalises gradient descent to with no loss of rate. The bound is dual to the subgradient method's of 44.06.01: putting these together, the central insight is that splitting off the smooth part and treating only by its prox escapes the nonsmooth black-box lower bound, and the fixed-point equation appears again in FISTA 44.06.03 and ADMM 44.06.04 as the optimality condition those methods solve.

Exercises Intermediate+

Advanced results Master

The proximal-gradient method is , and the results below sharpen its rate under strong convexity, fix the prox calculus that makes the backward step computable, and place the iteration inside the operator-theoretic frame from which acceleration and splitting descend.

Theorem 1 (linear convergence under strong convexity). If in addition is -strongly convex, then is -strongly convex and the proximal-gradient map with is a contraction in the Euclidean norm: , where is the condition number. Consequently and , a linear (geometric) rate replacing the sublinear [Beck, A. — First-Order Methods in Optimization (SIAM, 2017)]. The contraction factor is exactly that of gradient descent on a smooth strongly convex function; the prox of is firmly nonexpansive and so does not degrade it.

Theorem 2 (Moreau envelope as a smoothing, and the proximal point method). The envelope is convex, finite, with -Lipschitz gradient , satisfies with identical minimiser set, and increases to pointwise as . Gradient descent on with step is precisely the proximal point algorithm , the case of forward-backward applied to the single function ; it converges for any because is firmly nonexpansive with fixed-point set [Parikh, N. & Boyd, S. — Proximal Algorithms (2014)]. The envelope is the bridge: it turns a nonsmooth problem into a smooth one whose gradient steps are implicit (backward) steps on the original.

Theorem 3 (prox calculus). The proximal operator obeys a calculus making composite models tractable. Separable sum: acts coordinatewise for (Exercise 4). Affine precomposition with a tight frame: if then . Scaling and translation: and reduces to by rescaling. Moreau decomposition (extended): for , generalising Exercise 7 [Beck, A. — First-Order Methods in Optimization (SIAM, 2017)]. The decomposition recovers the orthogonal-projection identity onto a closed convex cone and its polar as the special case , .

Theorem 4 (acceleration: the FISTA rate). Inserting a Nesterov momentum extrapolation before the prox step — with , then — yields the fast iterative shrinkage-thresholding algorithm with [Beck, A. — First-Order Methods in Optimization (SIAM, 2017)]. This matches the lower bound for first-order methods on the -smooth class, so acceleration is optimal there; the backward step and the sufficient-decrease lemma are reused unchanged, only the point at which the gradient is evaluated moves to the extrapolated . The detailed estimate-sequence proof is the subject of 44.06.03.

Theorem 5 (forward-backward splitting as a resolvent iteration). Writing the optimality condition as with cocoercive and maximal monotone, the proximal-gradient step is the forward-backward operator , where is the resolvent of [Combettes, P. L. & Wajs, V. R. — Signal recovery by proximal forward-backward splitting]. Convergence of the iterates to a zero of holds for where is the cocoercivity constant of (here ), via averaged-operator / Krasnoselskii-Mann fixed-point theory. This operator viewpoint is the one ADMM 44.06.04 and Douglas-Rachford splitting generalise to sums of two arbitrary nonsmooth prox-friendly terms.

Synthesis. The proximal operator is exactly the resolvent of the maximal monotone subdifferential, and this single identification is the foundational reason the whole apparatus coheres: the second prox theorem generalises the smooth gradient step to a nonsmooth implicit step, the Moreau envelope is dual to it — a smooth function whose gradient is the prox residual — and firm nonexpansiveness is exactly what makes both the proximal point and forward-backward iterations converge. The central insight is that splitting and treating by its prox rather than a subgradient escapes the black-box barrier of 44.06.01, buying from the descent lemma and, with momentum, the optimal ; this is exactly why the method dominates the subgradient baseline on composite problems. Putting these together, the Moreau decomposition is subgradient inversion from 44.02.05 read at the resolvent level, tying the prox calculus to the conjugacy of 44.02.03, and the bridge from this unit to acceleration 44.06.03 and to ADMM 44.06.04 is precisely the operator identity , which those methods extend from one prox to a coordinated pair.

Full proof set Master

Proposition 1 (first prox theorem: existence and uniqueness). For proper closed convex and , exists and is unique for every .

Proof. The objective is the sum of a proper closed convex and a -strongly convex closed quadratic, hence proper closed and -strongly convex. Strong convexity gives coercivity ( as ): for a fixed subgradient at any , convexity gives , so , which is coercive. A proper closed coercive function on attains its infimum (Weierstrass on a sublevel set, which is compact by closedness and coercivity), and a strongly convex function has at most one minimiser, since if both minimise, the midpoint satisfies , a contradiction.

Proposition 2 (second prox theorem: the resolvent characterisation). .

Proof. By Proposition 1, is the unique minimiser of . By Fermat's rule 44.02.05 and the sum rule (the quadratic term is differentiable with gradient , so its subdifferential is a singleton and the sum rule applies without a qualification), minimises iff , i.e. . Rearranged, , i.e. , so ; single-valuedness of the inverse is Proposition 1.

Proposition 3 (firm nonexpansiveness). For proper closed convex and , ; in particular is nonexpansive.

Proof. Write , . By Proposition 2, and . Monotonicity of (Proposition 4 of 44.02.05) gives , i.e. , which is , the firm-nonexpansiveness inequality. Cauchy-Schwarz on the right then gives , so .

Proposition 4 (Moreau envelope: smoothness and gradient). For proper closed convex and , is convex, finite, and continuously differentiable with , and this gradient is -Lipschitz.

Proof. is finite by Proposition 1 and convex as the infimal convolution of with the convex quadratic (an infimal convolution of convex functions is convex). Let . Fix ; for any , using and ,

The symmetric lower bound, obtained by swapping the roles of and and using , gives after combining; together these pin the gradient as . Lipschitz continuity: ; firm nonexpansiveness (Proposition 3) makes such that , so is also firmly nonexpansive, hence nonexpansive, giving .

Proposition 5 (Moreau decomposition). For proper closed convex with conjugate , for all .

Proof. Let and . Proposition 2 (with ) gives , i.e. . The subdifferential inversion for closed proper convex , (Exercise 7 / Connections of 44.02.05, a consequence of and the Fenchel-Young equality of 44.02.03), gives , i.e. . By Proposition 2 applied to , this says . Hence .

Proposition 6 (sufficient-decrease lemma). Let be convex and -smooth, proper closed convex, , and . Then for every , .

Proof. This is part (ii) of the Key theorem; the proof there is complete: combine the descent lemma (1) for -smooth with , the prox optimality from Proposition 2, the subgradient inequality (2) for at , and convexity of at ; the cross-terms cancel and the polarisation identity delivers the bound.

Proposition 7 ( rate). Under Proposition 6's hypotheses with and , .

Proof. Take in Proposition 6 with , : , so the distance is nonincreasing and the suboptimalities telescope: . Take , , in Proposition 6: , so the sequence is nonincreasing, whence . Divide by .

Connections Master

  • The proximal operator is the resolvent of the subdifferential built in 44.02.05: the second prox theorem is Fermat's rule applied to the regularised objective, firm nonexpansiveness is the monotonicity of proved there, and the Moreau decomposition is exactly the subgradient inversion read at the resolvent level. The entire prox calculus is the algorithmic face of that unit's subdifferential theory.

  • The Moreau decomposition and the conjugate-based prox calculus rest on the Fenchel conjugate and the Fenchel-Young equality of 44.02.03: is well-posed because is closed proper convex, and the cone special case is the indicator-conjugate pairing specialised, so the duality of that unit is what makes the backward step splittable across and .

  • The proximal-gradient method is the direct successor to the subgradient method of 44.06.01: by splitting and replacing the subgradient step on the nonsmooth part by an exact prox step, it leaves the black-box model in which is unimprovable and attains from the descent lemma. The lower bound of that unit is precisely the barrier this structural exploitation circumvents, and the fixed-point condition is the composite Fermat rule those two units share.

Historical & philosophical context Master

The proximal operator and the regularisation now called the Moreau envelope were introduced by Jean-Jacques Moreau in a sequence of notes culminating in his 1965 paper Proximité et dualité dans un espace hilbertien [Moreau 1965], where he defined , established the decomposition , and identified the prox as the resolvent of the subdifferential, then a maximal monotone operator in the sense being developed concurrently by Minty and Browder. The proximal point algorithm was analysed as a general method for monotone inclusions by R. Tyrrell Rockafellar in 1976 [Rockafellar 1976], building on Bernard Martinet's 1970 introduction of the proximal regularisation for variational problems. The forward-backward splitting that interleaves an explicit gradient step with an implicit prox step traces to the operator-splitting work of Lions and Mercier (1979) and Passty (1979) on sums of monotone operators, and was brought into signal recovery and sparse regression by Combettes and Wajs [Combettes & Wajs 2005]. The composite analysis and its acceleration (FISTA) were given by Beck and Teboulle in 2009 [Beck & Teboulle 2009], the source whose sufficient-decrease estimate this unit follows.

Bibliography Master

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

@article{parikhboyd2014proximal,
  author  = {Parikh, Neal and Boyd, Stephen},
  title   = {Proximal Algorithms},
  journal = {Foundations and Trends in Optimization},
  volume  = {1},
  number  = {3},
  pages   = {127--239},
  year    = {2014}
}

@article{combetteswajs2005,
  author  = {Combettes, Patrick L. and Wajs, Val\'erie R.},
  title   = {Signal recovery by proximal forward-backward splitting},
  journal = {Multiscale Modeling \& Simulation},
  volume  = {4},
  number  = {4},
  pages   = {1168--1200},
  year    = {2005}
}

@article{becktebeoulle2009fista,
  author  = {Beck, Amir and Teboulle, Marc},
  title   = {A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems},
  journal = {SIAM Journal on Imaging Sciences},
  volume  = {2},
  number  = {1},
  pages   = {183--202},
  year    = {2009}
}

@article{moreau1965proximite,
  author  = {Moreau, Jean-Jacques},
  title   = {Proximit\'e et dualit\'e dans un espace hilbertien},
  journal = {Bulletin de la Soci\'et\'e Math\'ematique de France},
  volume  = {93},
  pages   = {273--299},
  year    = {1965}
}

@article{rockafellar1976proximal,
  author  = {Rockafellar, R. Tyrrell},
  title   = {Monotone operators and the proximal point algorithm},
  journal = {SIAM Journal on Control and Optimization},
  volume  = {14},
  number  = {5},
  pages   = {877--898},
  year    = {1976}
}

@book{bauschkecombettes2017,
  author    = {Bauschke, Heinz H. and Combettes, Patrick L.},
  title     = {Convex Analysis and Monotone Operator Theory in Hilbert Spaces},
  edition   = {2},
  series    = {CMS Books in Mathematics},
  publisher = {Springer},
  year      = {2017}
}