1. はじめに：機械学習と確率分布推定の重要性
私たちは日々、膨大な量のデータに囲まれていますが、そのデータがどのように生成されているのか、その背後にはどのような法則が隠れているのでしょうか？
機械学習は、この問いに答えるための強力なツールです。
中でも、確率分布推定は機械学習の中心的な問題の一つです。データが未知の確率分布から生成されていると仮定し、その確率分布を推定することで、データの背後にある真理に迫ることができます。
この推定された確率分布を用いれば、新しいデータに対する予測も可能になります。
本稿では、確率分布推定の基礎となる概念を発見法的に議論し、その面白さと重要性を明らかにしていきます。
さらに、代表的な推定法である最尤法、MAP推定、ベイズ推定の理論的な導出過程を詳しく解説し、それぞれの方法の特徴や関連性を探ります。
また、汎化誤差と過学習の概念について説明し、モデル選択の重要性について議論します。
確率分布推定の理論は、機械学習の発展を支える基盤であり、その理解は機械学習の可能性を広げるために不可欠です。
読者の皆さんに確率分布推定と機械学習の関係を楽しんでいただければ幸いです。
2. 確率的学習の基礎：データの背後にある確率分布を探る
2.1 確率分布とデータ：真理を探る旅のはじまり
私たちが手にするデータは、ある未知の確率分布 $q(x)$ から生成されていると考えられます。つまり、観測されるデータ $X_1, \dots, X_n$ は、確率分布 $q(x)$ に従って独立に生成されていると仮定できます。
この仮定のもと、私たちの目標は、データ $X_1, \dots, X_n$ から、真の確率分布 $q(x)$ を推定することです。この推定された確率分布を用いれば、新しいデータに対する予測も可能になります。つまり、確率分布推定は、データの背後にある真理を探る旅の第一歩なのです。
2.2 教師あり学習と教師なし学習：真理を探る二つの道
確率分布推定は、教師あり学習と教師なし学習という二つの枠組みで行われます。
教師あり学習では、データとそれに対応するラベルのペア $(X_1, Y_1), \dots, (X_n, Y_n)$ が与えられます。この場合、私たちの目標は、未知の条件付き確率分布 $q(y|x)$ を推定することです。つまり、入力データ $x$ が与えられたとき、それに対応する出力 $y$ がどのような確率分布に従っているのかを明らかにするのです。
一方、教師なし学習では、データ $X_1, \dots, X_n$ のみが与えられます。この場合、私たちの目標は、データ全体の確率分布 $q(x)$ を推定することです。
教師あり学習と教師なし学習は、真理を探る二つの道です。どちらの道を選ぶかは、手元にあるデータの種類や、解決したい問題の性質によって決まります。
2.3 モデル化：真理を探るための地図を描く
データの背後にある真の確率分布 $q(x)$ を直接推定することは困難です。そこで、私たちは確率分布のモデル化を行います。つまり、パラメトリックな確率分布族 $p(x|\theta)$ を用意し、そのパラメータ $\theta$ を調整することで、真の分布 $q(x)$ に近い分布を探すのです。
このモデル化は、まるで真理を探るための地図を描くようなものです。私たちは、データという実世界の情報をもとに、確率分布という地図を作成します。この地図を頼りに、真の分布という目的地を目指すのです。
さらに、パラメータ $\theta$ の事前分布 $\phi(\theta)$ を導入することで、パラメータ空間に確率的な構造を与えることができます。この事前分布は、パラメータに関する私たちの先験的な知識や信念を表現するものです。
2.4 予測分布：真理を探る旅の終着点
モデル $p(x|\theta)$ とパラメータの事前分布 $\phi(\theta)$ が与えられたとき、私たちの目標は、データ $D_n = (X_1, \dots, X_n)$ から予測分布 $p(x|D_n)$ を推定することです。この予測分布は、新しいデータ $x$ に対する確率分布を表します。
予測分布 $p(x|D_n)$ を推定する方法には、大きく分けて以下の3つがあります。
- 最尤法（Maximum Likelihood Estimation）
- MAP推定（Maximum A Posteriori Estimation）
- ベイズ推定（Bayesian Estimation）
これらの推定法は、それぞれ異なる原理に基づいていますが、いずれも予測分布を推定するという目標に向かって、真理を探る旅を導いてくれます。
次章以降では、これらの推定法の理論的な導出過程を詳しく見ていきましょう。
3. 最尤法：真理を探る最も素直な方法
3.1 最尤法の原理：データからパラメータを推定する
最尤法は、与えられたデータのもとで、モデルの尤度を最大化するパラメータを推定する方法です。つまり、データが得られる確率を最大にするようなパラメータを選ぶのです。
モデル $p(x|\theta)$ とデータ $D_n = (X_1, \dots, X_n)$ が与えられたとき、尤度関数 $L(\theta)$ は以下のように定義されます。
$$ L(\theta) = \prod_{j=1}^n p(X_j|\theta) $$
最尤法では、この尤度関数を最大化するパラメータ $\hat{\theta}_{ML}$ を推定します。
\hat{\theta}_{ML} = \arg\max_{\theta} L(\theta)
この推定されたパラメータ $\hat{\theta}_{ML}$ を用いて、予測分布
$$ p(x|\hat{\theta}_{ML}) $$
を構築します。
3.2 最尤法の面白さ：直感的に真理に迫る
最尤法の面白さは、その直感的な解釈にあります。最尤法は、データが得られる確率を最大にするパラメータを選ぶという、非常にシンプルな原理に基づいています。
この原理は、私たちの日常的な経験とも合致します。例えば、複数の仮説が与えられたとき、私たちは観測されたデータが最も起こりやすい仮説を選ぶ傾向があります。最尤法は、この直感的な推論を数学的に定式化したものとも言えます。
3.3 最尤法の応用例：線形回帰への適用
最尤法の応用例として、線形回帰を見てみましょう。線形回帰では、データ $(X_1, Y_1), \dots, (X_n, Y_n)$ が以下のモデルから生成されていると仮定します。
$$ Y_i = \theta_0 + \theta_1 X_i + \epsilon_i $$
ここで、$\epsilon_i$ は平均0、分散 $\sigma^2$ の正規分布に従う誤差項です。つまり、以下のような条件付き確率分布を考えます。
$$ p(Y_i|X_i, \theta_0, \theta_1, \sigma^2) = \frac{1}{\sqrt{2\pi\sigma^2}} \exp\left(-\frac{(Y_i - \theta_0 - \theta_1 X_i)^2}{2\sigma^2}\right) $$
この設定のもと、尤度関数は以下のように書けます。
$$ L(\theta_0, \theta_1, \sigma^2) = \prod_{i=1}^n \frac{1}{\sqrt{2\pi\sigma^2}} \exp\left(-\frac{(Y_i - \theta_0 - \theta_1 X_i)^2}{2\sigma^2}\right) $$
この尤度関数を最大化するパラメータ $(\hat{\theta}_0, \hat{\theta}_1, \hat{\sigma}^2)$ を求めることが、線形回帰における最尤推定の目標です。
実際に計算を進めると、最尤推定量 $(\hat{\theta}_0, \hat{\theta}_1)$ は、以下の正規方程式の解として与えられることがわかります。
\begin{aligned}
\sum_{i=1}^n (Y_i - \hat{\theta}_0 - \hat{\theta}_1 X_i) &= 0 \\
\sum_{i=1}^n (Y_i - \hat{\theta}_0 - \hat{\theta}_1 X_i) X_i &= 0
\end{aligned}
この結果は、最小二乗法による推定量と一致します。つまり、線形回帰において、最尤法は最小二乗法と等価なのです。
この事実は、最尤法と最小二乗法という一見異なる原理が、実は深く結びついていることを示しています。このような洞察は、機械学習の理論を探求する醍醐味の一つと言えるでしょう。
4. MAP推定とベイズ推定：事前分布の導入による推定の改良
4.1 MAP推定の原理：事後確率を最大化する
最尤法は直感的で分かりやすい方法ですが、パラメータの事前分布を考慮していないという欠点があります。そこで、事前分布 $\phi(\theta)$ を導入し、データが得られた後のパラメータの事後確率を最大化するのが MAP 推定です。
ベイズの定理より、パラメータの事後確率は以下のように表せます。
$$ p(\theta|D_n) = \frac{p(D_n|\theta)\phi(\theta)}{p(D_n)} \propto L(\theta)\phi(\theta) $$
MAP推定では、この事後確率 $p(\theta|D_n)$ を最大化するパラメータ $\hat{\theta}_{MAP}$ を推定します。
\hat{\theta}_{MAP} = \arg\max_{\theta} p(\theta|D_n)
4.2 MAP推定の利点：事前分布による正則化効果
MAP推定の利点は、事前分布による正則化効果です。事前分布を適切に設定することで、パラメータの過学習を防ぐことができます。
例えば、ガウス分布を事前分布として使うと、パラメータの大きさにペナルティを与えることになります。これは、Ridge回帰における正則化項に対応します。
\hat{\theta}_{MAP} = \arg\max_{\theta} \left( \ln L(\theta) - \frac{\lambda}{2} ||\theta||^2 \right)
ここで、$\lambda$ は正則化の強さを制御するハイパーパラメータです。
このように、MAP推定は最尤法に事前分布による正則化を加えた方法と見ることができます。
4.3 ベイズ推定の原理：事後分布を用いた推論
MAP推定はパラメータの事後確率を最大化する点推定ですが、ベイズ推定ではパラメータの事後分布 $p(\theta|D_n)$ 全体を用いて推論を行います。
具体的には、予測分布 $p(x|D_n)$ を以下のように計算します。
$$ p(x|D_n) = \int p(x|\theta) p(\theta|D_n) d\theta $$
これは、パラメータの事後分布で重み付けされたモデルの期待値とみなすことができます。
ここで、事後分布 $p(\theta|D_n)$ は以下のように定義されます。
$$ p(\theta|D_n) = \frac{1}{Z_{D_n}(\beta)}\phi(\theta)L(\theta)^\beta $$
$\beta$ は逆温度と呼ばれるハイパーパラメータで、$Z_{D_n}(\beta)$ は正規化定数です。
興味深いことに、この事後分布は統計力学のカノニカル分布と類似の形をしています。
$$ p(\theta|D_n) = \frac{1}{Z_{D_n}(\beta)}e^{-\beta H_{D_n}(\theta)} $$
ここで、$H_{D_n}(\theta) = -\ln L(\theta)$ は、統計力学のハミルトニアンに対応します。
また、最尤法は、ベイズ推定の逆温度 $\beta$ を無限大にした極限と見なすことができます。
4.4 ベイズ推定の利点：不確実性の定量化
ベイズ推定の大きな利点は、予測の不確実性を定量化できることです。事後分布 $p(\theta|D_n)$ は、パラメータの不確実性を表現しています。この不確実性を予測分布 $p(x|D_n)$ に反映させることで、予測の信頼区間を計算することができます。
ベイズ推定は、パラメータの不確実性をモデルに取り込むための原理的な方法であり、機械学習における重要な概念の一つです。
5. 汎化誤差と交差検証：モデルの真の性能を評価する
5.1 汎化誤差と学習誤差の違い
機械学習の目的は、未知のデータに対して良い予測性能を持つモデルを学習することです。この予測性能を評価する指標が汎化誤差です。
汎化誤差は、学習したモデルが未知のデータに対してどの程度の予測誤差を持つかを表します。数学的には、以下のように定義されます。
$$ G_n = \mathbb{E}_{X,Y}[L(Y, f(X))] $$
ここで、$L(Y, f(X))$ は予測値 $f(X)$ と真の値 $Y$ の間の損失関数です。
一方、学習誤差は、与えられたデータに対するモデルの適合度を評価する指標です。学習誤差が小さいモデルが、必ずしも汎化誤差が小さいとは限りません。
5.2 過学習の問題点と汎化誤差の推定
学習誤差を最小化するようにモデルを学習すると、モデルがデータに過剰に適合し、未知のデータに対する予測性能が悪化する過学習が起こることがあります。
過学習を防ぐためには、モデルの複雑さを適切に制御することが重要です。ただし、真の分布が未知である場合、汎化誤差を直接計算することはできません。そこで、交差検証という手法を用いて、汎化誤差を推定します。
5.3 交差検証：汎化誤差を推定する
交差検証では、データを訓練データとテストデータに分割し、訓練データでモデルを学習し、テストデータで性能を評価します。この過程を複数回繰り返すことで、汎化誤差の推定値を得ることができます。
代表的な交差検証の手法として、k-fold交差検証があります。これは、データを k 個に分割し、k 回の交差検証を行う方法です。各回の交差検証では、k 個のうち 1 つをテストデータ、残りを訓練データとして使います。
5.4 モデル選択への応用：最良のモデルを選ぶ
交差検証は、複数のモデルの中から最良のモデルを選ぶためにも使われます。各モデルの汎化誤差を交差検証で推定し、その推定値が最も小さいモデルを選択します。
このようなモデル選択の手法は、機械学習の実践において非常に重要です。モデルの複雑さとデータへの適合度のバランスを取ることで、過学習を防ぎ、未知のデータに対する予測性能を高めることができます。
6. WAICとベイズモデル選択：情報量規準による汎化誤差の推定
6.1 WAICの定義と意味
WAIC（Widely Applicable Information Criterion）は、ベイズモデル選択のための情報量規準の一つです。WAICは、以下のように定義されます。
\mathrm{WAIC} = -2(\ln Z_{D_n}(1) - \mathbb{E}_{p(\theta|D_n)}[V_{D_n}(\theta)])
ここで、$Z_{D_n}(1)$ はベイズ推定の正規化定数、$V_{D_n}(\theta)$ はデータに対する対数尤度のバラつきです。
WAICは、ベイズモデルの汎化誤差を推定するための情報量規準として、近年注目を集めています。
6.2 WAICによるモデル選択の利点
WAICによるモデル選択の利点は、以下の通りです。
- ベイズ推定の枠組みで汎化誤差を推定できる
- 事後分布の不確実性を考慮できる
- 複雑なモデルでも適用可能
WAICは、ベイズ推定の枠組みで汎化誤差を推定するため、事後分布の不確実性を考慮することができます。また、モデルの複雑さに対するペナルティを自動的に導入するため、複雑なモデルにも適用可能です。
7. まとめと今後の展望
本稿では、機械学習における確率分布推定の基礎を発見法的に議論しました。最尤法、MAP推定、ベイズ推定という代表的な推定法の原理と利点を詳しく解説し、それぞれの方法が真理を探求する旅をどのように導いてくれるのかを見てきました。
また、汎化誤差と交差検証の概念を導入し、モデルの真の性能を評価する方法についても触れました。さらに、ベイズモデル選択における情報量規準の一つであるWAICを紹介し、その利点について議論しました。
機械学習の究極の目標は、データの背後にある真理を明らかにし、未知のデータに対する予測性能を高めることです。確率分布推定は、この目標に向かう旅の中で欠かせない羅針盤の役割を果たします。
機械学習の世界は広大で、未知のことがまだまだたくさんあります。新しい推定法の開発、深層学習などの高度な手法の探求、因果推論への応用など、課題は尽きません。
確率分布推定の理論は、これらの高度な技術の基礎となる重要な概念であり、その理解は機械学習全体の発展に寄与すると考えられます。