110
. . . . . . Convex Optimization: Old Tricks for New Problems Ryota Tomioka The University of Tokyo 2012-08-15 @ DTU PhD Summer Course Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 1 / 73

Convex Optimization: Old Tricks for New Problems

  • Upload
    ryotat

  • View
    1.287

  • Download
    5

Embed Size (px)

DESCRIPTION

 

Citation preview

Page 1: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convex Optimization: Old Tricks for New Problems

Ryota Tomioka

The University of Tokyo

2012-08-15 @ DTU PhD Summer Course

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 1 / 73

Page 2: Convex Optimization: Old Tricks for New Problems

. . . . . .

Introduction

Why care about convex optimization (and sparsity)?

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 2 / 73

Page 3: Convex Optimization: Old Tricks for New Problems

. . . . . .

Why do we care about optimization — sparseestimation

High dimensional problems (dimension ≫ #samples)

Bioinformatics (microarray,SNP analysis,etc)

Text-mining (POS tagging,) Magnetic resonance imaging —

compressed sensingStructure inference

Collaborative filtering — low-rank structure Graphical model inference— sparse graph

structure

1

4

2

32

211

11

1

32

4

23

4

1

2

1 1Movies

Use

rs

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 3 / 73

Page 4: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ex. 1: SNP (single nucleotide polymorphism) analysisx i : input (SNP), yi = 1: has the illness,yi = −1: healthy

Goal: Infer the association from genetic variability x i to the illness yi .

Logistic regression

minimizew∈Rn

m∑i=1

log(1 + exp(−yi ⟨x i ,w⟩))︸ ︷︷ ︸data-fit

+ λ∥w∥1︸ ︷︷ ︸Regularization

E.g., # SNPs n = 500,000, #subjects m = 5,000MAP etimation with the logisticloss f.

log(1 + e−yz) = − log P(Y = y |z)

where P(Y = +1|z) = ez

1+ez .

f(x)=log(1+exp(−x))

y<x,w>

−5 0 50

0.5

1

z

σ (z

)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 4 / 73

Page 5: Convex Optimization: Old Tricks for New Problems

. . . . . .

L1-regularization and sparsity

Best convex approximation of ∥w∥0.

Threshold occurs for finite λ.Non-convex cases (p < 1) can be solved byre-weighted L1 minimization

−3 −2 −1 0 1 2 30

1

2

3

4

|x|0.01

|x|0.5

|x|x2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 5 / 73

Page 6: Convex Optimization: Old Tricks for New Problems

. . . . . .

L1-regularization and sparsity

Best convex approximation of ∥w∥0.

−3 −2 −1 0 1 2 30

1

2

3

4

|x|0.01

|x|0.5

|x|x2

Threshold occurs for finite λ.Non-convex cases (p < 1) can be solved byre-weighted L1 minimization

−3 −2 −1 0 1 2 30

1

2

3

4

|x|0.01

|x|0.5

|x|x2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 5 / 73

Page 7: Convex Optimization: Old Tricks for New Problems

. . . . . .

L1-regularization and sparsity

Best convex approximation of ∥w∥0.Threshold occurs for finite λ.

Non-convex cases (p < 1) can be solved byre-weighted L1 minimization

−3 −2 −1 0 1 2 30

1

2

3

4

|x|0.01

|x|0.5

|x|x2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 5 / 73

Page 8: Convex Optimization: Old Tricks for New Problems

. . . . . .

L1-regularization and sparsity

Best convex approximation of ∥w∥0.Threshold occurs for finite λ.Non-convex cases (p < 1) can be solved byre-weighted L1 minimization

−2 −1.5 −1 −0.5 0 0.5 1 1.5 20

0.5

1

1.5

2

−3 −2 −1 0 1 2 30

1

2

3

4

|x|0.01

|x|0.5

|x|x2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 5 / 73

Page 9: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ex. 2: Compressed sensing [Candes, Romberg, & Tao 06]

Signal (MRI image) recovery from (noisy) low-dimensionalmeasurements.

minimizew∈Rn

12∥y −Ωw∥2

2 + λ∥Φw∥1

y : Noisy signalw : Original signalΩ: Rn → Rm: Observation matrix (random, fourier transform)Φ: Trnasformation s.t. the original signal is sparse

NB: If Φ−1 exists, we can solve instead

minimizew∈Rn

12∥y − Aw∥2

2 + λ∥w∥1,

where A = ΩΦ−1.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 6 / 73

Page 10: Convex Optimization: Old Tricks for New Problems

. . . . . .

50 100 150 200 250

50

100

150

200

250

50 100 150 200 250

50

100

150

200

250

−3

−2

−1

0

1

2

3

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 7 / 73

Page 11: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ex. 3: Estimation of a low-rank matrix [Fazel+ 01; Srebro+ 05]

Goal: Recover a low-rank matrix X from partial (noisy) measurement Y

minimizeX

12∥Ω(X − Y )∥2 + λ∥X∥S1

where ∥X∥S1 :=r∑

j=1

σj(X ) (Schatten 1-norm)

Aka trace norm, nuclear norm⇒ Linear sum of singular values⇒ Sparsity in the SV spectrum⇒ Low-rank

1

4

2

32

211

11

1

32

4

23

4

1

2

1 1Movies

Use

rs

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 8 / 73

Page 12: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ex. 4: Low-rank tensor completion [Tomioka+11]

SensorsTim

e

Featu

res

Factors

(loadings)

Core

(interactions)

n1

n2

n3

n1

n2

n3

r1

r2

r3r1

r2r3

Tucker decompositionXijk =

r1∑

a=1

r2∑

b=1

r3∑

c=1

CabcU(1)ia U

(2)jb U

(c)kc

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 9 / 73

Page 13: Convex Optimization: Old Tricks for New Problems

. . . . . .

Simple vs. structured sparse estimation problems

Simple sparse estimation problem

minimizew

L(w) + λ∥w∥1

SNP analysis Compressed sensing with Φ−1 (e.g., wavelet) Collaborative filtering (matrix completion)

Structured sparse estimation problem

minimizew

L(w) + λ∥Φw∥1

Compressed sensing without Φ−1 (e.g., total variation) Low-rank tensor completion

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 10 / 73

Page 14: Convex Optimization: Old Tricks for New Problems

. . . . . .

Common criticisms

Convex optimization is another developed field (and it is boring).We can just use it as a black box.

Yes, but we can do much better by knowing the structure of ourproblems.

Convexity is too restrictive. Convexity depends on parametrization. A seemingly non-convex

problem could be reformulated into a convex problem.I am only interested in making things work.

Yes, convex optimization works. But it can also be used foranalyzing how algorithms perform at the end.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 11 / 73

Page 15: Convex Optimization: Old Tricks for New Problems

. . . . . .

Common criticisms

Convex optimization is another developed field (and it is boring).We can just use it as a black box.

Yes, but we can do much better by knowing the structure of ourproblems.

Convexity is too restrictive. Convexity depends on parametrization. A seemingly non-convex

problem could be reformulated into a convex problem.

I am only interested in making things work. Yes, convex optimization works. But it can also be used for

analyzing how algorithms perform at the end.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 11 / 73

Page 16: Convex Optimization: Old Tricks for New Problems

. . . . . .

Common criticisms

Convex optimization is another developed field (and it is boring).We can just use it as a black box.

Yes, but we can do much better by knowing the structure of ourproblems.

Convexity is too restrictive. Convexity depends on parametrization. A seemingly non-convex

problem could be reformulated into a convex problem.I am only interested in making things work.

Yes, convex optimization works. But it can also be used foranalyzing how algorithms perform at the end.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 11 / 73

Page 17: Convex Optimization: Old Tricks for New Problems

. . . . . .

Bayesian inference as a convex optimization

minimizeq

Eq[f (w)]︸ ︷︷ ︸average energy

+Eq[log q(w)]︸ ︷︷ ︸entropy

s.t. q(w) ≥ 0,∫

q(w)dw = 1

wheref (w) = − log P(D|w)︸ ︷︷ ︸

neg. log likelihood

− log P(w)︸ ︷︷ ︸neg. log prior

⇒ q(w) =1Z

e−f (w) (Bayesian posterior)

Inner approximations

Variational BayesEmpirical Bayes

Outer approximations

Belief propagationSee Wainwright & Jordan08.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 12 / 73

Page 18: Convex Optimization: Old Tricks for New Problems

. . . . . .

Bayesian inference as a convex optimization

minimizeq

Eq[f (w)]︸ ︷︷ ︸average energy

+Eq[log q(w)]︸ ︷︷ ︸entropy

s.t. q(w) ≥ 0,∫

q(w)dw = 1

wheref (w) = − log P(D|w)︸ ︷︷ ︸

neg. log likelihood

− log P(w)︸ ︷︷ ︸neg. log prior

⇒ q(w) =1Z

e−f (w) (Bayesian posterior)

Inner approximations

Variational BayesEmpirical Bayes

Outer approximations

Belief propagationSee Wainwright & Jordan08.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 12 / 73

Page 19: Convex Optimization: Old Tricks for New Problems

. . . . . .

Bayesian inference as a convex optimization

minimizeq

Eq[f (w)]︸ ︷︷ ︸average energy

+Eq[log q(w)]︸ ︷︷ ︸entropy

s.t. q(w) ≥ 0,∫

q(w)dw = 1

wheref (w) = − log P(D|w)︸ ︷︷ ︸

neg. log likelihood

− log P(w)︸ ︷︷ ︸neg. log prior

⇒ q(w) =1Z

e−f (w) (Bayesian posterior)

Inner approximations

Variational BayesEmpirical Bayes

Outer approximations

Belief propagationSee Wainwright & Jordan08.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 12 / 73

Page 20: Convex Optimization: Old Tricks for New Problems

. . . . . .

Bayesian inference as a convex optimization

minimizeq

Eq[f (w)]︸ ︷︷ ︸average energy

+Eq[log q(w)]︸ ︷︷ ︸entropy

s.t. q(w) ≥ 0,∫

q(w)dw = 1

wheref (w) = − log P(D|w)︸ ︷︷ ︸

neg. log likelihood

− log P(w)︸ ︷︷ ︸neg. log prior

⇒ q(w) =1Z

e−f (w) (Bayesian posterior)

Inner approximations

Variational BayesEmpirical Bayes

Outer approximations

Belief propagationSee Wainwright & Jordan08.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 12 / 73

Page 21: Convex Optimization: Old Tricks for New Problems

. . . . . .

Overview

Convex optimization basics Convex sets Convex function Conditions that guarantee convexity Convex optimization problem

Looking into more structures Proximity operators Conjugate duality and dual ascent Augmented Lagrangian and ADMM

References:Boyd &Vandenberghe. (2004)Convex optimization.

Bertsekas (1999)NonlinearProgramming.

Rockafellar (1970)Convex Analysis.

Moreau (1965) Proximité etdualité dans un espaceHilbertien.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 13 / 73

Page 22: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convexity

Learning objectivesConvex setsConvex functionConditions that guarantee convexityConvex optimization problem

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 14 / 73

Page 23: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convex setA subset V ⊆ Rn is a convex set

⇔ line segment between two arbitrary points x ,y ∈ V is included in V ;that is,

∀x ,y ∈ V , ∀λ ∈ [0,1], λx + (1 − λ)y ∈ V .

x

y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 15 / 73

Page 24: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convex functionA function f : Rn → R ∪ +∞ is a convex function

⇔ the function f is below any line segment between two points on f ;that is,

∀x ,y ∈ Rn, ∀λ ∈ [0,1], f ((1 − λ)x + λy) ≤ (1 − λ)f (x) + λf (y)

(Jensen’s inequality)

x

f(x)

y

f(y)

f(x)

f(y)

Non−convexConvex

Johan Jensen

1859 – 1925

NB: when the strict inequality < holds, f is called strictly convex.Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 16 / 73

Page 25: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convex function

A function f : Rn → R ∪ +∞ is a convex function

⇔ the epigraph of f is a convex set; that is

Vf := (t ,x) : (t ,x) ∈ Rn+1, t ≥ f (x) is convex.

(x,f(x))

(y,f(y))

Epigraph

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 17 / 73

Page 26: Convex Optimization: Old Tricks for New Problems

. . . . . .

Jointly convexA function f (x , y) can be convex wrt x (y ) for any fixed y (x),respectively, zbut can fail to be convex for x and y simultaneously.

−1 −0.5 0 0.5 1 −1−0.5

00.5

1

−1

−0.5

0

0.5

1

y

x

x*y

f (x , y) is convex ⇒ () f (x , y) is convex for x and y individually

To be more explicit, we sometimes say jointly convex.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 18 / 73

Page 27: Convex Optimization: Old Tricks for New Problems

. . . . . .

Jointly convexA function f (x , y) can be convex wrt x (y ) for any fixed y (x),respectively, zbut can fail to be convex for x and y simultaneously.

−1 −0.5 0 0.5 1 −1−0.5

00.5

1

−1

−0.5

0

0.5

1

y

x

x*y

f (x , y) is convex ⇒ () f (x , y) is convex for x and y individually

To be more explicit, we sometimes say jointly convex.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 18 / 73

Page 28: Convex Optimization: Old Tricks for New Problems

. . . . . .

Why do we allow infinity?

f (x) = 1/x is convex for x > 0.

f (x) =

1/x if x > 0,+∞ otherwise.

and we can forget about the domain.

The indicator function δC(x) of a set C:

δC(x) =

0 if x ∈ C,

+∞ otherwise.

Is this a convex function? (consider the epigraph)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 19 / 73

Page 29: Convex Optimization: Old Tricks for New Problems

. . . . . .

Why do we allow infinity?

f (x) = 1/x is convex for x > 0.

f (x) =

1/x if x > 0,+∞ otherwise.

and we can forget about the domain.The indicator function δC(x) of a set C:

δC(x) =

0 if x ∈ C,

+∞ otherwise.

Is this a convex function? (consider the epigraph)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 19 / 73

Page 30: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #1: Hessian........Hessian ∇2f (x) is positive semidefinite (if f is differentiable)

Examples(Negative) entropy is a convex function.

f (p) =n∑

i=1

pi log pi ,

∇2f (p) = diag(1/p1, . . . , 1/pn) ⪰ 0.

0 1

log determinant is a concave (−f is convex) function

f (X ) = log |X | (X ⪰ 0),

∇2f (X ) = −X−⊤ ⊗ X−1 ⪯ 0X

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 20 / 73

Page 31: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #1: Hessian........Hessian ∇2f (x) is positive semidefinite (if f is differentiable)

Examples(Negative) entropy is a convex function.

f (p) =n∑

i=1

pi log pi ,

∇2f (p) = diag(1/p1, . . . , 1/pn) ⪰ 0.

0 1

log determinant is a concave (−f is convex) function

f (X ) = log |X | (X ⪰ 0),

∇2f (X ) = −X−⊤ ⊗ X−1 ⪯ 0X

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 20 / 73

Page 32: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #1: Hessian........Hessian ∇2f (x) is positive semidefinite (if f is differentiable)

Examples(Negative) entropy is a convex function.

f (p) =n∑

i=1

pi log pi ,

∇2f (p) = diag(1/p1, . . . , 1/pn) ⪰ 0.

0 1

log determinant is a concave (−f is convex) function

f (X ) = log |X | (X ⪰ 0),

∇2f (X ) = −X−⊤ ⊗ X−1 ⪯ 0X

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 20 / 73

Page 33: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #2: Maximum over convex functions..

......

Maximum over convex functions fj(x)∞j=1

f (x) := maxj

fj(x) (fj(x) is convex for all j)

is convex.

The same as saying “intersection of convex sets is a convex set”Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 21 / 73

Page 34: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #2: Maximum over convex functions..

......

Maximum over convex functions f (x ;α) : α ∈ Rm

f (x) := supα∈Rm

f (x ;α)

is convex.

ExampleQuadratic over linear is a convex function

f (x , y) = supα∈R

(−α2

2x + αy

)(x > 0)

=y2

2x

Similarly

f (Σ,y) =12

y⊤Σ−1y (Σ ≻ 0) is a convex function (show it!)

0

0.05

0.1 −1−0.5

00.5

1

0102030

y

Σ

yΣ−

1 y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 22 / 73

Page 35: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #2: Maximum over convex functions..

......

Maximum over convex functions f (x ;α) : α ∈ Rm

f (x) := supα∈Rm

f (x ;α)

is convex.

ExampleQuadratic over linear is a convex function

f (x , y) = supα∈R

(−α2

2x + αy

)(x > 0)

=y2

2x

Similarly

f (Σ,y) =12

y⊤Σ−1y (Σ ≻ 0) is a convex function (show it!)

0

0.05

0.1 −1−0.5

00.5

1

0102030

y

Σ

yΣ−

1 yRyota Tomioka (Univ Tokyo) Optimization 2012-08-15 22 / 73

Page 36: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #2: Maximum over convex functions..

......

Maximum over convex functions f (x ;α) : α ∈ Rm

f (x) := supα∈Rm

f (x ;α)

is convex.

ExampleQuadratic over linear is a convex function

f (x , y) = supα∈R

(−α2

2x + αy

)(x > 0)

=y2

2x

Similarly

f (Σ,y) =12

y⊤Σ−1y (Σ ≻ 0) is a convex function (show it!)

0

0.05

0.1 −1−0.5

00.5

1

0102030

y

Σ

yΣ−

1 yRyota Tomioka (Univ Tokyo) Optimization 2012-08-15 22 / 73

Page 37: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #3: Partial minimum..

......

Partial minimum of a convex function f (x , y)

f (x) := miny∈Rn

f (x , y) is convex.

ExamplesHierarchical prior minimization

f (x) = mind1,...,dn≥0

12

n∑j=1

(x2

j

dj+

dpj

p

)(p ≥ 1)

=1q

n∑j=1

|xj |q (q =2p

1 + p)

Schatten 1- norm (sum of singularvalues)

f (X ) = minΣ⪰0

12

(Tr(

XΣ−1X⊤)+ Tr (Σ)

)= Tr

((X⊤X )1/2

)=

r∑j=1

σj(X ).

q=1q=1.5q=2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 23 / 73

Page 38: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #3: Partial minimum..

......

Partial minimum of a convex function f (x , y)

f (x) := miny∈Rn

f (x , y) is convex.

ExamplesHierarchical prior minimization

f (x) = mind1,...,dn≥0

12

n∑j=1

(x2

j

dj+

dpj

p

)(p ≥ 1)

=1q

n∑j=1

|xj |q (q =2p

1 + p)

Schatten 1- norm (sum of singularvalues)

f (X ) = minΣ⪰0

12

(Tr(

XΣ−1X⊤)+ Tr (Σ)

)= Tr

((X⊤X )1/2

)=

r∑j=1

σj(X ).

q=1q=1.5q=2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 23 / 73

Page 39: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #3: Partial minimum..

......

Partial minimum of a convex function f (x , y)

f (x) := miny∈Rn

f (x , y) is convex.

ExamplesHierarchical prior minimization

f (x) = mind1,...,dn≥0

12

n∑j=1

(x2

j

dj+

dpj

p

)(p ≥ 1)

=1q

n∑j=1

|xj |q (q =2p

1 + p)

Schatten 1- norm (sum of singularvalues)

f (X ) = minΣ⪰0

12

(Tr(

XΣ−1X⊤)+ Tr (Σ)

)

= Tr((X⊤X )1/2

)=

r∑j=1

σj(X ).

q=1q=1.5q=2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 23 / 73

Page 40: Convex Optimization: Old Tricks for New Problems

. . . . . .

.Condition #3: Partial minimum..

......

Partial minimum of a convex function f (x , y)

f (x) := miny∈Rn

f (x , y) is convex.

ExamplesHierarchical prior minimization

f (x) = mind1,...,dn≥0

12

n∑j=1

(x2

j

dj+

dpj

p

)(p ≥ 1)

=1q

n∑j=1

|xj |q (q =2p

1 + p)

Schatten 1- norm (sum of singularvalues)

f (X ) = minΣ⪰0

12

(Tr(

XΣ−1X⊤)+ Tr (Σ)

)= Tr

((X⊤X )1/2

)=

r∑j=1

σj(X ).

q=1q=1.5q=2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 23 / 73

Page 41: Convex Optimization: Old Tricks for New Problems

. . . . . .

Convex optimization problem

f : convex function, g: concave function (−g is convex), C: convex set.

minimizex

f (x), maximizey

g(y),

s.t. x ∈ C. s.t. y ∈ C.

Why?local optimum ⇒ global optimumduality (later) can be used to check convergence

⇒ We can be sure that we are doing the right thing!

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 24 / 73

Page 42: Convex Optimization: Old Tricks for New Problems

. . . . . .

Coming up next:

Gradient descent:

w t+1 = w t − ηt∇f (w t)

What do we do if we have Constraints Non-differentiable terms, like ∥w∥1

⇒ projection/proximity operator

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 25 / 73

Page 43: Convex Optimization: Old Tricks for New Problems

. . . . . .

Proximity operators and iterativeshrinkage/thresholding methods

Learning objectives(Projected) gradient methodIterative shrinkage/thresholding (IST) methodAcceleration

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 26 / 73

Page 44: Convex Optimization: Old Tricks for New Problems

. . . . . .

Proximity view on gradient descent“Linearize and Prox”

w t+1 = argminw

(∇f (w t)(w − w t) +

12ηt

∥w − w t∥2)

= w t − ηt∇f (w t)

Step-size should satisfyηt ≤ 1/L(f ).L(f ): the Lipschitz constant

∥∇f (y)−∇f (x)∥ ≤ L(f )∥y − x∥.

L(f )=upper bound on themaximum eigenvalue of theHessian

wtwt+1w*

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 27 / 73

Page 45: Convex Optimization: Old Tricks for New Problems

. . . . . .

Constraint minimization problem

What do we do, if we have a constraint?

minimizew∈Rn

f (w),

s.t. w ∈ C.

can be equivalently written as

minimizew∈Rn

f (w) + δC(w),

where δC(w) is the indicator function of the set C.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 28 / 73

Page 46: Convex Optimization: Old Tricks for New Problems

. . . . . .

Constraint minimization problem

What do we do, if we have a constraint?

minimizew∈Rn

f (w),

s.t. w ∈ C.

can be equivalently written as

minimizew∈Rn

f (w) + δC(w),

where δC(w) is the indicator function of the set C.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 28 / 73

Page 47: Convex Optimization: Old Tricks for New Problems

. . . . . .

Projected gradient method (Bertsekas 99; Nesterov 03)

Linearize the objective f , δC is the indicator of the constraint C

w t+1 = argminw

(∇f (w t)(w − w t) + δC(w) +

12ηt

∥w − w t∥22

)= argmin

w

(δC(w) +

12ηt

∥w − (w t − ηt∇f (w t))∥22

)= projC(w

t − ηt∇f (w t)).

Requires ηt ≤ 1/L(f ).Convergence rate

f (wk )− f (w∗) ≤L(f )∥w0 − w∗∥2

22k

Need the projection projC tobe easy to compute

−2 −1.5 −1 −0.5 0 0.5 1 1.5 2−2

−1.5

−1

−0.5

0

0.5

1

1.5

2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 29 / 73

Page 48: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ideas for regularized minimizationConstrained minimization problem

minimizew∈Rn

f (w) + δC(w).

⇒ need to compute the projection

w t+1 = argminw

(δC(w) +

12ηt

∥w − y∥22

)

Regularized minimization problem

minimizew∈Rn

f (w) + λ∥w∥1

⇒ need to compute the proximity operator

w t+1 = argminw

(λ∥w∥1 +

12ηt

∥w − y∥22

)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 30 / 73

Page 49: Convex Optimization: Old Tricks for New Problems

. . . . . .

Ideas for regularized minimizationConstrained minimization problem

minimizew∈Rn

f (w) + δC(w).

⇒ need to compute the projection

w t+1 = argminw

(δC(w) +

12ηt

∥w − y∥22

)Regularized minimization problem

minimizew∈Rn

f (w) + λ∥w∥1

⇒ need to compute the proximity operator

w t+1 = argminw

(λ∥w∥1 +

12ηt

∥w − y∥22

)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 30 / 73

Page 50: Convex Optimization: Old Tricks for New Problems

. . . . . .

Proximal Operator: generalization of projection

proxg(y) = argminw

(g(w) +

12∥w − y∥2

2

)

g = δC : Projection onto a convex set projC(y).g(w) = λ∥w∥1: Soft-Threshold

proxλ(y) = argminw

(λ∥w∥1 +

12∥w − y∥2

2

)

=

yj + λ (yj < −λ),

0 (−λ ≤ yj ≤ λ),

yj − λ (yj > λ).

λ−λ y

ST(y)

Prox can be computed easily for a separable f .Non-differentiability is OK.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 31 / 73

Page 51: Convex Optimization: Old Tricks for New Problems

. . . . . .

Exercise

Derive prox operator proxg forRidge regularization

g(w) = λ

n∑j=1

w2j

Group lasso regularization [Yuan & Lin 2006]

g(w1, . . . ,wn) = λ

n∑j=1

∥w j∥2

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 32 / 73

Page 52: Convex Optimization: Old Tricks for New Problems

. . . . . .

Iterative Shrinkage Thresholding (IST)

w t+1 = argminw

(∇f (w t)(w − w t) + λ∥w∥1 +

12ηt

∥w − w t∥22

)= argmin

w

(λ∥w∥1 +

12ηt

∥w − (w t − ηt∇f (w t))∥22

)= proxληt

(w t − ηt∇f (w t)).

The same condition for ηt , thesame O(1/k) convergence (Beck& Teboulle 09)

f (wk )− f (w∗) ≤ L(f )∥w0 − w∗∥2

2k

If the Prox operator proxλ is easy,it is simple to implement.AKA Forward-Backward Splitting(Lions & Mercier 76)Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 33 / 73

Page 53: Convex Optimization: Old Tricks for New Problems

. . . . . .

IST summary

Solve minimization problem

minimizew∈Rn

f (w) + λ∥w∥1

by iteratively computing

w t+1 = proxληt(w t − ηt∇f (w t)),

where

proxλ(y) = argminw

(λ∥w∥1 +

12∥w − y∥2

2

).

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 34 / 73

Page 54: Convex Optimization: Old Tricks for New Problems

. . . . . .

FISTA: accelerated version of IST (Beck & Teboulle 09;

Nesterov 07)

...1 Initialize w0 appropriately, z1 = w0,s1 = 1.

...2 Update w t :

w t = proxληt(z t − ηt∇f (z t)).

...3 Update z t :

z t+1 = w t +

(st − 1st+1

)(w t − w t−1),

where st+1 =(1 +

√1 + 4s2

t)/

2.

The same per iteration complexity. Converges as O(1/k2).Roughly speaking, z t predicts where the IST step should becomputed.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 35 / 73

Page 55: Convex Optimization: Old Tricks for New Problems

. . . . . .

Effect of acceleration

0 2000 4000 6000 8000 1000010-8

10-6

10-4

10-2

100

102

ISTAMTWISTFISTA

Without acceleration

With acceleration

Number of iterations

From Beck & Teboulle 2009 SIAM J. IMAGING SCIENCES

Vol. 2, No. 1, pp. 183-202

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 36 / 73

Page 56: Convex Optimization: Old Tricks for New Problems

. . . . . .

MATLAB Exercise 1: implement an L1 regularizedlogistic regression via IST

minimizew∈Rn

m∑i=1

log(1 + exp(−yi ⟨x i ,w⟩))︸ ︷︷ ︸data-fit

+ λn∑

j=1|wj |︸ ︷︷ ︸

Regularization

Hint: define

fℓ(z) =m∑

i=1

log(1 + exp(−zi)).

Then the problem is

minimize fℓ(Aw) + λn∑

j=1|wj | where A =

y1x1⊤

y2x2⊤

...ymxm

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 37 / 73

Page 57: Convex Optimization: Old Tricks for New Problems

. . . . . .

Some more hints...1 Compute the gradient of the loss term

∇w fℓ(Aw) = −A⊤(

exp(−zi)

1 + exp(−zi)

)m

i=1(z = Aw)

...2 The gradient step becomes

w t+ 12 = w t + ηtA⊤

(exp(−zi)

1 + exp(−zi)

)m

i=1

...3 Then compute the proximity operator

w t+1 = proxληt(w t+ 1

2 )

=

w

t+ 12

j + ληt (wt+ 1

2j < −ληt),

0 (−ληt ≤ wt+ 1

2j ≤ ληt),

wt+ 1

2j − ληt (w

t+ 12

j > ληt).

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 38 / 73

Page 58: Convex Optimization: Old Tricks for New Problems

. . . . . .

Matrix completion via IST (Mazumder et al. 10)Loss function:

f (X ) =12∥Ω(X − Y )∥2.

gradient:

∇f (X ) = Ω⊤(Ω(X − Y ))

Regularization:

g(X ) = λr∑

j=1

σj(X ) (S1-norm).

Prox operator (Singular ValueThresholding):

proxλ(Z ) = U max(S − λI ,0)V⊤.

Iteration:

X t+1 = proxληt

((I − ηtΩ

⊤Ω)(X t)︸ ︷︷ ︸fill in missing

+ ηtΩ⊤Ω(Y t)︸ ︷︷ ︸

observed

)

When ηt = 1, fill missings with predicted values X t , overwrite theobserved with observed values, then soft-threshold.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 39 / 73

Page 59: Convex Optimization: Old Tricks for New Problems

. . . . . .

Matrix completion via IST (Mazumder et al. 10)Loss function:

f (X ) =12∥Ω(X − Y )∥2.

gradient:

∇f (X ) = Ω⊤(Ω(X − Y ))

Regularization:

g(X ) = λr∑

j=1

σj(X ) (S1-norm).

Prox operator (Singular ValueThresholding):

proxλ(Z ) = U max(S − λI ,0)V⊤.

Iteration:

X t+1 = proxληt

((I − ηtΩ

⊤Ω)(X t)︸ ︷︷ ︸fill in missing

+ ηtΩ⊤Ω(Y t)︸ ︷︷ ︸

observed

)

When ηt = 1, fill missings with predicted values X t , overwrite theobserved with observed values, then soft-threshold.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 39 / 73

Page 60: Convex Optimization: Old Tricks for New Problems

. . . . . .

Matrix completion via IST (Mazumder et al. 10)Loss function:

f (X ) =12∥Ω(X − Y )∥2.

gradient:

∇f (X ) = Ω⊤(Ω(X − Y ))

Regularization:

g(X ) = λr∑

j=1

σj(X ) (S1-norm).

Prox operator (Singular ValueThresholding):

proxλ(Z ) = U max(S − λI ,0)V⊤.

Iteration:

X t+1 = proxληt

((I − ηtΩ

⊤Ω)(X t)︸ ︷︷ ︸fill in missing

+ ηtΩ⊤Ω(Y t)︸ ︷︷ ︸

observed

)

When ηt = 1, fill missings with predicted values X t , overwrite theobserved with observed values, then soft-threshold.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 39 / 73

Page 61: Convex Optimization: Old Tricks for New Problems

. . . . . .

Matrix completion via IST (Mazumder et al. 10)Loss function:

f (X ) =12∥Ω(X − Y )∥2.

gradient:

∇f (X ) = Ω⊤(Ω(X − Y ))

Regularization:

g(X ) = λr∑

j=1

σj(X ) (S1-norm).

Prox operator (Singular ValueThresholding):

proxλ(Z ) = U max(S − λI ,0)V⊤.

Iteration:

X t+1 = proxληt

((I − ηtΩ

⊤Ω)(X t)︸ ︷︷ ︸fill in missing

+ ηtΩ⊤Ω(Y t)︸ ︷︷ ︸

observed

)

When ηt = 1, fill missings with predicted values X t , overwrite theobserved with observed values, then soft-threshold.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 39 / 73

Page 62: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate duality and dual ascent

Convex conjugate functionLagrangian relaxation and dual problem

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 40 / 73

Page 63: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate dualityThe convex conjugate f ∗ of a function f :

f ∗(y) = supx∈Rn

(⟨x ,y⟩ − f (x))

f*(y)f(x)

Since the maximum over linear functions is always convex, f need notbe convex.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 41 / 73

Page 64: Convex Optimization: Old Tricks for New Problems

. . . . . .

Demo

Trydemo_conjugate(@(x)x.ˆ2/2,-5:0.1:5);

demo_conjugate(@(x)abs(x),-5:0.1:5);

demo_conjugate(@(x)x.*log(x)+(1-x).*log(1-x),...0.001:0.001:0.999);

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 42 / 73

Page 65: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate duality (dual view).Convex conjugate function..

......Every pair (y , f ∗(y)) corresponds to a tangent line ⟨x ,y⟩ − f ∗(y) of theoriginal function f (x).

Becausef ∗(y) = supx (⟨x ,y⟩ − f (x))implies

If t < f ∗(y), there is a x s.t.

f (x) < ⟨x ,y⟩ − t .

If t ≥ f ∗(y),

f (x) ≥ ⟨x ,y⟩ − t

for every x . -f*(y)y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 43 / 73

Page 66: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate duality (dual view).Convex conjugate function..

......Every pair (y , f ∗(y)) corresponds to a tangent line ⟨x ,y⟩ − f ∗(y) of theoriginal function f (x).

Becausef ∗(y) = supx (⟨x ,y⟩ − f (x))implies

If t < f ∗(y), there is a x s.t.

f (x) < ⟨x ,y⟩ − t .

If t ≥ f ∗(y),

f (x) ≥ ⟨x ,y⟩ − t

for every x .

-f*(y)y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 43 / 73

Page 67: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate duality (dual view).Convex conjugate function..

......Every pair (y , f ∗(y)) corresponds to a tangent line ⟨x ,y⟩ − f ∗(y) of theoriginal function f (x).

Becausef ∗(y) = supx (⟨x ,y⟩ − f (x))implies

If t < f ∗(y), there is a x s.t.

f (x) < ⟨x ,y⟩ − t .

If t ≥ f ∗(y),

f (x) ≥ ⟨x ,y⟩ − t

for every x .-f*(y) y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 43 / 73

Page 68: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conjugate duality (dual view).Convex conjugate function..

......Every pair (y , f ∗(y)) corresponds to a tangent line ⟨x ,y⟩ − f ∗(y) of theoriginal function f (x).

Becausef ∗(y) = supx (⟨x ,y⟩ − f (x))implies

If t < f ∗(y), there is a x s.t.

f (x) < ⟨x ,y⟩ − t .

If t ≥ f ∗(y),

f (x) ≥ ⟨x ,y⟩ − t

for every x . -f*(y)y

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 43 / 73

Page 69: Convex Optimization: Old Tricks for New Problems

. . . . . .

Example of conjugate duality f ∗(y) = supx∈Rn (⟨x , y⟩ − f (x))

Quadratic function

f (x) =x2

2σ2 f ∗(y) =σ2y2

2

f*(y)f(x)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 44 / 73

Page 70: Convex Optimization: Old Tricks for New Problems

. . . . . .

Example of conjugate duality f ∗(y) = supx∈Rn (⟨x , y⟩ − f (x))

Logistic loss function

f (x) = log(1+exp(−x))

f ∗(−y) = y log(y)+ (1− y) log(1− y)

0−1

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 45 / 73

Page 71: Convex Optimization: Old Tricks for New Problems

. . . . . .

Example of conjugate duality f ∗(y) = supx∈Rn (⟨x , y⟩ − f (x))

Logistic loss function

f (x) = log(1+exp(−x)) f ∗(−y) = y log(y)+ (1− y) log(1− y)

0−1

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 45 / 73

Page 72: Convex Optimization: Old Tricks for New Problems

. . . . . .

Example of conjugate duality f ∗(y) = supx∈Rn (⟨x , y⟩ − f (x))

L1 regularizer

f (x) = |x |

f ∗(y) =

0 (−1 ≤ y ≤ 1)+∞ (otherwise)

f(x)

f*(y)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 46 / 73

Page 73: Convex Optimization: Old Tricks for New Problems

. . . . . .

Example of conjugate duality f ∗(y) = supx∈Rn (⟨x , y⟩ − f (x))

L1 regularizer

f (x) = |x | f ∗(y) =

0 (−1 ≤ y ≤ 1)+∞ (otherwise)

f(x)

f*(y)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 46 / 73

Page 74: Convex Optimization: Old Tricks for New Problems

. . . . . .

Bi-conjugate f ∗∗ may be different from f

For nonconvex f ,

f(x)

f*(y)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 47 / 73

Page 75: Convex Optimization: Old Tricks for New Problems

. . . . . .

Lagrangian relaxationOur optimization problem:

minimizew∈Rn

f (Aw) + g(w)

For examplef (z) = 1

2∥z − y∥22

(squared loss)

Equivalently written as

minimizez∈Rm,w∈Rn

f (z) + g(w),

s.t. z = Aw (equality constraint)

Lagrangian relaxation

minimizez,w

L(z ,w ,α) = f (z) + g(w) +α⊤(z − Aw)

As long as z = Aw , the relaxation is exact.supα L(z ,w ,α) recovers the original problem.Minimum of L is no greater than the minimum of the original.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 48 / 73

Page 76: Convex Optimization: Old Tricks for New Problems

. . . . . .

Lagrangian relaxationOur optimization problem:

minimizew∈Rn

f (Aw) + g(w)

For examplef (z) = 1

2∥z − y∥22

(squared loss)

Equivalently written as

minimizez∈Rm,w∈Rn

f (z) + g(w),

s.t. z = Aw (equality constraint)

Lagrangian relaxation

minimizez,w

L(z ,w ,α) = f (z) + g(w) +α⊤(z − Aw)

As long as z = Aw , the relaxation is exact.supα L(z ,w ,α) recovers the original problem.Minimum of L is no greater than the minimum of the original.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 48 / 73

Page 77: Convex Optimization: Old Tricks for New Problems

. . . . . .

Lagrangian relaxationOur optimization problem:

minimizew∈Rn

f (Aw) + g(w)

For examplef (z) = 1

2∥z − y∥22

(squared loss)

Equivalently written as

minimizez∈Rm,w∈Rn

f (z) + g(w),

s.t. z = Aw (equality constraint)

Lagrangian relaxation

minimizez,w

L(z ,w ,α) = f (z) + g(w) +α⊤(z − Aw)

As long as z = Aw , the relaxation is exact.supα L(z ,w ,α) recovers the original problem.Minimum of L is no greater than the minimum of the original.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 48 / 73

Page 78: Convex Optimization: Old Tricks for New Problems

. . . . . .

Lagrangian relaxationOur optimization problem:

minimizew∈Rn

f (Aw) + g(w)

For examplef (z) = 1

2∥z − y∥22

(squared loss)

Equivalently written as

minimizez∈Rm,w∈Rn

f (z) + g(w),

s.t. z = Aw (equality constraint)

Lagrangian relaxation

minimizez,w

L(z ,w ,α) = f (z) + g(w) +α⊤(z − Aw)

As long as z = Aw , the relaxation is exact.supα L(z ,w ,α) recovers the original problem.Minimum of L is no greater than the minimum of the original.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 48 / 73

Page 79: Convex Optimization: Old Tricks for New Problems

. . . . . .

Weak duality

infz,w

L(z ,w ,α) ≤ infw

(f (Aw) + g(w)) =: p∗

proof

infz,w

L(z ,w ,α) = min(

infz=Aw

L(z ,w ,α), infz =Aw

L(z,w ,α)

)= min

(p∗, inf

z =AwL(z ,w ,α)

)≤ p∗

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 49 / 73

Page 80: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual problem

From the above argument

d(α) := infz,w

L(z ,w ,α)

is a lower bound for p∗ for any α. Why don’t we maximize over α?

Dual problem

maximizeα∈Rm

d(α)

Note

supα

infz,w

L(z,w ,α) = d∗ ≤ p∗ = infz,w

supα

L(z ,w ,α)

If d∗ = p∗, strong duality holds. This is the case if f and g both closedand convex.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 50 / 73

Page 81: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual problem

From the above argument

d(α) := infz,w

L(z ,w ,α)

is a lower bound for p∗ for any α. Why don’t we maximize over α?

Dual problem

maximizeα∈Rm

d(α)

Note

supα

infz,w

L(z,w ,α) = d∗ ≤ p∗ = infz,w

supα

L(z ,w ,α)

If d∗ = p∗, strong duality holds. This is the case if f and g both closedand convex.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 50 / 73

Page 82: Convex Optimization: Old Tricks for New Problems

. . . . . .

Fenchel’s dualityFor convex1 functions f and g, and a matrix A ∈ Rm×n

supα∈Rm

(−f ∗(−α)− g∗(A⊤α)

)= inf

w∈Rn

(f (Aw) + g(w)

)Werner Fenchel

1905 – 1988

Only need conjugate functions f ∗ and g∗ to compute the dual.We can make a list of them (like Laplace transform)

MATLAB Exercise 1.5:Compute the Fenchel dual of L1-logistic regression problem inEx.1 and implement the stopping criterion: stop optimization if

(objprim − objdual)/objprim < ϵ (relative duality gap).

1More precisely, proper, closed, and convex.Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 51 / 73

Page 83: Convex Optimization: Old Tricks for New Problems

. . . . . .

Derivation of Fenchel’s duality theorem

d(α) = infz,w

L(z,w ,α)

= infz,w

(f (z) + g(w) +α⊤(z − Aw)

)= inf

z(f (z) + ⟨α, z⟩) + inf

w

(g(w)−

⟨A⊤α,w

⟩)= − sup

z(⟨−α, z⟩ − f (z))− sup

w

(⟨A⊤α,w

⟩− g(w)

)= −f ∗(−α)− g∗(A⊤α)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 52 / 73

Page 84: Convex Optimization: Old Tricks for New Problems

. . . . . .

Augmented Lagrangian and ADMM

Learning objectivesStructured sparse estimationAugmented LagrangianAlternating direction method of multipliers (ADMM)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 53 / 73

Page 85: Convex Optimization: Old Tricks for New Problems

. . . . . .

Recap: Simple vs. structured sparse estimationproblems

Simple sparse estimation problem

minimizew

f (w) + λ∥w∥1

SNP analysis Compressed sensing with Φ−1 (e.g., wavelet) Collaborative filtering (matrix completion)

Structured sparse estimation problem

minimizew

f (w) + λ∥Φw∥1

Compressed sensing without Φ−1 (e.g., total variation) Low-rank tensor completion

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 54 / 73

Page 86: Convex Optimization: Old Tricks for New Problems

. . . . . .

Total Variation based image denoising [Rudin, Osher, Fatemi 92]

minimizeW

12∥W − M∥2

F + λ∑i,j

∥∥∥( ∂x Wij∂y Wij

)∥∥∥2

Original W0 Observed M

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 55 / 73

Page 87: Convex Optimization: Old Tricks for New Problems

. . . . . .

In one dimensionFused lasso [Tibshirani et al. 05]

minimizew

12∥w − y∥2

2 + λn−1∑j=1

∣∣wj+1 − wj∣∣

TrueNoisy

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 56 / 73

Page 88: Convex Optimization: Old Tricks for New Problems

. . . . . .

Structured sparse estimation

TV denoising

minimizeW

12∥W − M∥2

F + λ∑i,j

∥∥∥( ∂x Wij∂y Wij

)∥∥∥2

Fused lasso

minimizew

12∥w − y∥2

2 + λn−1∑j=1

∣∣wj+1 − wj∣∣

.Structured sparse estimation problem..

......

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 57 / 73

Page 89: Convex Optimization: Old Tricks for New Problems

. . . . . .

Structured sparse estimation

TV denoising

minimizeW

12∥W − M∥2

F + λ∑i,j

∥∥∥( ∂x Wij∂y Wij

)∥∥∥2

Fused lasso

minimizew

12∥w − y∥2

2 + λn−1∑j=1

∣∣wj+1 − wj∣∣

.Structured sparse estimation problem..

......

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 57 / 73

Page 90: Convex Optimization: Old Tricks for New Problems

. . . . . .

Structured sparse estimation problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Not easy to compute prox operator (because it is non-separable)⇒ difficult to apply IST-type methods.

Can we use the Lagrangian relaxation trick?

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 58 / 73

Page 91: Convex Optimization: Old Tricks for New Problems

. . . . . .

Structured sparse estimation problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Not easy to compute prox operator (because it is non-separable)⇒ difficult to apply IST-type methods.

Can we use the Lagrangian relaxation trick?

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 58 / 73

Page 92: Convex Optimization: Old Tricks for New Problems

. . . . . .

Forming the LagrangianStructured sparsity problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Equivalently written as

minimizew∈Rn

f (w) + λ∥z∥1︸ ︷︷ ︸separable!

,

s.t. z = Aw (equality constraint)

.Lagrangian function..

......

L(w , z,α) = f (w) + λ∥z∥1 +α⊤(z − Aw).

α: Lagrangian multiplier vector.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 59 / 73

Page 93: Convex Optimization: Old Tricks for New Problems

. . . . . .

Forming the LagrangianStructured sparsity problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Equivalently written as

minimizew∈Rn

f (w) + λ∥z∥1︸ ︷︷ ︸separable!

,

s.t. z = Aw (equality constraint)

.Lagrangian function..

......

L(w , z,α) = f (w) + λ∥z∥1 +α⊤(z − Aw).

α: Lagrangian multiplier vector.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 59 / 73

Page 94: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual ascent.Dual problem..

......maxα

infz,w

(f (w) + λ∥z∥1 +α⊤(z − Aw)

)We can compute the dual objective d(α) by separately minimizing

(1) minw

(f (w)−α⊤Aw

)

= −f ∗(A⊤α),

(2) minz

(λ∥z∥1 +α⊤z

)

= −(λ∥ · ∥1)∗(−α).

But also we get the gradient of d(α) (for free) as follows:

∇αd(α) = z∗ − Aw∗,

where w∗: argmin of (1), z∗: argmin of (2). See Chapter 6, Bertsekas 1999.

Gradient ascent (in the dual)!

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 60 / 73

Page 95: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual ascent.Dual problem..

......maxα

infz,w

(f (w) + λ∥z∥1 +α⊤(z − Aw)

)We can compute the dual objective d(α) by separately minimizing

(1) minw

(f (w)−α⊤Aw

)= −f ∗(A⊤α),

(2) minz

(λ∥z∥1 +α⊤z

)= −(λ∥ · ∥1)

∗(−α).

But also we get the gradient of d(α) (for free) as follows:

∇αd(α) = z∗ − Aw∗,

where w∗: argmin of (1), z∗: argmin of (2). See Chapter 6, Bertsekas 1999.

Gradient ascent (in the dual)!

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 60 / 73

Page 96: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual ascent.Dual problem..

......maxα

infz,w

(f (w) + λ∥z∥1 +α⊤(z − Aw)

)We can compute the dual objective d(α) by separately minimizing

(1) minw

(f (w)−α⊤Aw

)= −f ∗(A⊤α),

(2) minz

(λ∥z∥1 +α⊤z

)= −(λ∥ · ∥1)

∗(−α).

But also we get the gradient of d(α) (for free) as follows:

∇αd(α) = z∗ − Aw∗,

where w∗: argmin of (1), z∗: argmin of (2). See Chapter 6, Bertsekas 1999.

Gradient ascent (in the dual)!

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 60 / 73

Page 97: Convex Optimization: Old Tricks for New Problems

. . . . . .

Dual ascent (Arrow, Hurwicz, & Uzawa 1958)

Minimize the Lagrangian wrt x and z :w t+1 = argminw

(f (w)−α⊤Aw

).

z t+1 = argminz(λ∥z∥1 +α⊤z

),

Update the Lagrangian multiplier αt :αt+1 = αt + ηt(z t+1 − Aw t+1).

Pro: Very simple.Con: When f ∗ or g∗ isnon-differentiable, it is a dualsubgradient method (convergencemore tricky)

NB: f ∗ is differentiable ⇔ f is strictlyconvex.

H. Uzawa

primal

dual

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 61 / 73

Page 98: Convex Optimization: Old Tricks for New Problems

. . . . . .

Forming the augmented LagrangianStructured sparsity problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Equivalently written as (for any η > 0)

minimizew∈Rn

f (w) + λ∥z∥1︸ ︷︷ ︸separable!

2∥z − Aw∥2

2︸ ︷︷ ︸penalty term

,

s.t. z = Aw (equality constraint)

.Augmented Lagrangian function..

......

Lη(w , z,α) = f (w) + λ∥z∥1 +α⊤(z − Aw) +η

2∥z − Aw∥2

2

α: Lagrangian multiplier, η: penalty parameter

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 62 / 73

Page 99: Convex Optimization: Old Tricks for New Problems

. . . . . .

Forming the augmented LagrangianStructured sparsity problem

minimizew∈Rn

f (w)︸ ︷︷ ︸data-fit

+ λ∥Aw∥1︸ ︷︷ ︸regularization

Equivalently written as (for any η > 0)

minimizew∈Rn

f (w) + λ∥z∥1︸ ︷︷ ︸separable!

2∥z − Aw∥2

2︸ ︷︷ ︸penalty term

,

s.t. z = Aw (equality constraint)

.Augmented Lagrangian function..

......

Lη(w , z,α) = f (w) + λ∥z∥1 +α⊤(z − Aw) +η

2∥z − Aw∥2

2

α: Lagrangian multiplier, η: penalty parameter

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 62 / 73

Page 100: Convex Optimization: Old Tricks for New Problems

. . . . . .

Augmented Lagrangian Method.Augmented Lagrangian function..

......Lη(w , z ,α) = f (w) + λ∥z∥1 +α⊤(z − Aw) +

η

2∥z − Aw∥2.

.Augmented Lagrangian method (Hestenes 69, Powell 69)..

......

Minimize the AL function wrt w and z:(w t+1, z t+1) = argmin

w∈Rn,z∈RmLη(w , z ,αt).

Update the Lagrangian multiplier:αt+1 = αt + η(z t+1 − Aw t+1).

Pro: The dual is always differentiable due to the penalty term.Con: Cannot minimize over w and z independently

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 63 / 73

Page 101: Convex Optimization: Old Tricks for New Problems

. . . . . .

Alternating Direction Method of Multipliers (ADMM;Gabay & Mercier 76)

Minimize the AL function Lη(w , z t ,αt) wrt w :w t+1 = argmin

w∈Rn

(f (w) +

η

2∥z t − Aw +αt/η∥2

2

).

Minimize the AL function Lη(w t+1, z,αt) wrt z:z t+1 = argmin

z∈Rm

(λ∥z∥1 +

η

2∥z − Aw t+1 +αt/η∥2

2

).

Update the Lagrangian multiplier:αt+1 = αt + η(z t+1 − Aw t+1).

Looks ad-hoc but convergence can be shown rigorously.Stability does not rely on the choice of step-size η.The newly updated w t+1 enters the computation of z t+1.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 64 / 73

Page 102: Convex Optimization: Old Tricks for New Problems

. . . . . .

MATLAB Exercise 2: implement an ADMM for fusedlasso

Fused lasso

minimizew

12∥w − y∥2

2 + λ

n−1∑j=1

∣∣wj+1 − wj∣∣

What is the loss function f?What is the matrix A for fused lasso?How does the w -update step look?How does the z-update step look?

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 65 / 73

Page 103: Convex Optimization: Old Tricks for New Problems

. . . . . .

Conclusion

Three approaches for various sparse estimation problems Iterative shrinkage/thresholding – proximity operator Uzawa’s method – convex conjugate function ADMM – combination of the above two

Above methods go beyond black-box models (e.g., gradientdescent or Newton’s method) – takes better care of the problemstructures.These methods are simple enough to be implemented rapidly, butshould not be considered as a silver bullet.⇒ Trade-off between:

Quick implementation – test new ideas rapidly Efficient optimization – more inspection/try-and-error/cross

validation

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 66 / 73

Page 104: Convex Optimization: Old Tricks for New Problems

. . . . . .

Topics we did not cover

Beyond polynomial convergence O(1/k2) Dual Augmented Lagrangian (DAL) converges super-linearly

o(exp(−k)). Softwarehttp://mloss.org/software/view/183/(This is limited to non-structured sparse estimation.)

Beyond convexity Generalized eigenvalue problems. Difference of convex (DC) programming. Dual ascent (or dual decomposition) for sequence labeling in

natural language processing; see [Wainwright, Jaakkola, Willsky05; Koo et al. 10]

Stochastic optimization Good tutorial by Nathan Srebro (ICML2010)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 67 / 73

Page 105: Convex Optimization: Old Tricks for New Problems

. . . . . .

Optimization for Machine Learning

A new book “Optimization for Machine Learning” (2011)

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 68 / 73

Page 106: Convex Optimization: Old Tricks for New Problems

. . . . . .

Possible projects

...1 Compare the three approaches, namely IST, dual ascent, andADMM, and discuss empirically (and theoretically) their pros andcons.

...2 Apply one of the methods discussed in the lecture to model somereal problem with (structured) sparsity or low-rank matrix.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 69 / 73

Page 107: Convex Optimization: Old Tricks for New Problems

. . . . . .

References

Recent surveys

Tomioka, Suzuki, & Sugiyama (2011) Augmented Lagrangian Methods for Learning,Selecting, and Combining Features. In Sra, Nowozin, Wright., editors, Optimization forMachine Learning, MIT Press.

Combettes & Pesquet (2010) Proximal splitting methods in signal processing. InFixed-Point Algorithms for Inverse Problems in Science and Engineering. Springer-Verlag.

Boyd, Parikh, Peleato, & Eckstein (2010) Distributed optimization and statistical learningvia the alternating direction method of multipliers.

Textbooks

Rockafellar (1970) Convex Analysis. Princeton University Press.

Bertsekas (1999) Nonlinear Programming. Athena Scientific.

Nesterov (2003) Introductory Lectures on Convex Optimization: A Basic Course. Springer.

Boyd & Vandenberghe. (2004) Convex optimization, Cambridge University Press.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 70 / 73

Page 108: Convex Optimization: Old Tricks for New Problems

. . . . . .

ReferencesIST/FISTA

Moreau (1965) Proximité et dualité dans un espace Hilbertien. Bul letin de la S. M. F.

Nesterov (2007) Gradient Methods for Minimizing Composite Objective Function.

Beck & Teboulle (2009) A Fast Iterative Shrinkage-Thresholding Algorithm for LinearInverse Problems. SIAM J Imag Sci 2, 183–202.

Dual ascent

Arrow, Hurwicz, & Uzawa (1958) Studies in Linear and Non-Linear Programming. StanfordUniversity Press.

Chapter 6 in Bertsekas (1999).

Wainwright, Jaakkola, & Willsky (2005) Map estimation via agreement on trees:message-passing and linear programming. IEEE Trans IT, 51(11).

Augmented Lagrangian

Rockafellar (1976) Augmented Lagrangians and applications of the proximal pointalgorithm in convex programming. Math. of Oper. Res. 1.

Bertsekas (1982) Constrained Optimization and Lagrange Multiplier Methods. AcademicPress.

Tomioka, Suzuki, & Sugiyama (2011) Super-Linear Convergence of Dual AugmentedLagrangian Algorithm for Sparse Learning. JMLR 12.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 71 / 73

Page 109: Convex Optimization: Old Tricks for New Problems

. . . . . .

References

ADMM

Gabay & Mercier (1976) A dual algorithm for the solution of nonlinear variational problemsvia finite element approximation. Comput Math Appl 2, 17–40.

Lions & Mercier (1979) Splitting Algorithms for the Sum of Two Nonlinear Operators. SIAMJ Numer Anal 16, 964–979.

Eckstein & Bertsekas (1992) On the Douglas-Rachford splitting method and the proximalpoint algorithm for maximal monotone operators.

Matrices

Srebro, Rennie, & Jaakkola (2005) Maximum-Margin Matrix Factorization. Advances inNIPS 17, 1329–1336.

Cai, Candès, & Shen (2008) A singular value thresholding algorithm for matrix completion.

Tomioka, Suzuki, Sugiyama, & Kashima (2010) A Fast Augmented Lagrangian Algorithmfor Learning Low-Rank Matrices. In ICML 2010.

Mazumder, Hastie, & Tibshirani (2010) Spectral Regularization Algorithms for LearningLarge Incomplete Matrices. JMLR 11, 2287–2322.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 72 / 73

Page 110: Convex Optimization: Old Tricks for New Problems

. . . . . .

ReferencesMulti-task/Mutliple kernel learning

Evgeniou, Micchelli, & Pontil (2005) Learning Multiple Tasks with Kernel Methods. JMLR 6,615–637.

Lanckriet, Christiani, Bartlett, Ghaoui, & Jordan (2004) Learning the Kernel Matrix withSemidefinite Programming.

Bach, Thibaux, & Jordan (2005) Computing regularization paths for learning multiplekernels. Advances in NIPS, 73–80.

Structured sparsity

Tibshirani, Saunders, Rosset, Zhu and Knight. (2005) Sparsity and smoothness via thefused lasso. J. Roy. Stat. Soc. B, 67.

Rudin, Osher, Fetemi. (1992) Nonlinear total variation based noise removal algorithms.Physica D: Nonlinear Phenomena, 60.

Goldstein & Osher (2009) Split Bregman method for L1 regularization problems. SIAM J.Imag. Sci. 2.

Mairal, Jenatton, Obozinski, & Bach. (2011) Convex and network flow optimization forstructured sparsity.

Bayes & Probabilistic Inference

Wainwright & Jordan (2008) Graphical Models, Exponential Families, and VariationalInference.

Ryota Tomioka (Univ Tokyo) Optimization 2012-08-15 73 / 73