Mímisbrunnr知恵の泉

← 機械学習 一覧

🎓 レベル:発展 | 重要度:B(標準)

📎 前提:勾配降下法 | 応用:サポートベクターマシン(SVM)(KKT・双対の実例)

要点(BLUF)


なぜ凸性が機械学習で嬉しいのか

機械学習の学習は、結局「損失(経験リスク)を最小化する」最適化問題です(学習問題の定式化(仮説・損失・経験リスク))。最適化が難しいのは局所最適に捕まるから。勾配を下っても、それが谷底(大域最適)とは限りません。

ところが目的関数が凸なら、この心配が消えます。**「見つけた局所最適は、必ず大域最適」**が保証されるからです。これが凸最適化を学ぶ最大の動機です。順に、凸集合 → 凸関数 → その嬉しい性質、と積み上げます。


凸集合

定義:集合 C⊆RnC \subseteq \mathbb{R}^n が凸集合であるとは、CC 内の任意の2点 x,yx, y と任意の λ∈[0,1]\lambda \in [0,1] について

λx+(1−λ)y∈C\lambda x + (1-\lambda) y \in C

が成り立つこと。

要するに、集合内の2点を結ぶ線分が、はみ出さずに丸ごと集合に収まるということです。λx+(1−λ)y\lambda x + (1-\lambda)y は xx と yy を結ぶ線分上の点(λ=1\lambda=1 で xx、λ=0\lambda=0 で yy)を表します。

制約 gi(x)≤0g_i(x) \le 0 がどれも凸関数なら、実行可能領域 {x:gi(x)≤0}\{x : g_i(x)\le 0\} は凸集合の共通部分なので凸になります。これが後で効きます。


凸関数

定義:関数 f:Rn→Rf : \mathbb{R}^n \to \mathbb{R}(定義域は凸集合)が凸関数(お椀型で、極小がそのまま全体の最小になる扱いやすい関数)であるとは、任意の x,yx, y と λ∈[0,1]\lambda \in [0,1] について

f(λx+(1−λ)y)  ≤  λf(x)+(1−λ)f(y)f(\lambda x + (1-\lambda) y) \;\le\; \lambda f(x) + (1-\lambda) f(y)

が成り立つこと。

要するに、2点を結んだ「弦(直線)」が、関数のグラフより上にあるということです。左辺は2点の中間(線分上の点)での関数値、右辺はその弦の高さ。弦が常にグラフを上から押さえているのが凸です。不等号が常に真の不等号 <<(x≠yx\ne y、λ∈(0,1)\lambda\in(0,1))なら狭義凸といいます。

一次条件(微分可能なとき)

ff が微分可能なら、凸性は次と同値です:

f(y)  ≥  f(x)+∇f(x)⊤(y−x)∀x,yf(y) \;\ge\; f(x) + \nabla f(x)^\top (y - x) \qquad \forall x, y

要するに、関数が自分の接平面より常に上にあるということです。右辺は点 xx での一次近似(接平面)。凸関数は接線(接平面)が必ず下から支える「お椀型」だと言っています。

この条件の重要な帰結:もし ∇f(x⋆)=0\nabla f(x^\star)=0 なら、任意の yy で f(y)≥f(x⋆)f(y) \ge f(x^\star)。つまり勾配がゼロになる点は、それだけで大域最適です(無制約のとき)。非凸では「勾配ゼロ=停留点」止まりで、極小・極大・鞍点の区別がつかないのと対照的です。

二次条件(2回微分可能なとき)

ff が2回微分可能なら、凸性はヘッセ行列 ∇2f(x)\nabla^2 f(x) が半正定値であることと同値です:

∇2f(x)  ⪰  0∀x\nabla^2 f(x) \;\succeq\; 0 \qquad \forall x

ここで ∇2f(x)⪰0\nabla^2 f(x) \succeq 0 は「任意の vv に対し v⊤∇2f(x) v≥0v^\top \nabla^2 f(x)\, v \ge 0」、同値に「全固有値が非負」。要するに、どの方向に動いても曲率が下に凸(上に反らない)ということです。固有値がすべて正(正定値 ∇2f≻0\nabla^2 f \succ 0)なら狭義凸になります(十分条件)。

⚠️ 逆は成り立ちません。f(x)=x4f(x)=x^4 は狭義凸ですが f′′(0)=0f''(0)=0 で正定値ではありません。「狭義凸 ⇒ ヘッセ正定値」は偽、「ヘッセ正定値 ⇒ 狭義凸」が真です。

機械学習での具体例:二乗誤差 f(w)=∥Xw−y∥2f(w)=\|Xw-y\|^2 はヘッセが ∇2f=2X⊤X⪰0\nabla^2 f = 2X^\top X \succeq 0 なので凸(線形回帰(最小二乗法と確率的解釈))。L2L_2 正則化項 λ∥w∥2\lambda\|w\|^2 はヘッセ 2λI≻02\lambda I \succ 0 で狭義凸。ロジスティック回帰の負の対数尤度も凸です。


なぜ凸だと「局所最適=大域最適」なのか(証明)

これが凸最適化の心臓です。背理法できちんと示します。

主張:ff が凸関数のとき、x⋆x^\star が局所最適(ある半径 r>0r>0 の近傍で最小)なら、x⋆x^\star は大域最適である。

証明(背理法): x⋆x^\star を局所最適とし、これが大域最適でないと仮定します。すると、ある点 yy が存在して

f(y)<f(x⋆)f(y) < f(x^\star)

が成り立ちます。ここで x⋆x^\star と yy を結ぶ線分上の点を、x⋆x^\star に十分近い側に取ります:

z=λx⋆+(1−λ)y,λ∈(0,1)z = \lambda x^\star + (1-\lambda) y, \qquad \lambda \in (0,1)

凸関数の定義より、

f(z)  ≤  λf(x⋆)+(1−λ)f(y)f(z) \;\le\; \lambda f(x^\star) + (1-\lambda) f(y)

仮定 f(y)<f(x⋆)f(y) < f(x^\star) を右辺に使うと、(1−λ)>0(1-\lambda)>0 なので

λf(x⋆)+(1−λ)f(y)  <  λf(x⋆)+(1−λ)f(x⋆)=f(x⋆)\lambda f(x^\star) + (1-\lambda) f(y) \;<\; \lambda f(x^\star) + (1-\lambda) f(x^\star) = f(x^\star)

よって

f(z)<f(x⋆)f(z) < f(x^\star)

ところが、λ\lambda を 1 に十分近く取れば zz は x⋆x^\star の半径 rr の近傍に入れます(zz と x⋆x^\star の距離は ∥z−x⋆∥=(1−λ)∥y−x⋆∥\|z-x^\star\| = (1-\lambda)\|y-x^\star\| で、λ→1\lambda\to 1 でいくらでも小さくできるため)。すると「近傍内に x⋆x^\star より小さい点 zz がある」ことになり、x⋆x^\star が局所最適であることに矛盾します。

したがって、f(y)<f(x⋆)f(y) < f(x^\star) となる yy は存在せず、x⋆x^\star は大域最適です。■\quad\blacksquare

直観:弦がグラフを上から押さえる(凸の定義)以上、「局所的にはここが底だが、遠くにもっと低い谷がある」という状況は作れません。もっと低い点があるなら、そこへ向かう線分上を少し進むだけで必ず下がってしまうからです。だから凸では谷は1つ(連結な底)。

graph LR
  A["f が凸関数"] --> B["弦が常にグラフより上<br/>近傍に下がる方向が必ずある"]
  B --> C["局所最適より低い点 y があれば<br/>線分上の z で f が更に下がる"]
  C --> D["局所最適の定義に矛盾"]
  D --> E["局所最適 = 大域最適<br/>勾配を下るだけで谷底に届く"]

この性質のおかげで、凸問題では「勾配がゼロの点(停留点)を1つ見つける」=「大域最適を見つける」になります。勾配降下法(勾配降下法)が安心して使えるのはこのためです。


制約付き最適化とラグランジュ関数

実問題には制約が付きます。標準形は:

min⁡xf(x)s.t.gi(x)≤0,i=1,…,mhj(x)=0,j=1,…,p\begin{aligned} \min_{x}\quad & f(x) \\ \text{s.t.}\quad & g_i(x) \le 0, \quad i=1,\dots,m \\ & h_j(x) = 0, \quad j=1,\dots,p \end{aligned}

この最適値を p⋆p^\star(主問題の最適値)と書きます。制約付きを直接扱うのは面倒なので、制約を目的関数にペナルティとして取り込むのがラグランジュの発想です。

ラグランジュ関数:

L(x,λ,ν)=f(x)+∑i=1mλi gi(x)+∑j=1pνj hj(x)L(x, \lambda, \nu) = f(x) + \sum_{i=1}^{m} \lambda_i\, g_i(x) + \sum_{j=1}^{p} \nu_j\, h_j(x)

要するに、「制約を破ったらペナルティを払う」形に書き換え、ペナルティの強さ(乗数)を変数として一緒に扱うわけです。乗数は「その制約をどれだけ厳しく効かせるかの値段(影の価格)」と読めます。


ラグランジュ双対

双対関数

ラグランジュ関数を xx について最小化したものを双対関数と呼びます:

g(λ,ν)=inf⁡x L(x,λ,ν)g(\lambda, \nu) = \inf_{x}\, L(x, \lambda, \nu)

重要な性質:双対関数は(f,gi,hjf, g_i, h_j が何であれ)常に凹関数です。理由は、LL は固定した xx に対して (λ,ν)(\lambda,\nu) のアフィン(一次)関数であり、アフィン関数の下限(点ごとの inf)は凹関数になるから。主問題が非凸でも双対は凹なので扱いやすい、という強みがあります。

弱双対(常に成立)

双対問題は、この下界を最大化します:

d⋆=max⁡λ≥0, ν g(λ,ν)d^\star = \max_{\lambda \ge 0,\, \nu}\, g(\lambda, \nu)

弱双対定理:d⋆≤p⋆d^\star \le p^\star が常に成り立つ。

証明:任意の実行可能点 x~\tilde{x}(gi(x~)≤0g_i(\tilde x)\le 0, hj(x~)=0h_j(\tilde x)=0)と任意の λ≥0,ν\lambda\ge 0,\nu について、

L(x~,λ,ν)=f(x~)+∑iλigi(x~)⏟≤0 (λi≥0, gi≤0)+∑jνjhj(x~)⏟=0 (hj=0) ≤ f(x~)L(\tilde x,\lambda,\nu)=f(\tilde x)+\underbrace{\sum_i \lambda_i g_i(\tilde x)}_{\le 0\ (\lambda_i\ge0,\,g_i\le0)}+\underbrace{\sum_j \nu_j h_j(\tilde x)}_{=0\ (h_j=0)}\ \le\ f(\tilde x)

一方、inf の定義から g(λ,ν)=inf⁡xL(x,λ,ν)≤L(x~,λ,ν)g(\lambda,\nu)=\inf_x L(x,\lambda,\nu)\le L(\tilde x,\lambda,\nu)。両者を繋ぐと、任意の実行可能 x~\tilde x で

g(λ,ν) ≤ f(x~)g(\lambda,\nu)\ \le\ f(\tilde x)

右辺で実行可能 x~\tilde x にわたり下限を取ると g(λ,ν)≤p⋆g(\lambda,\nu)\le p^\star。最後に左辺で λ≥0,ν\lambda\ge0,\nu にわたり上限を取れば d⋆≤p⋆d^\star\le p^\star。■\quad\blacksquare

要するに、双対問題は主問題の「下からの最良の見積もり」を与える。差 p⋆−d⋆≥0p^\star - d^\star \ge 0 を双対ギャップと呼びます。

強双対とスレーター条件

強双対:d⋆=p⋆d^\star = p^\star(ギャップがゼロ)。これは一般には成り立ちませんが、凸問題では十分条件があります。

スレーター条件:主問題が凸(f,gif, g_i が凸、hjh_j がアフィン)で、すべての不等式制約を真に満たす実行可能点が存在する(ある xx で gi(x)<0g_i(x) < 0 がすべて成立。アフィン制約は等号でよい)。

このとき強双対が成立します(d⋆=p⋆d^\star=p^\star)。要するに、「制約の内側にちゃんと余白がある凸問題」なら、双対問題を解けば主問題の答えがそのまま得られるということ。SVMの双対形式(サポートベクターマシン(SVM))が主問題と同じ答えを返し、しかもカーネル化できるのはこの強双対のおかげです。


KKT条件

強双対が成り立つとき、最適解 (x⋆,λ⋆,ν⋆)(x^\star, \lambda^\star, \nu^\star) は次の4条件をすべて満たします。これがKKT(Karush–Kuhn–Tucker)条件です。

(1) 停留性∇f(x⋆)+∑iλi⋆∇gi(x⋆)+∑jνj⋆∇hj(x⋆)=0(2) 主実行可能性gi(x⋆)≤0,hj(x⋆)=0(3) 双対実行可能性λi⋆≥0(4) 相補性条件λi⋆ gi(x⋆)=0∀i\begin{aligned} &\textbf{(1) 停留性} &&\nabla f(x^\star) + \sum_i \lambda_i^\star \nabla g_i(x^\star) + \sum_j \nu_j^\star \nabla h_j(x^\star) = 0 \\ &\textbf{(2) 主実行可能性} &&g_i(x^\star) \le 0,\quad h_j(x^\star) = 0 \\ &\textbf{(3) 双対実行可能性}&&\lambda_i^\star \ge 0 \\ &\textbf{(4) 相補性条件} &&\lambda_i^\star\, g_i(x^\star) = 0 \quad \forall i \end{aligned}

各条件の意味:

KKTの2つの顔(凸の場合):

graph TB
  K["KKT条件<br/>制約付き凸最適の最適性の特徴づけ"]
  K --> S1["(1)停留性<br/>∇L が 0"]
  K --> S2["(2)主実行可能性<br/>g_i ≤ 0, h_j = 0"]
  K --> S3["(3)双対実行可能性<br/>λ_i ⪰ 0"]
  K --> S4["(4)相補性条件<br/>λ_i・g_i = 0"]
  S4 --> SV["効かない制約は λ_i = 0<br/>正の λ_i の制約だけアクティブ"]

SVMがKKTの実例

SVM(サポートベクターマシン(SVM))は凸(二次計画)+スレーター条件を満たすので、強双対とKKTがそのまま使えます。ハードマージンSVMの主問題は

min⁡w,b 12∥w∥2s.t.yi(w⊤xi+b)≥1 ⟺ gi=1−yi(w⊤xi+b)≤0\min_{w,b}\ \tfrac12\|w\|^2 \quad \text{s.t.}\quad y_i(w^\top x_i + b) \ge 1 \ \Longleftrightarrow\ g_i = 1 - y_i(w^\top x_i + b) \le 0

各データ点 ii に乗数 λi≥0\lambda_i \ge 0 が付きます。ここで相補性条件(制約が効いていない所では対応する係数がゼロになる条件) λi gi=0\lambda_i\, g_i = 0 が決定的です:

さらに停留性条件 ∇wL=0\nabla_w L = 0 から w=∑iλiyixiw = \sum_i \lambda_i y_i x_i が出て、λi>0\lambda_i>0 の点(サポートベクター)だけで ww が決まることが分かります。「決定境界が一部の点だけで決まる」というSVMの核心が、相補性条件から直接導かれるわけです。


正則化も制約付き最適化として読める

Ridge/Lasso(正則化(Ridge・Lasso・Elastic Net))は2通りに書けて、両者は等価です:

min⁡w ∥Xw−y∥2+α∥w∥qq⏟ペナルティ形式(ラグランジュ形式)⟺min⁡w ∥Xw−y∥2 s.t. ∥w∥qq≤t⏟制約形式\underbrace{\min_w\ \|Xw-y\|^2 + \alpha\|w\|_q^q}_{\text{ペナルティ形式(ラグランジュ形式)}} \qquad\Longleftrightarrow\qquad \underbrace{\min_w\ \|Xw-y\|^2 \ \text{s.t.}\ \|w\|_q^q \le t}_{\text{制約形式}}

左のペナルティ形式は、まさに右の制約付き問題のラグランジュ関数(α\alpha が乗数)です。各 tt に対応する α\alpha が存在し、強双対により最適解が一致します。q=2q=2 がRidge(凸・狭義凸)、q=1q=1 がLasso(凸だが非平滑)。「正則化=係数の大きさに予算制約を課す」という見方は、ここから来ています。


⚠️ よくある誤解・落とし穴


まとめ

凸最適化は「凸集合上で凸関数を最小化する」問題で、局所最適=大域最適という強い保証(背理法で証明)が成り立つため、勾配を下るだけで解けます。制約が付くとラグランジュ関数で扱い、双対問題が下界(弱双対は常に成立)を、強双対(凸+スレーター条件)が等号を与えます。最適性はKKT条件(停留性・主実行可能性・双対実行可能性・相補性)で特徴づけられ、凸ではKKTが必要十分。SVMのサポートベクターや正則化の予算制約は、この理論のそのままの応用です。深層学習の損失は非凸なので保証は外れますが、凸理論は「学習がうまくいく理想形」の基準として効きます。


対応するシミュレーション

simulations/convex_nonconvex.py:凸関数 f(x)=x2f(x)=x^2 と非凸関数 f(x)=x4−4x2+0.5xf(x)=x^4-4x^2+0.5x(谷が2つ)に対し、複数の初期点から勾配降下を回します。凸関数ではどの初期点からも同じ大域的最小に到達するのに対し、非凸関数では初期点しだいで異なる局所的最小に捕まることを可視化します(深いネットの損失が非凸であることの含意)。

凸=大域最小 / 非凸=初期値依存の局所解

関連ノート

📌 統計の土台:正則化のベイズ的解釈(MAP推定)は統計サイトのベイズ推定・MAP推定へ繋がります。