0. イントロ
確率論や統計学では、ランダムな量の値そのものよりも、その量が典型的な値からどの程度ずれうるかを知りたい場面がよくあります。
例として、独立同分布(independent and identically distributed; i.i.d.)な確率変数 X 1 , … , X n X_1, \dots, X_n X 1 , … , X n の標本平均
X ˉ n : = 1 n ∑ i = 1 n X i \bar{X}_n := \frac{1}{n} \sum_{i=1}^n X_i X ˉ n := n 1 i = 1 ∑ n X i
を考えます。
E [ ∣ X 1 ∣ ] < ∞ \mathbb{E}[|X_1|] < \infty E [ ∣ X 1 ∣ ] < ∞ なら、大数の法則により、n → ∞ n \to \infty n → ∞ で X ˉ n \bar{X}_n X ˉ n は期待値 μ : = E [ X 1 ] \mu := \mathbb{E}[X_1] μ := E [ X 1 ] に確率 1 1 1 で収束します。
しかし、実際のデータ解析では n n n は有限です。
そこで知りたいのは、与えられた n n n と ε > 0 \varepsilon > 0 ε > 0 に対して、次の確率がどれだけ小さいかです。
Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) \Pr\left(|\bar{X}_n - \mu| \geq \varepsilon\right) Pr ( ∣ X ˉ n − μ ∣ ≥ ε )
この確率を、n n n 、ε \varepsilon ε 、分布の性質(値域や分散など)を使って上から評価する不等式を 集中不等式(concentration inequality) と呼びます。
中心極限定理も X ˉ n \bar{X}_n X ˉ n のゆらぎを記述しますが、その主張は n → ∞ n \to \infty n → ∞ の極限における分布の形であり、有限の n n n での誤差の大きさは含みません。
集中不等式は、有限の n n n でそのまま成り立つ上界を与えます。
本記事で使う例:成功確率の推定
本記事では、次の問題を例にして各不等式を比較します。
成功確率 p p p の試行を独立に n n n 回行い、i i i 回目が成功なら X i = 1 X_i = 1 X i = 1 、失敗なら X i = 0 X_i = 0 X i = 0 とします。
p p p を成功割合 p ^ : = 1 n ∑ i = 1 n X i \hat{p} := \frac{1}{n} \sum_{i=1}^n X_i p ^ := n 1 ∑ i = 1 n X i で推定するとき、誤差が ε \varepsilon ε 以上になる確率を δ \delta δ 以下にするには、n n n をいくつ以上にすればよいでしょうか。
条件は次の 2 つとし、それぞれ δ = 0.05 \delta = 0.05 δ = 0.05 と δ = 10 − 6 \delta = 10^{-6} δ = 1 0 − 6 の場合を考えます。
(A) p = 0.5 p = 0.5 p = 0.5 , ε = 0.05 \varepsilon = 0.05 ε = 0.05 :1 回の試行の分散 p ( 1 − p ) p(1-p) p ( 1 − p ) が最大の 1 / 4 1/4 1/4 になる場合
(B) p = 0.01 p = 0.01 p = 0.01 , ε = 0.005 \varepsilon = 0.005 ε = 0.005 :成功がまれで、分散 p ( 1 − p ) ≈ 0.0099 p(1-p) \approx 0.0099 p ( 1 − p ) ≈ 0.0099 が小さい場合
まず、(A) の条件で p ^ \hat{p} p ^ の分布を Python で調べます。
次のコードは、n = 10 , 100 , 1000 n = 10, 100, 1000 n = 10 , 100 , 1000 のそれぞれについて p ^ \hat{p} p ^ を 10 5 10^5 1 0 5 回生成し、ヒストグラムを描きます。
import numpy as np
import matplotlib . pyplot as plt
rng = np . random . default_rng ( 0 )
p , eps , trials = 0.5 , 0.05 , 100_000
for n in [ 10 , 100 , 1000 ] :
# 成功確率 p の試行を n 回行ったときの成功割合を trials 回分生成する
p_hat = rng . binomial ( n , p , size = trials ) / n
# 誤差が eps 以上になった割合(1e-12 は浮動小数点の丸め誤差対策)
miss = np . mean ( np . abs ( p_hat - p ) >= eps - 1e-12 )
bins = ( np . arange ( n + 2 ) - 0.5 ) / n # 各 k/n を中心とする幅 1/n のビン
plt . hist ( p_hat , bins = bins , density = True , histtype = "step" , lw = 2 ,
label = f"$n = { n } $ ( { miss : .3f } )" )
plt . axvspan ( p - eps , p + eps , color = "gray" , alpha = 0.2 )
plt . xlim ( 0.2 , 0.8 )
plt . xlabel ( r"$\hat p$" )
plt . ylabel ( "Density" )
plt . legend ( loc = "upper left" )
plt . show ( )
図 1:(A) の条件(p = 0.5)での成功割合の分布。灰色の帯は |p̂ − p| < 0.05 の範囲、凡例の括弧内は誤差が 0.05 以上になった割合。
図 1 のように、n n n を増やすと p ^ \hat{p} p ^ の分布は p = 0.5 p = 0.5 p = 0.5 の周りに集中し、誤差が 0.05 0.05 0.05 以上になる割合は n = 10 n = 10 n = 10 で 0.752 0.752 0.752 、n = 100 n = 100 n = 100 で 0.367 0.367 0.367 、n = 1000 n = 1000 n = 1000 で 0.002 0.002 0.002 と減少します。
集中不等式は、分布を細かく計算せずに、この割合の上界を与えます。
記事中にコードを載せていない図 2 と図 5 も含め、本記事のすべての図は code/concentration/make_figures.py で再生成できます。
本記事のロードマップ
各節で扱う不等式と、その不等式が使う情報は次のとおりです。
Markov の不等式 (第 2 節):非負性と期待値
Chebyshev の不等式 (第 3 節):分散
Chernoff 法 (第 4 節):積率母関数
Hoeffding の不等式 (第 5 節):独立性と値域
Bernstein の不等式 (第 6 節):独立性、値域、分散
McDiarmid の不等式 (第 7 節):独立性と、1 つの入力を変えたときの出力の変化の上限
第 8 節で例の結果を比較し、第 9 節で仮定が満たされない場合の道具を紹介します。
後の節の不等式ほど常に強いわけではありません。
どの不等式が小さい上界を与えるかは、使える仮定と問題の条件によって変わります。
1. 記法と基本事項
Pr ( A ) \Pr(A) Pr ( A ) は事象 A A A の確率、E [ X ] \mathbb{E}[X] E [ X ] は確率変数 X X X の期待値、Var ( X ) : = E [ ( X − E [ X ] ) 2 ] \operatorname{Var}(X) := \mathbb{E}[(X - \mathbb{E}[X])^2] Var ( X ) := E [( X − E [ X ] ) 2 ] は分散を表します。
1 A \mathbf{1}_A 1 A は事象 A A A の指示関数で、A A A が起きれば 1 1 1 、起きなければ 0 0 0 をとります。
「X ≥ 0 X \geq 0 X ≥ 0 a.s.」は、確率 1 1 1 で X ≥ 0 X \geq 0 X ≥ 0 が成り立つこと(almost surely)を表します。
確率変数 X 1 , … , X n X_1, \dots, X_n X 1 , … , X n の和を S n : = ∑ i = 1 n X i S_n := \sum_{i=1}^n X_i S n := ∑ i = 1 n X i 、標本平均を X ˉ n : = S n / n \bar{X}_n := S_n / n X ˉ n := S n / n と書きます。
log \log log は自然対数を表します。
裾確率と union bound
確率変数 X X X が期待値から t > 0 t > 0 t > 0 以上離れる確率を 裾確率(tail probability) と呼びます。
離れる向きによって、次の 3 種類があります。
上側の裾確率:Pr ( X − E [ X ] ≥ t ) \Pr(X - \mathbb{E}[X] \geq t) Pr ( X − E [ X ] ≥ t )
下側の裾確率:Pr ( X − E [ X ] ≤ − t ) \Pr(X - \mathbb{E}[X] \leq -t) Pr ( X − E [ X ] ≤ − t )
両側の裾確率:Pr ( ∣ X − E [ X ] ∣ ≥ t ) \Pr(|X - \mathbb{E}[X]| \geq t) Pr ( ∣ X − E [ X ] ∣ ≥ t )
集中不等式は、これらの裾確率の上界(tail bound)を与えます。
任意の事象 A , B A, B A , B について Pr ( A ∪ B ) ≤ Pr ( A ) + Pr ( B ) \Pr(A \cup B) \leq \Pr(A) + \Pr(B) Pr ( A ∪ B ) ≤ Pr ( A ) + Pr ( B ) が成り立つので(union bound)、両側の裾確率は上側と下側の裾確率の上界の和で抑えられます。
そのため、上側と下側で同じ上界が成り立つ場合、両側の上界には係数 2 2 2 が付きます。
2. Markov の不等式
Markov の不等式は、非負性と期待値だけから裾確率を評価します。
本記事の残りの不等式は、いずれも証明のどこかでこの不等式を使います。
X ≥ 0 X \geq 0 X ≥ 0 a.s. かつ E [ X ] < ∞ \mathbb{E}[X] < \infty E [ X ] < ∞ とします。
このとき、任意の t > 0 t > 0 t > 0 に対して
Pr ( X ≥ t ) ≤ E [ X ] t \Pr(X \geq t) \leq \frac{\mathbb{E}[X]}{t} Pr ( X ≥ t ) ≤ t E [ X ] が成り立ちます。
証明 X ≥ 0 X \geq 0 X ≥ 0 a.s. なので、確率 1 1 1 で X ≥ t 1 { X ≥ t } X \geq t \mathbf{1}_{\{X \geq t\}} X ≥ t 1 { X ≥ t } が成り立ちます(X ≥ t X \geq t X ≥ t のとき右辺は t t t 、それ以外のとき右辺は 0 0 0 です)。
両辺の期待値をとると E [ X ] ≥ t Pr ( X ≥ t ) \mathbb{E}[X] \geq t \Pr(X \geq t) E [ X ] ≥ t Pr ( X ≥ t ) となり、両辺を t t t で割ると定理 1 を得ます。□ \square □
t = c E [ X ] t = c\, \mathbb{E}[X] t = c E [ X ] (c > 0 c > 0 c > 0 )とおくと、Pr ( X ≥ c E [ X ] ) ≤ 1 / c \Pr(X \geq c\, \mathbb{E}[X]) \leq 1/c Pr ( X ≥ c E [ X ]) ≤ 1/ c です。
例えば、非負の確率変数が期待値の 10 10 10 倍以上になる確率は 1 / 10 1/10 1/10 以下です。
仮定が弱い代わりに、上界は t t t について 1 / t 1/t 1/ t の速さでしか減少しません。
指示関数を非負の関数で抑える
証明で使ったのは、指示関数 1 { X ≥ t } \mathbf{1}_{\{X \geq t\}} 1 { X ≥ t } を非負の関数で上から抑え、その期待値をとる、という手順です。
一般に、確率変数 Y Y Y と非負の関数 φ \varphi φ が、Y Y Y のとりうるすべての値 y y y で φ ( y ) ≥ 1 { y ≥ t } \varphi(y) \geq \mathbf{1}_{\{y \geq t\}} φ ( y ) ≥ 1 { y ≥ t } を満たせば、
Pr ( Y ≥ t ) = E [ 1 { Y ≥ t } ] ≤ E [ φ ( Y ) ] \Pr(Y \geq t) = \mathbb{E}\left[\mathbf{1}_{\{Y \geq t\}}\right] \leq \mathbb{E}[\varphi(Y)] Pr ( Y ≥ t ) = E [ 1 { Y ≥ t } ] ≤ E [ φ ( Y )]
が成り立ちます。
φ \varphi φ の選び方によって、異なる不等式が得られます(図 2)。
図 2:指示関数 1{y ≥ t}(黒、t = 1)を上から抑える非負の関数。(a) は t の近く、(b) は広い範囲で、(b) の破線の枠が (a) の範囲。Markov の不等式は y/t(y ≥ 0 の場合)、Chebyshev の不等式は y²/t²、Chernoff 法は exp(λ(y − t)) を使う。
Markov の不等式 (第 2 節):非負の Y = X Y = X Y = X に φ ( y ) = y / t \varphi(y) = y/t φ ( y ) = y / t を使います。
Chebyshev の不等式 (第 3 節):Y = X − E [ X ] Y = X - \mathbb{E}[X] Y = X − E [ X ] に φ ( y ) = y 2 / t 2 \varphi(y) = y^2/t^2 φ ( y ) = y 2 / t 2 を使います。この φ \varphi φ は y ≤ − t y \leq -t y ≤ − t でも 1 1 1 以上なので、両側の裾確率を一度に抑えます。
Chernoff 法 (第 4 節):Y = X − E [ X ] Y = X - \mathbb{E}[X] Y = X − E [ X ] に φ ( y ) = e λ ( y − t ) \varphi(y) = e^{\lambda(y - t)} φ ( y ) = e λ ( y − t ) (λ > 0 \lambda > 0 λ > 0 )を使います。
Y Y Y が値をとりやすい範囲で φ \varphi φ が指示関数に近いほど、上界は小さくなります。
Chernoff 法では λ \lambda λ によって φ \varphi φ の形を調整できます。
図 2 (a) の t t t の近くでは、λ = 4 \lambda = 4 λ = 4 の曲線は λ = 1.5 \lambda = 1.5 λ = 1.5 の曲線と比べて、y < t y < t y < t で指示関数に近い代わりに、y > t y > t y > t で速く増加します。
第 4 節では、この λ \lambda λ を最適化します。
図 2 (b) の広い範囲では、遠くでの増え方の違いが分かります。
Markov の不等式の y / t y/t y / t は y y y の 1 次関数で、Chebyshev の不等式の y 2 / t 2 y^2/t^2 y 2 / t 2 は 0 0 0 から左右どちらに離れても増加します。
Chernoff 法の e λ ( y − t ) e^{\lambda(y - t)} e λ ( y − t ) は、y y y が小さくなるほど 0 0 0 に近づく一方、y > t y > t y > t では指数関数的に増加します。
そのため、Chernoff 法で E [ φ ( Y ) ] \mathbb{E}[\varphi(Y)] E [ φ ( Y )] が有限になるには、Y Y Y の積率母関数(第 4 節)が有限であること、つまり Y Y Y の上側の裾が指数関数以上の速さで減衰することが必要です。
3. Chebyshev の不等式
Chebyshev の不等式は、分散を使って期待値の周りへの集中を評価します。
X X X の分散 Var ( X ) \operatorname{Var}(X) Var ( X ) が有限とします。
このとき、任意の t > 0 t > 0 t > 0 に対して
Pr ( ∣ X − E [ X ] ∣ ≥ t ) ≤ Var ( X ) t 2 \Pr(|X - \mathbb{E}[X]| \geq t) \leq \frac{\operatorname{Var}(X)}{t^2} Pr ( ∣ X − E [ X ] ∣ ≥ t ) ≤ t 2 Var ( X ) が成り立ちます。
証明 非負の確率変数 ( X − E [ X ] ) 2 (X - \mathbb{E}[X])^2 ( X − E [ X ] ) 2 に Markov の不等式を適用すると、
Pr ( ∣ X − E [ X ] ∣ ≥ t ) = Pr ( ( X − E [ X ] ) 2 ≥ t 2 ) ≤ E [ ( X − E [ X ] ) 2 ] t 2 = Var ( X ) t 2 \begin{aligned}
\Pr(|X - \mathbb{E}[X]| \geq t)
&= \Pr\left((X - \mathbb{E}[X])^2 \geq t^2\right) \\
&\leq \frac{\mathbb{E}\left[(X - \mathbb{E}[X])^2\right]}{t^2}
= \frac{\operatorname{Var}(X)}{t^2}
\end{aligned} Pr ( ∣ X − E [ X ] ∣ ≥ t ) = Pr ( ( X − E [ X ] ) 2 ≥ t 2 ) ≤ t 2 E [ ( X − E [ X ] ) 2 ] = t 2 Var ( X )
を得ます。□ \square □
標本平均への適用
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n が互いに無相関で、期待値 μ \mu μ と分散 σ 2 \sigma^2 σ 2 が共通とします。
このとき Var ( X ˉ n ) = σ 2 / n \operatorname{Var}(\bar{X}_n) = \sigma^2/n Var ( X ˉ n ) = σ 2 / n なので、定理 2 より
Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ σ 2 n ε 2 \Pr(|\bar{X}_n - \mu| \geq \varepsilon) \leq \frac{\sigma^2}{n\varepsilon^2} Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ n ε 2 σ 2
です。
Var ( X ˉ n ) \operatorname{Var}(\bar{X}_n) Var ( X ˉ n ) の計算に使ったのは分散の加法性だけなので、独立性より弱い無相関性で足ります。
例への適用 1 回の試行の分散は σ 2 = p ( 1 − p ) \sigma^2 = p(1-p) σ 2 = p ( 1 − p ) なので、右辺を δ \delta δ 以下にするには
n ≥ p ( 1 − p ) ε 2 δ n \geq \frac{p(1-p)}{\varepsilon^2 \delta} n ≥ ε 2 δ p ( 1 − p )
とすれば十分です。
(A) で δ = 0.05 \delta = 0.05 δ = 0.05 なら n ≥ 2,000 n \geq 2{,}000 n ≥ 2 , 000 です。
必要な n n n は 1 / δ 1/\delta 1/ δ に比例するため、δ = 10 − 6 \delta = 10^{-6} δ = 1 0 − 6 では n ≥ 10 8 n \geq 10^8 n ≥ 1 0 8 になります。
4. Chernoff 法
Chebyshev の不等式の上界は t t t の多項式でしか減少しません。
指数関数を使うと、t t t について指数的に減少する上界が得られます。
この手法を Chernoff 法、または Cramér–Chernoff 法と呼びます(Chernoff, 1952; Boucheron et al., 2013, §2.2)。
一般形
λ ∈ R \lambda \in \mathbb{R} λ ∈ R の関数 E [ e λ X ] \mathbb{E}[e^{\lambda X}] E [ e λ X ] を、X X X の 積率母関数(moment generating function; MGF) と呼びます。
本記事では、中心化した確率変数の MGF の対数
ψ X ( λ ) : = log E [ e λ ( X − E [ X ] ) ] \psi_X(\lambda) := \log \mathbb{E}\left[e^{\lambda (X - \mathbb{E}[X])}\right] ψ X ( λ ) := log E [ e λ ( X − E [ X ]) ]
を使います。
λ > 0 \lambda > 0 λ > 0 で ψ X ( λ ) \psi_X(\lambda) ψ X ( λ ) が有限なら、x ↦ e λ x x \mapsto e^{\lambda x} x ↦ e λ x は単調増加なので、Markov の不等式より
Pr ( X − E [ X ] ≥ t ) = Pr ( e λ ( X − E [ X ] ) ≥ e λ t ) ≤ e − λ t E [ e λ ( X − E [ X ] ) ] = exp ( − λ t + ψ X ( λ ) ) \begin{aligned}
\Pr(X - \mathbb{E}[X] \geq t)
&= \Pr\left(e^{\lambda (X - \mathbb{E}[X])} \geq e^{\lambda t}\right) \\
&\leq e^{-\lambda t}\, \mathbb{E}\left[e^{\lambda (X - \mathbb{E}[X])}\right]
= \exp\left(-\lambda t + \psi_X(\lambda)\right)
\end{aligned} Pr ( X − E [ X ] ≥ t ) = Pr ( e λ ( X − E [ X ]) ≥ e λ t ) ≤ e − λ t E [ e λ ( X − E [ X ]) ] = exp ( − λ t + ψ X ( λ ) )
が成り立ちます。
この式は ψ X ( λ ) \psi_X(\lambda) ψ X ( λ ) が有限なすべての λ > 0 \lambda > 0 λ > 0 で成り立つので、右辺を λ \lambda λ について最小化して
Pr ( X − E [ X ] ≥ t ) ≤ exp ( − sup λ > 0 { λ t − ψ X ( λ ) } ) \Pr(X - \mathbb{E}[X] \geq t) \leq \exp\left(-\sup_{\lambda > 0} \left\{\lambda t - \psi_X(\lambda)\right\}\right) Pr ( X − E [ X ] ≥ t ) ≤ exp ( − λ > 0 sup { λ t − ψ X ( λ ) } )
を得ます。
この上界を Chernoff bound と呼びます。
ψ X \psi_X ψ X を上から評価できれば、具体的な上界が得られます。
下側の裾確率は、− X -X − X に同じ議論を適用して評価します。
独立な確率変数の和
指数関数を使うと、独立な確率変数の和の MGF は、各変数の MGF の積に分解されます。
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n が独立なら
E [ e λ ( S n − E [ S n ] ) ] = ∏ i = 1 n E [ e λ ( X i − E [ X i ] ) ] \mathbb{E}\left[e^{\lambda (S_n - \mathbb{E}[S_n])}\right] = \prod_{i=1}^n \mathbb{E}\left[e^{\lambda (X_i - \mathbb{E}[X_i])}\right] E [ e λ ( S n − E [ S n ]) ] = i = 1 ∏ n E [ e λ ( X i − E [ X i ]) ]
であり、両辺の対数をとると ψ S n ( λ ) = ∑ i = 1 n ψ X i ( λ ) \psi_{S_n}(\lambda) = \sum_{i=1}^n \psi_{X_i}(\lambda) ψ S n ( λ ) = ∑ i = 1 n ψ X i ( λ ) です。
したがって、各 X i X_i X i の ψ X i \psi_{X_i} ψ X i を評価すれば、和 S n S_n S n の裾確率を評価できます。
第 5 節の Hoeffding の不等式と第 6 節の Bernstein の不等式は、この手順で導かれます。
sub-Gaussian
ある ν > 0 \nu > 0 ν > 0 が存在して、すべての λ ∈ R \lambda \in \mathbb{R} λ ∈ R で
ψ X ( λ ) ≤ ν λ 2 2 \psi_X(\lambda) \leq \frac{\nu \lambda^2}{2} ψ X ( λ ) ≤ 2 ν λ 2 が成り立つとき、X X X は分散パラメータ ν \nu ν の sub-Gaussian であるといいます。
分散 ν \nu ν の正規分布に従う確率変数では、ψ X ( λ ) = ν λ 2 / 2 \psi_X(\lambda) = \nu\lambda^2/2 ψ X ( λ ) = ν λ 2 /2 が等号で成り立ちます。
つまり sub-Gaussian とは、中心化した MGF が正規分布の MGF 以下に収まる確率変数です。
sub-Gaussian な X X X に Chernoff 法を適用すると、λ t − ν λ 2 / 2 \lambda t - \nu\lambda^2/2 λ t − ν λ 2 /2 は λ = t / ν \lambda = t/\nu λ = t / ν で最大値 t 2 / ( 2 ν ) t^2/(2\nu) t 2 / ( 2 ν ) をとるので、
Pr ( X − E [ X ] ≥ t ) ≤ exp ( − t 2 2 ν ) \Pr(X - \mathbb{E}[X] \geq t) \leq \exp\left(-\frac{t^2}{2\nu}\right) Pr ( X − E [ X ] ≥ t ) ≤ exp ( − 2 ν t 2 )
を得ます。
定義は λ < 0 \lambda < 0 λ < 0 も含むので、下側の裾確率にも同じ上界が成り立ち、両側では係数 2 2 2 が付きます(Boucheron et al., 2013, §2.3)。
また、独立な X i X_i X i がそれぞれ分散パラメータ ν i \nu_i ν i の sub-Gaussian なら、ψ S n = ∑ i ψ X i \psi_{S_n} = \sum_i \psi_{X_i} ψ S n = ∑ i ψ X i より、S n S_n S n は分散パラメータ ∑ i = 1 n ν i \sum_{i=1}^n \nu_i ∑ i = 1 n ν i の sub-Gaussian です。
第 5 節では、この性質を使って Hoeffding の不等式を導きます。
Bernoulli 確率変数の和に対する相対誤差の上界
Chernoff 法を Bernoulli 確率変数の和に適用すると、期待値に対する相対誤差の上界が得られます(Mitzenmacher & Upfal, 2017, 第 4 章)。
定理 3(相対誤差の Chernoff bound)
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n を独立な確率変数とし、X i ∈ { 0 , 1 } X_i \in \{0, 1\} X i ∈ { 0 , 1 } 、Pr ( X i = 1 ) = p i \Pr(X_i = 1) = p_i Pr ( X i = 1 ) = p i とします。
m : = E [ S n ] = ∑ i = 1 n p i m := \mathbb{E}[S_n] = \sum_{i=1}^n p_i m := E [ S n ] = ∑ i = 1 n p i とおくと、任意の η > 0 \eta > 0 η > 0 に対して
Pr ( S n ≥ ( 1 + η ) m ) ≤ ( e η ( 1 + η ) 1 + η ) m ≤ exp ( − η 2 m 2 + η ) \begin{aligned}
\Pr\left(S_n \geq (1 + \eta) m\right)
&\leq \left(\frac{e^{\eta}}{(1 + \eta)^{1 + \eta}}\right)^m \\
&\leq \exp\left(-\frac{\eta^2 m}{2 + \eta}\right)
\end{aligned} Pr ( S n ≥ ( 1 + η ) m ) ≤ ( ( 1 + η ) 1 + η e η ) m ≤ exp ( − 2 + η η 2 m ) が成り立ちます。
また、0 < η < 1 0 < \eta < 1 0 < η < 1 に対して
Pr ( S n ≤ ( 1 − η ) m ) ≤ exp ( − η 2 m 2 ) \Pr\left(S_n \leq (1 - \eta) m\right) \leq \exp\left(-\frac{\eta^2 m}{2}\right) Pr ( S n ≤ ( 1 − η ) m ) ≤ exp ( − 2 η 2 m ) が成り立ちます。
1 つ目の式の 2 番目の不等号は、η ≥ 0 \eta \geq 0 η ≥ 0 で成り立つ log ( 1 + η ) ≥ 2 η / ( 2 + η ) \log(1 + \eta) \geq 2\eta/(2 + \eta) log ( 1 + η ) ≥ 2 η / ( 2 + η ) から従います。
この定理は、誤差を期待値 m m m との比 η \eta η で測ります。
m m m が n n n に比べて小さいとき、第 5 節の Hoeffding の不等式より小さい上界を与えることがあります(第 8 節)。
例への適用 p i = p p_i = p p i = p なら m = n p m = np m = n p です。
事象 ∣ p ^ − p ∣ ≥ ε |\hat{p} - p| \geq \varepsilon ∣ p ^ − p ∣ ≥ ε は、相対誤差 η : = ε / p \eta := \varepsilon/p η := ε / p が上側または下側に生じる事象です。
η < 1 \eta < 1 η < 1 のとき、下側の指数 η 2 m / 2 \eta^2 m/2 η 2 m /2 は上側の指数 η 2 m / ( 2 + η ) \eta^2 m/(2 + \eta) η 2 m / ( 2 + η ) 以上なので、両側を合わせて
Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ 2 exp ( − η 2 n p 2 + η ) \Pr(|\hat{p} - p| \geq \varepsilon) \leq 2\exp\left(-\frac{\eta^2 np}{2 + \eta}\right) Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ 2 exp ( − 2 + η η 2 n p )
です。
右辺を δ \delta δ 以下にする条件は n ≥ ( 2 + η ) log ( 2 / δ ) / ( η 2 p ) n \geq (2 + \eta)\log(2/\delta) / (\eta^2 p) n ≥ ( 2 + η ) log ( 2/ δ ) / ( η 2 p ) で、(B)(η = 0.5 \eta = 0.5 η = 0.5 )で δ = 0.05 \delta = 0.05 δ = 0.05 なら n ≥ 3,689 n \geq 3{,}689 n ≥ 3 , 689 です。
5. Hoeffding の不等式
独立な確率変数がそれぞれ既知の有界区間に値をとるとき、Chernoff 法から Hoeffding の不等式が導かれます(Hoeffding, 1963)。
鍵になるのは、有界な確率変数が sub-Gaussian であることを示す次の補題です。
a ≤ X ≤ b a \leq X \leq b a ≤ X ≤ b a.s. なら、すべての λ ∈ R \lambda \in \mathbb{R} λ ∈ R で
ψ X ( λ ) ≤ λ 2 ( b − a ) 2 8 \psi_X(\lambda) \leq \frac{\lambda^2 (b - a)^2}{8} ψ X ( λ ) ≤ 8 λ 2 ( b − a ) 2 が成り立ちます。
つまり、X X X は分散パラメータ ( b − a ) 2 / 4 (b - a)^2/4 ( b − a ) 2 /4 の sub-Gaussian です。
証明は Boucheron et al. (2013) の第 2 章を参照してください。
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n を独立な確率変数とし、各 i i i で a i ≤ X i ≤ b i a_i \leq X_i \leq b_i a i ≤ X i ≤ b i a.s. とします。
このとき、任意の t > 0 t > 0 t > 0 に対して
Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − 2 t 2 ∑ i = 1 n ( b i − a i ) 2 ) \Pr(S_n - \mathbb{E}[S_n] \geq t) \leq \exp\left(-\frac{2t^2}{\sum_{i=1}^n (b_i - a_i)^2}\right) Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − ∑ i = 1 n ( b i − a i ) 2 2 t 2 ) が成り立ちます。
下側の裾確率にも同じ上界が成り立ち、両側の裾確率の上界は右辺の 2 倍です。
証明 Hoeffding の補題より、各 X i X_i X i は分散パラメータ ( b i − a i ) 2 / 4 (b_i - a_i)^2/4 ( b i − a i ) 2 /4 の sub-Gaussian です。
第 4 節で見たように、独立な和 S n S_n S n は分散パラメータ ν : = ∑ i = 1 n ( b i − a i ) 2 / 4 \nu := \sum_{i=1}^n (b_i - a_i)^2/4 ν := ∑ i = 1 n ( b i − a i ) 2 /4 の sub-Gaussian なので、
Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − t 2 2 ν ) = exp ( − 2 t 2 ∑ i = 1 n ( b i − a i ) 2 ) \begin{aligned}
\Pr(S_n - \mathbb{E}[S_n] \geq t)
&\leq \exp\left(-\frac{t^2}{2\nu}\right) \\
&= \exp\left(-\frac{2t^2}{\sum_{i=1}^n (b_i - a_i)^2}\right)
\end{aligned} Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − 2 ν t 2 ) = exp ( − ∑ i = 1 n ( b i − a i ) 2 2 t 2 )
を得ます。□ \square □
標本平均と必要な標本数
各 X i ∈ [ a , b ] X_i \in [a, b] X i ∈ [ a , b ] a.s. で、期待値 μ \mu μ が共通なら、定理 4 で t = n ε t = n\varepsilon t = n ε とおいて
Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ 2 exp ( − 2 n ε 2 ( b − a ) 2 ) \Pr(|\bar{X}_n - \mu| \geq \varepsilon) \leq 2\exp\left(-\frac{2n\varepsilon^2}{(b - a)^2}\right) Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ 2 exp ( − ( b − a ) 2 2 n ε 2 )
を得ます。
ε \varepsilon ε を固定すると、この上界は n n n について指数的に減少します。
第 3 節の Chebyshev の上界 σ 2 / ( n ε 2 ) \sigma^2/(n\varepsilon^2) σ 2 / ( n ε 2 ) は、1 / n 1/n 1/ n の速さでしか減少しません。
右辺を δ \delta δ とおいて解くと、次の 2 つの形が得られます。
n ≥ ( b − a ) 2 2 ε 2 log 2 δ n \geq \dfrac{(b - a)^2}{2\varepsilon^2} \log\dfrac{2}{\delta} n ≥ 2 ε 2 ( b − a ) 2 log δ 2 なら、Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ δ \Pr(|\bar{X}_n - \mu| \geq \varepsilon) \leq \delta Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ δ が成り立ちます。
任意の n n n について、確率 1 − δ 1 - \delta 1 − δ 以上で ∣ X ˉ n − μ ∣ < ( b − a ) log ( 2 / δ ) 2 n |\bar{X}_n - \mu| < (b - a)\sqrt{\dfrac{\log(2/\delta)}{2n}} ∣ X ˉ n − μ ∣ < ( b − a ) 2 n log ( 2/ δ ) が成り立ちます。
例への適用 X i ∈ [ 0 , 1 ] X_i \in [0, 1] X i ∈ [ 0 , 1 ] なので、条件は n ≥ log ( 2 / δ ) / ( 2 ε 2 ) n \geq \log(2/\delta)/(2\varepsilon^2) n ≥ log ( 2/ δ ) / ( 2 ε 2 ) です。
(A) で δ = 0.05 \delta = 0.05 δ = 0.05 なら n ≥ 738 n \geq 738 n ≥ 738 、δ = 10 − 6 \delta = 10^{-6} δ = 1 0 − 6 でも n ≥ 2,902 n \geq 2{,}902 n ≥ 2 , 902 です。
必要な n n n は log ( 1 / δ ) \log(1/\delta) log ( 1/ δ ) に比例して増えるので、1 / δ 1/\delta 1/ δ に比例して増える Chebyshev の不等式との差は、δ \delta δ が小さいほど広がります。
一方、(B) では δ = 0.05 \delta = 0.05 δ = 0.05 でも n ≥ 73,778 n \geq 73{,}778 n ≥ 73 , 778 が必要です。
Hoeffding の不等式は値域 [ 0 , 1 ] [0, 1] [ 0 , 1 ] しか使わないため、(B) の分散 p ( 1 − p ) ≈ 0.0099 p(1-p) \approx 0.0099 p ( 1 − p ) ≈ 0.0099 が、値域 [ 0 , 1 ] [0, 1] [ 0 , 1 ] の確率変数の分散の最大値 1 / 4 1/4 1/4 の約 1 / 25 1/25 1/25 であることを反映できません。
この点を改善するのが、第 6 節の Bernstein の不等式です。
定理 4 の a i , b i a_i, b_i a i , b i は、分布について事前に分かっている値域です。
手元の標本で観測された最小値 min i X i \min_i X_i min i X i と最大値 max i X i \max_i X_i max i X i を代わりに使うと、まだ観測されていない値の可能性を無視することになり、上界は保証されません。
6. Bernstein の不等式
Hoeffding の不等式は値域しか使いません。
値域が広くても、ほとんどの値が期待値の近くに集まる確率変数では、分散も使う Bernstein の不等式がより小さい上界を与えます(Boucheron et al., 2013, §2.8)。
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n を独立な確率変数とし、ある M > 0 M > 0 M > 0 について、各 i i i で ∣ X i − E [ X i ] ∣ ≤ M |X_i - \mathbb{E}[X_i]| \leq M ∣ X i − E [ X i ] ∣ ≤ M a.s. とします。
V : = ∑ i = 1 n Var ( X i ) V := \sum_{i=1}^n \operatorname{Var}(X_i) V := ∑ i = 1 n Var ( X i ) とおくと、任意の t > 0 t > 0 t > 0 に対して
Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − t 2 2 ( V + M t / 3 ) ) \Pr(S_n - \mathbb{E}[S_n] \geq t) \leq \exp\left(-\frac{t^2}{2(V + Mt/3)}\right) Pr ( S n − E [ S n ] ≥ t ) ≤ exp ( − 2 ( V + M t /3 ) t 2 ) が成り立ちます。
下側の裾確率にも同じ上界が成り立ち、両側の裾確率の上界は右辺の 2 倍です。
2 つの減衰の速さ
分母の V + M t / 3 V + Mt/3 V + M t /3 の 2 つの項のどちらが大きいかで、上界の減衰の速さが変わります。
V + M t / 3 ≤ 2 max { V , M t / 3 } V + Mt/3 \leq 2\max\{V, Mt/3\} V + M t /3 ≤ 2 max { V , M t /3 } より、
exp ( − t 2 2 ( V + M t / 3 ) ) ≤ exp ( − min { t 2 4 V , 3 t 4 M } ) \exp\left(-\frac{t^2}{2(V + Mt/3)}\right) \leq \exp\left(-\min\left\{\frac{t^2}{4V}, \frac{3t}{4M}\right\}\right) exp ( − 2 ( V + M t /3 ) t 2 ) ≤ exp ( − min { 4 V t 2 , 4 M 3 t } )
が成り立ちます。
t ≤ 3 V / M t \leq 3V/M t ≤ 3 V / M のとき、指数は t 2 / ( 4 V ) t^2/(4V) t 2 / ( 4 V ) 以上です。上界は、分散 V V V の正規分布と同じく t 2 t^2 t 2 に比例する指数で減衰します。
t > 3 V / M t > 3V/M t > 3 V / M のとき、指数は 3 t / ( 4 M ) 3t/(4M) 3 t / ( 4 M ) 以上です。上界は、指数分布と同じく t t t に比例する指数で減衰します。
標本平均と Hoeffding の不等式との比較
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n が i.i.d. で、E [ X i ] = μ \mathbb{E}[X_i] = \mu E [ X i ] = μ 、Var ( X i ) = σ 2 \operatorname{Var}(X_i) = \sigma^2 Var ( X i ) = σ 2 、∣ X i − μ ∣ ≤ M |X_i - \mu| \leq M ∣ X i − μ ∣ ≤ M a.s. とします。
定理 5 に V = n σ 2 V = n\sigma^2 V = n σ 2 、t = n ε t = n\varepsilon t = n ε を代入すると、
Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ 2 exp ( − n ε 2 2 ( σ 2 + M ε / 3 ) ) \Pr(|\bar{X}_n - \mu| \geq \varepsilon) \leq 2\exp\left(-\frac{n\varepsilon^2}{2(\sigma^2 + M\varepsilon/3)}\right) Pr ( ∣ X ˉ n − μ ∣ ≥ ε ) ≤ 2 exp ( − 2 ( σ 2 + M ε /3 ) n ε 2 )
を得ます。
X i ∈ [ a , b ] X_i \in [a, b] X i ∈ [ a , b ] のとき、分散は最大で ( b − a ) 2 / 4 (b - a)^2/4 ( b − a ) 2 /4 です。
上式の σ 2 \sigma^2 σ 2 をこの最大値で置き換え、M ε / 3 M\varepsilon/3 M ε /3 を無視すると、指数は 2 n ε 2 / ( b − a ) 2 2n\varepsilon^2/(b - a)^2 2 n ε 2 / ( b − a ) 2 となり、第 5 節の Hoeffding の上界の指数と一致します。
つまり、Hoeffding の不等式は分散が最大の場合を想定した上界と見なせます。
2 つの指数を比べると、σ 2 + M ε / 3 < ( b − a ) 2 / 4 \sigma^2 + M\varepsilon/3 < (b - a)^2/4 σ 2 + M ε /3 < ( b − a ) 2 /4 のとき、Bernstein の上界のほうが小さくなります。
例への適用 σ 2 = p ( 1 − p ) \sigma^2 = p(1-p) σ 2 = p ( 1 − p ) 、M = max { p , 1 − p } M = \max\{p, 1 - p\} M = max { p , 1 − p } とすると、条件は
n ≥ 2 ( σ 2 + M ε / 3 ) ε 2 log 2 δ n \geq \frac{2(\sigma^2 + M\varepsilon/3)}{\varepsilon^2} \log\frac{2}{\delta} n ≥ ε 2 2 ( σ 2 + M ε /3 ) log δ 2
です。
(B) で δ = 0.05 \delta = 0.05 δ = 0.05 なら n ≥ 3,409 n \geq 3{,}409 n ≥ 3 , 409 で、Hoeffding の不等式の 73,778 73{,}778 73 , 778 の約 1 / 22 1/22 1/22 です。
(A) では σ 2 = 1 / 4 \sigma^2 = 1/4 σ 2 = 1/4 が最大値なので、n ≥ 763 n \geq 763 n ≥ 763 となり、Hoeffding の不等式の 738 738 738 よりわずかに大きくなります。
7. McDiarmid の不等式
第 5 節と第 6 節では、確率変数の和を扱いました。
McDiarmid の不等式(有界差分不等式、bounded differences inequality)は、和に限らず、独立な確率変数の関数 f ( X 1 , … , X n ) f(X_1, \dots, X_n) f ( X 1 , … , X n ) を扱います(McDiarmid, 1989; Boucheron et al., 2013, §6.1)。
関数 f : X 1 × ⋯ × X n → R f: \mathcal{X}_1 \times \cdots \times \mathcal{X}_n \to \mathbb{R} f : X 1 × ⋯ × X n → R と定数 c 1 , … , c n ≥ 0 c_1, \dots, c_n \geq 0 c 1 , … , c n ≥ 0 が、すべての i i i と、i i i 番目の入力だけが異なる任意の 2 点について
∣ f ( x 1 , … , x i , … , x n ) − f ( x 1 , … , x i ′ , … , x n ) ∣ ≤ c i \left|f(x_1, \dots, x_i, \dots, x_n) - f(x_1, \dots, x_i', \dots, x_n)\right| \leq c_i ∣ f ( x 1 , … , x i , … , x n ) − f ( x 1 , … , x i ′ , … , x n ) ∣ ≤ c i を満たすとき、f f f は 有界差分条件(bounded differences condition) を満たすといいます。
有界差分条件は、i i i 番目の入力だけを変えても出力は c i c_i c i 以下しか変わらない、という安定性を表します。
X 1 , … , X n X_1, \dots, X_n X 1 , … , X n を独立な確率変数とし、X i X_i X i は X i \mathcal{X}_i X i に値をとるとします。
f f f が定数 c 1 , … , c n c_1, \dots, c_n c 1 , … , c n で有界差分条件を満たすとき、Z : = f ( X 1 , … , X n ) Z := f(X_1, \dots, X_n) Z := f ( X 1 , … , X n ) について、任意の t > 0 t > 0 t > 0 に対して
Pr ( Z − E [ Z ] ≥ t ) ≤ exp ( − 2 t 2 ∑ i = 1 n c i 2 ) \Pr(Z - \mathbb{E}[Z] \geq t) \leq \exp\left(-\frac{2t^2}{\sum_{i=1}^n c_i^2}\right) Pr ( Z − E [ Z ] ≥ t ) ≤ exp ( − ∑ i = 1 n c i 2 2 t 2 ) が成り立ちます。
下側の裾確率にも同じ上界が成り立ち、両側の裾確率の上界は右辺の 2 倍です。
Hoeffding の不等式との関係
f f f が和 f ( x 1 , … , x n ) = ∑ i = 1 n x i f(x_1, \dots, x_n) = \sum_{i=1}^n x_i f ( x 1 , … , x n ) = ∑ i = 1 n x i で、x i ∈ [ a i , b i ] x_i \in [a_i, b_i] x i ∈ [ a i , b i ] の場合を考えます。
i i i 番目の入力を変えたときの和の変化は b i − a i b_i - a_i b i − a i 以下なので、c i = b i − a i c_i = b_i - a_i c i = b i − a i を定理 6 に代入すると、定理 4(Hoeffding の不等式)と同じ式になります。
McDiarmid の不等式は、Hoeffding の不等式の「独立な有界確率変数の和」という仮定を、「各入力の変化に対して安定な関数」に広げた結果です。
例:経験分布関数の一様な誤差
和ではない関数の例として、X 1 , … , X n X_1, \dots, X_n X 1 , … , X n が i.i.d. で累積分布関数 F F F に従うとき、経験分布関数 F n ( x ) : = 1 n ∑ i = 1 n 1 { X i ≤ x } F_n(x) := \frac{1}{n} \sum_{i=1}^n \mathbf{1}_{\{X_i \leq x\}} F n ( x ) := n 1 ∑ i = 1 n 1 { X i ≤ x } と F F F の差の最大値
Z : = sup x ∈ R ∣ F n ( x ) − F ( x ) ∣ Z := \sup_{x \in \mathbb{R}} |F_n(x) - F(x)| Z := x ∈ R sup ∣ F n ( x ) − F ( x ) ∣
を考えます。
Z Z Z は Kolmogorov–Smirnov 統計量と呼ばれ、標本が分布 F F F に従うかどうかの検定に使われます。
X i X_i X i を 1 個変えると、各 x x x で F n ( x ) F_n(x) F n ( x ) は 1 / n 1/n 1/ n 以下しか変わらないので、その上限 Z Z Z も 1 / n 1/n 1/ n 以下しか変わりません。
c i = 1 / n c_i = 1/n c i = 1/ n より ∑ i = 1 n c i 2 = 1 / n \sum_{i=1}^n c_i^2 = 1/n ∑ i = 1 n c i 2 = 1/ n なので、定理 6 より
Pr ( Z − E [ Z ] ≥ t ) ≤ e − 2 n t 2 \Pr(Z - \mathbb{E}[Z] \geq t) \leq e^{-2nt^2} Pr ( Z − E [ Z ] ≥ t ) ≤ e − 2 n t 2
です。
Z Z Z の分布を有限の n n n で正確に求めるのは容易ではありませんが、この上界は F F F によらず成り立ちます。
この上界を Python で確かめます。
次のコードは、一様分布 U n i f o r m ( 0 , 1 ) \mathrm{Uniform}(0, 1) Uniform ( 0 , 1 ) (F ( x ) = x F(x) = x F ( x ) = x )から n = 100 , 1000 n = 100, 1000 n = 100 , 1000 個の標本を 10 4 10^4 1 0 4 回生成して Z Z Z を計算し、Pr ( Z − E [ Z ] ≥ t ) \Pr(Z - \mathbb{E}[Z] \geq t) Pr ( Z − E [ Z ] ≥ t ) を推定します。
E [ Z ] \mathbb{E}[Z] E [ Z ] には 10 4 10^4 1 0 4 回の平均を使います。
import numpy as np
import matplotlib . pyplot as plt
rng = np . random . default_rng ( 0 )
reps = 10_000
s = np . linspace ( 0 , 1.5 , 151 ) # 横軸は sqrt(n) * t
for n in [ 100 , 1000 ] :
# Uniform(0, 1) の標本を reps 組生成し、各組を昇順に並べる
u = np . sort ( rng . random ( ( reps , n ) ) , axis = 1 )
i = np . arange ( 1 , n + 1 )
# Z = sup_x |F_n(x) - x|(経験分布関数と真の分布関数 F(x) = x の差の最大値)
z = np . maximum ( i / n - u , u - ( i - 1 ) / n ) . max ( axis = 1 )
tail = [ np . mean ( z - z . mean ( ) >= x / np . sqrt ( n ) ) for x in s ]
plt . semilogy ( s , tail , lw = 2 , label = f"Simulation ($n = { n } $)" )
plt . semilogy ( s , np . exp ( - 2 * s ** 2 ) , "k--" , lw = 2 ,
label = r"McDiarmid bound $e^{-2nt^2}$" )
plt . ylim ( 1e-4 , 1.5 )
plt . xlabel ( r"$\sqrt{n}\,t$" )
plt . ylabel ( r"$\Pr(Z - \mathbb{E}[Z] \geq t)$" )
plt . legend ( )
plt . show ( )
図 3:Kolmogorov–Smirnov 統計量 Z の上側の裾確率(シミュレーション)と McDiarmid の上界。横軸を √n t とすると、上界は n によらない 1 本の曲線になる。縦軸は対数目盛。
横軸を n t \sqrt{n}\, t n t にとると、上界 e − 2 n t 2 = e − 2 ( n t ) 2 e^{-2nt^2} = e^{-2(\sqrt{n}\, t)^2} e − 2 n t 2 = e − 2 ( n t ) 2 は n n n によらない 1 本の曲線になります。
図 3 では、n = 100 n = 100 n = 100 と n = 1000 n = 1000 n = 1000 のシミュレーション結果がほぼ重なり、どちらも上界より下にあります。
つまり、Z − E [ Z ] Z - \mathbb{E}[Z] Z − E [ Z ] のゆらぎは 1 / n 1/\sqrt{n} 1/ n の大きさで、McDiarmid の不等式はその大きさを正しく捉えています。
機械学習でも同じ構造が現れます。
値が [ 0 , 1 ] [0, 1] [ 0 , 1 ] に収まる損失関数の集合を G \mathcal{G} G とし、sup g ∈ G ∣ 1 n ∑ i = 1 n g ( X i ) − E [ g ( X 1 ) ] ∣ \sup_{g \in \mathcal{G}} \left|\frac{1}{n} \sum_{i=1}^n g(X_i) - \mathbb{E}[g(X_1)]\right| sup g ∈ G n 1 ∑ i = 1 n g ( X i ) − E [ g ( X 1 )] を考えると、上と同じ議論で c i = 1 / n c_i = 1/n c i = 1/ n の有界差分条件が成り立ちます。
この量は、訓練データ上の平均損失と期待損失の差の上限であり、汎化誤差の評価に使われます。
Rademacher 複雑度による汎化誤差の評価では、McDiarmid の不等式でこの量をその期待値の近くに抑え、期待値を Rademacher 複雑度で評価します(Bartlett & Mendelson, 2002)。
コラム:証明の考え方(martingale による分解)
McDiarmid の不等式は、Z − E [ Z ] Z - \mathbb{E}[Z] Z − E [ Z ] を、X 1 , X 2 , … X_1, X_2, \dots X 1 , X 2 , … を 1 つずつ明らかにしたときの変化の和に分解して証明します。
Y 0 : = E [ Z ] Y_0 := \mathbb{E}[Z] Y 0 := E [ Z ] とし、k = 1 , … , n k = 1, \dots, n k = 1 , … , n について
Y k : = E [ Z ∣ X 1 , … , X k ] Y_k := \mathbb{E}[Z \mid X_1, \dots, X_k] Y k := E [ Z ∣ X 1 , … , X k ] とおくと、Y n = Z Y_n = Z Y n = Z なので
Z − E [ Z ] = ∑ k = 1 n ( Y k − Y k − 1 ) Z - \mathbb{E}[Z] = \sum_{k=1}^n (Y_k - Y_{k-1}) Z − E [ Z ] = k = 1 ∑ n ( Y k − Y k − 1 ) と書けます。
X 1 , … , X k − 1 X_1, \dots, X_{k-1} X 1 , … , X k − 1 が与えられたもとで、増分 Y k − Y k − 1 Y_k - Y_{k-1} Y k − Y k − 1 の条件付き期待値は 0 0 0 です。
このような性質をもつ確率変数の列 ( Y k ) (Y_k) ( Y k ) を martingale と呼びます。
さらに、独立性と有界差分条件から、X 1 , … , X k − 1 X_1, \dots, X_{k-1} X 1 , … , X k − 1 を固定すると、Y k Y_k Y k は X k X_k X k の関数として長さ c k c_k c k 以下の区間に値をとります。
そこで、Hoeffding の補題を条件付きで各増分に適用し、k = n , n − 1 , … , 1 k = n, n-1, \dots, 1 k = n , n − 1 , … , 1 の順に条件付き期待値をとると、
E [ e λ ( Z − E [ Z ] ) ] ≤ exp ( λ 2 8 ∑ k = 1 n c k 2 ) \mathbb{E}\left[e^{\lambda (Z - \mathbb{E}[Z])}\right] \leq \exp\left(\frac{\lambda^2}{8} \sum_{k=1}^n c_k^2\right) E [ e λ ( Z − E [ Z ]) ] ≤ exp ( 8 λ 2 k = 1 ∑ n c k 2 ) を得ます。
つまり、Z Z Z は分散パラメータ ∑ k c k 2 / 4 \sum_k c_k^2/4 ∑ k c k 2 /4 の sub-Gaussian であり、第 5 節と同様に Chernoff 法を適用すると定理 6 が得られます。
8. 比較と使い分け
例の結果
第 3 節から第 6 節で求めた、Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ \Pr(|\hat{p} - p| \geq \varepsilon) \leq \delta Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ を保証する標本数を表 1 にまとめます。
比較のため、二項分布から数値計算した、実際に必要な標本数も示します。
Chebyshev と Bernstein の不等式には真の分散 p ( 1 − p ) p(1-p) p ( 1 − p ) を使っています。
実際には p p p は未知なので、分散の上界(例えば p ≤ 0.01 p \leq 0.01 p ≤ 0.01 が分かっていれば 0.0099 0.0099 0.0099 )を使います。
表 1 :Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ \Pr(|\hat{p} - p| \geq \varepsilon) \leq \delta Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ を保証する標本数 n n n の下限
不等式 使う情報 (A) δ = 0.05 \delta = 0.05 δ = 0.05 (A) δ = 10 − 6 \delta = 10^{-6} δ = 1 0 − 6 (B) δ = 0.05 \delta = 0.05 δ = 0.05 (B) δ = 10 − 6 \delta = 10^{-6} δ = 1 0 − 6 Chebyshev(第 3 節) 分散 2,000 100,000,000 7,920 396,000,000 相対誤差の Chernoff bound(第 4 節) Bernoulli 和の期待値 1,550 6,094 3,689 14,509 Hoeffding(第 5 節) 値域 738 2,902 73,778 290,174 Bernstein(第 6 節) 値域、分散 763 2,999 3,409 13,406 二項分布から計算した値 分布そのもの 約 400 約 2,400 約 1,600 約 10,400
最終行は、その値以上のすべての n n n で Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ \Pr(|\hat{p} - p| \geq \varepsilon) \leq \delta Pr ( ∣ p ^ − p ∣ ≥ ε ) ≤ δ となる最小の n n n です。
n ( p ± ε ) n(p \pm \varepsilon) n ( p ± ε ) が整数になる n n n では、∣ p ^ − p ∣ = ε |\hat{p} - p| = \varepsilon ∣ p ^ − p ∣ = ε となる境界上の値が事象に含まれるため、二項分布から計算した確率は n n n について単調には減少しません。
そのため、概数で示しています。
次のコードは、二項分布から計算した Pr ( ∣ p ^ − p ∣ ≥ ε ) \Pr(|\hat{p} - p| \geq \varepsilon) Pr ( ∣ p ^ − p ∣ ≥ ε ) と、各不等式による上界を n n n の関数として描きます。
import numpy as np
import matplotlib . pyplot as plt
from scipy . stats import binom
def exact_tail ( n , p , eps ) :
"""S ~ Bin(n, p) について Pr(|S/n - p| >= eps) を計算する"""
# 1e-9 は浮動小数点の丸め誤差対策
upper = np . ceil ( n * ( p + eps ) - 1e-9 )
lower = np . floor ( n * ( p - eps ) + 1e-9 )
return binom . sf ( upper - 1 , n , p ) + binom . cdf ( lower , n , p )
def bounds ( n , p , eps ) :
"""各不等式による Pr(|p_hat - p| >= eps) の上界"""
var , M , eta = p * ( 1 - p ) , max ( p , 1 - p ) , eps / p
return {
"Chebyshev" : var / ( n * eps ** 2 ) ,
"Chernoff (relative)" : 2 * np . exp ( - eta ** 2 * n * p / ( 2 + eta ) ) ,
"Hoeffding" : 2 * np . exp ( - 2 * n * eps ** 2 ) ,
"Bernstein" : 2 * np . exp ( - n * eps ** 2 / ( 2 * ( var + M * eps / 3 ) ) ) ,
}
settings = [ ( "A" , 0.5 , 0.05 , 4000 ) , ( "B" , 0.01 , 0.005 , 20000 ) ]
fig , axes = plt . subplots ( 1 , 2 , figsize = ( 14 , 5.6 ) , sharey = True )
for ax , ( label , p , eps , n_max ) in zip ( axes , settings ) :
n = np . arange ( 10 , n_max + 1 , 10 )
ax . semilogy ( n , exact_tail ( n , p , eps ) , "k" , lw = 1.2 , label = "Exact (binomial)" )
for name , value in bounds ( n , p , eps ) . items ( ) :
ax . semilogy ( n , np . minimum ( value , 1 ) , lw = 2.2 , label = name )
ax . axhline ( 0.05 , color = "gray" , ls = "--" )
ax . set ( title = rf"( { label } ) $p = { p } $, $\varepsilon = { eps } $" ,
xlabel = "$n$" , xlim = ( 0 , n_max ) , ylim = ( 1e-10 , 2 ) )
axes [ 0 ] . set_ylabel ( r"$\Pr(|\hat p - p| \geq \varepsilon)$" )
axes [ 1 ] . legend ( loc = "lower left" )
plt . show ( )
図 4:誤差確率 Pr(|p̂ − p| ≥ ε) の正確な値(黒)と、各不等式による上界。左が (A)、右が (B)。灰色の破線は δ = 0.05。縦軸は対数目盛。
表 1 と図 4 から、次の 3 点が読み取れます。
δ \delta δ を小さくしたときの増え方 :δ \delta δ を 0.05 0.05 0.05 から 10 − 6 10^{-6} 1 0 − 6 にすると、Chebyshev の不等式による n n n は 1 / δ 1/\delta 1/ δ に比例して 5 × 10 4 5 \times 10^4 5 × 1 0 4 倍になります。指数的な上界(Chernoff、Hoeffding、Bernstein)による n n n は log ( 2 / δ ) \log(2/\delta) log ( 2/ δ ) に比例し、約 3.9 3.9 3.9 倍にとどまります。図 4 では、この違いが直線(指数的な減衰)と、なだらかな曲線(1 / n 1/n 1/ n の減衰)の違いとして現れます。
分散が小さいとき :(B) では、分散を使う Bernstein の不等式と、期待値に対する相対誤差を使う Chernoff bound が、値域だけを使う Hoeffding の不等式より約 20 20 20 倍小さい n n n を与えます。δ = 0.05 \delta = 0.05 δ = 0.05 では、Chebyshev の不等式も Hoeffding の不等式より小さい n n n を与えます。
保守性 :どの上界も実際に必要な n n n より大きく、最も小さい上界でも (A) で約 1.2 1.2 1.2 〜1.8 1.8 1.8 倍、(B) で約 1.3 1.3 1.3 〜2.1 2.1 2.1 倍です。集中不等式は分布の形を細かく使わない代わりに、その分だけ保守的になります。
使い分け
表 2 に、各不等式の仮定と上界をまとめます。
表 2 :各不等式の仮定と上界
不等式 主な仮定 対象 上界 Markov 非負、期待値が有限 非負の確率変数 X X X E [ X ] / t \mathbb{E}[X]/t E [ X ] / t Chebyshev 分散が有限 一般の確率変数 X X X Var ( X ) / t 2 \operatorname{Var}(X)/t^2 Var ( X ) / t 2 Chernoff 法 MGF が有限 一般の確率変数 X X X exp ( − sup λ > 0 { λ t − ψ X ( λ ) } ) \exp(-\sup_{\lambda > 0}\{\lambda t - \psi_X(\lambda)\}) exp ( − sup λ > 0 { λ t − ψ X ( λ )}) Hoeffding 独立、有界 和 S n S_n S n exp ( − 2 t 2 / ∑ i ( b i − a i ) 2 ) \exp(-2t^2/\sum_i (b_i - a_i)^2) exp ( − 2 t 2 / ∑ i ( b i − a i ) 2 ) Bernstein 独立、有界、分散 和 S n S_n S n exp ( − t 2 / ( 2 ( V + M t / 3 ) ) ) \exp(-t^2/(2(V + Mt/3))) exp ( − t 2 / ( 2 ( V + M t /3 ))) McDiarmid 独立、有界差分条件 関数 f ( X 1 , … , X n ) f(X_1, \dots, X_n) f ( X 1 , … , X n ) exp ( − 2 t 2 / ∑ i c i 2 ) \exp(-2t^2/\sum_i c_i^2) exp ( − 2 t 2 / ∑ i c i 2 )
状況に応じた選び方の目安は次のとおりです。
非負性と期待値しか分からない → Markov の不等式
分散は分かるが、独立性を仮定できない → Chebyshev の不等式(無相関なら標本平均にも使えます)
MGF を計算できる → Chernoff 法で直接評価する
独立な Bernoulli 確率変数の和で、期待値に対する相対誤差を評価したい → 相対誤差の Chernoff bound
独立な有界確率変数の和で、分散の情報がない → Hoeffding の不等式
独立な有界確率変数の和で、分散が値域に比べて小さい → Bernstein の不等式
和ではない関数で、各入力の影響が小さい → McDiarmid の不等式
9. 仮定が満たされないとき
独立でない場合
Markov の不等式と Chebyshev の不等式は、独立性を必要としません。
一方、本記事で示した形の Hoeffding、Bernstein、McDiarmid の不等式は独立性を仮定しており、依存のある確率変数にはそのまま適用できません。
ただし例外もあり、Hoeffding (1963) は、有限母集団からの非復元抽出(各標本は独立ではない)でも、復元抽出の場合と同じ上界が成り立つことを示しています。
また、増分が有界な martingale には Azuma–Hoeffding の不等式が使えます(Azuma, 1967)。
第 7 節のコラムの証明は、この不等式と同じ martingale の議論に基づいています。
有界でない場合
有界でない確率変数も集中しえます。
例えば、正規分布は有界ではありませんが、第 4 節で見たように sub-Gaussian です。
sub-Gaussian を含み、より重い裾も許す確率変数のクラスとして sub-exponential があり、Bernstein 型の上界が成り立ちます(Vershynin, 2018, 第 2 章)。
MGF が有限でない分布でも、高次のモーメントが有限なら、第 2 節の方法で t t t の多項式で減少する上界が得られます。
10. まとめ
本記事で扱った不等式は、図 5 のように Markov の不等式から 2 つの経路で導かれます。
図 5:本記事で扱った不等式の関係。各箱の下段は、その不等式が使う主な仮定や情報。
Markov → Chebyshev :二乗偏差に Markov の不等式を適用し、分散で裾確率を評価します。
Markov → Chernoff 法 → Hoeffding、Bernstein、McDiarmid :指数関数に Markov の不等式を適用し、MGF を評価して指数的に減少する上界を得ます。McDiarmid の不等式は、martingale による分解によって、和以外の関数にも適用できるようにした結果です。
どの不等式も、有限個のデータから作った量が期待値からどれだけ外れうるか、という問いに答えます。
第 8 節の例が示すように、どの不等式が最も小さい上界を与えるかは、使える仮定(分散、値域、独立性、関数の安定性)と問題の条件によって変わります。
References
Azuma, K. (1967). Weighted sums of certain dependent random variables . Tohoku Mathematical Journal, 19(3), 357–367.
Bartlett, P. L., & Mendelson, S. (2002). Rademacher and Gaussian complexities: Risk bounds and structural results . Journal of Machine Learning Research, 3, 463–482.
Boucheron, S., Lugosi, G., & Massart, P. (2013). Concentration Inequalities: A Nonasymptotic Theory of Independence . Oxford University Press.
Chernoff, H. (1952). A measure of asymptotic efficiency for tests of a hypothesis based on the sum of observations . The Annals of Mathematical Statistics, 23(4), 493–507.
Hoeffding, W. (1963). Probability inequalities for sums of bounded random variables . Journal of the American Statistical Association, 58(301), 13–30.
McDiarmid, C. (1989). On the method of bounded differences . In Surveys in Combinatorics, 1989 (London Mathematical Society Lecture Note Series 141), 148–188. Cambridge University Press.
Mitzenmacher, M., & Upfal, E. (2017). Probability and Computing: Randomization and Probabilistic Techniques in Algorithms and Data Analysis (2nd ed.). Cambridge University Press.
Vershynin, R. (2018). High-Dimensional Probability: An Introduction with Applications in Data Science . Cambridge University Press.