はじめに
『ベイズ推論による機械学習入門』の学習時のノートです。基本的な内容は「数式の行間を読んでみた」とそれを「RとPythonで組んでみた」になります。「数式」と「プログラム」から理解するのが目標です。
この記事は、項の内容です。「尤度関数を平均が未知の多次元ガウス分布(多変量正規分布)」、「事前分布を多次元ガウス分布」とした場合の「パラメータの事後分布」と「未観測値の予測分布」の計算をR言語で実装します。
省略してある内容等ありますので、本とあわせて読んでください。初学者な自分が理解できるレベルまで落として書き下していますので、分かる人にはかなりくどくなっています。同じような立場の人のお役に立てれば幸いです。
【数式読解編】
【他の節の内容】
【この節の内容】
・Rでやってみよう
人工的に生成したデータを用いて、ベイズ推論を行ってみましょう。
利用するパッケージを読み込みます。
# 3.4.1項で利用パッケージ library(tidyverse) library(mvnfast)
mvnfast
パッケージは、多次元ガウス分布に関するパッケージです。多次元ガウス分布に従う乱数生成関数rmvn()
と確率密度関数dmvn()
を使います。
・観測モデルの構築
まずは、観測モデルを設定します。この例では、尤度$p(\mathbf{X} | \boldsymbol{\mu})$をデータごとに独立した多次元ガウス分布$\mathcal{N}(\mathbf{x}_n | \boldsymbol{\mu}, \boldsymbol{\Lambda}^{-1})$とします。
尤度のパラメータを設定します。この実装例では2次元のグラフで表現するため、$D = 2$のときのみ動作します。
# 真の平均パラメータを指定 mu_truth_d <- c(25, 50) # (既知の)分散共分散行列を指定 sigma2_dd <- matrix(c(600, 90, 90, 400), nrow = 2, ncol = 2) # (既知の)精度行列を計算 lambda_dd <- solve(sigma2_dd) # 精度行列
平均パラメータ$\boldsymbol{\mu} = (\mu_1, \cdots, \mu_D)$をmu_truth_d
として値を指定します。この例では未知の値であり、この値を求めるのが目的です。
分散共分散行列$\boldsymbol{\Sigma} = (\sigma_{1,1}^2, \cdots, \sigma_{D,D}^2)$をsigma2_dd
として値を指定します。作図時に標準偏差に変換して利用します。matrix()
のデフォルトの仕様上、$(\sigma_{1,1}^2, \sigma_{2,1}^2, \sigma_{1,2}^2, \sigma_{2,2}^2)$の順番に値を指定します。$\sigma_{i,i}^2$は$i$次元方向の分散、$\sigma_{i,j}^2$は$i$次元と$j$次元方向の共分散です。
分散共分散行列の逆行列を精度行列と呼びます。sigma2_dd
から精度パラメータ(精度行列)$\boldsymbol{\Lambda} = \boldsymbol{\Sigma}^{-1}$を計算してlambda_dd
とします。逆行列はsolve()
で計算できます。
グラフ用の点を作成します。
# 作図用のxのx軸の値を作成 x_1_vec <- seq( mu_truth_d[1] - 3 * sqrt(sigma2_dd[1, 1]), mu_truth_d[1] + 3 * sqrt(sigma2_dd[1, 1]), length.out = 500 ) # 作図用のxのy軸の値を作成 x_2_vec <- seq( mu_truth_d[2] - 3 * sqrt(sigma2_dd[2, 2]), mu_truth_d[2] + 3 * sqrt(sigma2_dd[2, 2]), length.out = 500 ) # 作図用のxの点を作成 x_point_mat <- cbind( rep(x_1_vec, times = length(x_2_vec)), rep(x_2_vec, each = length(x_1_vec)) )
作図用に、2次元ガウス分布に従う変数$\mathbf{x}_n = (x_{n,1}, x_{n,2})$がとり得る点(2値の組み合わせ)を作成します。
$x_{n,1}$がとり得る値(x軸の値)をseq()
で作成してx_1_vec
とします。この例では、平均値を中心に標準偏差の3倍を範囲とします。length.out
引数を使うと指定した要素数で等間隔に切り分けます。by
引数を使うと切り分ける間隔を指定できます。処理が重い場合は、この値を調整してください。$i$軸の標準偏差$\sigma_{i,i}$は、分散の平方根$\sigma_{i,i} = \sqrt{\sigma_{i,i}^2}$です。平方根はsqrt()
で計算できます。同様に、$x_{n,2}$がとり得る値(y軸の値)も作成してx_2_vec
とします。
x_1_vec
とx_2_vec
の要素の全ての組み合わせを持つようにマトリクスを作成してx_point_mat
とします。これは、確率密度を等高線図にする際に格子状の点(2軸の全ての値が直交する点)を渡す必要があるためです。
作成した$\mathbf{x}$の点は次のようになります。
# 確認 x_point_mat[1:5, ]
## [,1] [,2] ## [1,] -48.48469 -10 ## [2,] -48.19016 -10 ## [3,] -47.89564 -10 ## [4,] -47.60111 -10 ## [5,] -47.30658 -10
1列目がx軸の値、2列目がy軸の値に対応しています。
尤度の確率密度を計算して、作図用のデータフレームを作成します。
# 尤度を計算:式(2.72) model_df <- tibble( x_1 = x_point_mat[, 1], x_2 = x_point_mat[, 2], density = mvnfast::dmvn( X = x_point_mat, mu = mu_truth_d, sigma = solve(lambda_dd) ) )
x_point_mat
の値(の組み合わせ)ごとに確率密度を計算します。多次元ガウス分布の確率密度は、mvnfast::dmvn()
で計算できます。データの引数X
にx_point_mat
、平均の引数mu
にmu_truth_d
、分散共分散行列の引数sigma
にlambda_dd
を逆行列$\boldsymbol{\Sigma} = \boldsymbol{\Lambda}^{-1}$に変換して指定します。勿論sigma2_dd
を指定すればいいのですが、ここでは式に合わせてlambda_dd
を使っています。
作成したデータフレームを確認しましょう。
# 確認 head(model_df)
## # A tibble: 6 x 3 ## x_1 x_2 density ## <dbl> <dbl> <dbl> ## 1 -48.5 -10 0.000000165 ## 2 -48.2 -10 0.000000170 ## 3 -47.9 -10 0.000000175 ## 4 -47.6 -10 0.000000181 ## 5 -47.3 -10 0.000000186 ## 6 -47.0 -10 0.000000192
ggplot2
パッケージを利用して作図するには、データフレームを渡す必要があります。
尤度を作図します。
# 尤度を作図 ggplot(model_df, aes(x = x_1, y = x_2, z = density, color = ..level..)) + geom_contour() + # 尤度 labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("mu=(", paste(round(mu_truth_d, 1), collapse = ", "), ")", ", lambda=(", paste(round(lambda_dd, 5), collapse = ", "), ")"), x = expression(x[1]), y = expression(x[2]), color = "density")
geom_contour()
で、等高線グラフを描画します。格子状の点を渡す必要があります。
真のパラメータを求めることは、この真の分布を求めることを意味します。
・データの生成
続いて、構築したモデルに従って観測データ$\mathbf{X} = \{\mathbf{x}_1, \mathbf{x}_2, \cdots, \mathbf{x}_N\}$を生成します。
多次元ガウス分布に従う$N$個のデータをランダムに生成します。
# (観測)データ数を指定 N <- 50 # 多次元ガウス分布に従うデータを生成 x_nd <- mvnfast::rmvn(n = N, mu = mu_truth_d, sigma = solve(lambda_dd))
生成するデータ数$N$をN
として値を指定します。
多次元ガウス分布に従う乱数は、mvnfast::rmvn()
で生成できます。試行回数の引数n
にN
を指定します。他の引数についてはmvnfast::dmvn()
と同じです。生成したN
個のデータをx_nd
とします。
観測したデータを確認しましょう。
# 確認 summary(x_nd)
## V1 V2 ## Min. :-12.67 Min. : 5.762 ## 1st Qu.: 10.06 1st Qu.:38.524 ## Median : 19.50 Median :50.496 ## Mean : 24.00 Mean :51.146 ## 3rd Qu.: 33.73 3rd Qu.:64.710 ## Max. : 80.78 Max. :96.189
観測データの散布図を尤度と重ねて確認します。
# 観測データのデータフレームを作成 x_df <- tibble( x_n1 = x_nd[, 1], x_n2 = x_nd[, 2] ) # 観測データの散布図を作図 ggplot() + geom_point(data = x_df, aes(x = x_n1, y = x_n2)) + # 観測データ geom_contour(data = model_df, aes(x = x_1, y = x_2, z = density, color = ..level..)) + # 尤度 labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("N=", N, ", mu=(", paste(mu_truth_d, collapse = ", "), ")", ", lambda=(", paste(round(lambda_dd, 5), collapse = ", "), ")"), x = expression(x[1]), y = expression(x[2]), color = "density")
作図用に観測データをデータフレームに格納しておき、確率密度のグラフと重ねて表示します。
・事前分布の設定
次に、尤度に対する共役事前分布を設定します。多次元ガウス分布の平均パラメータ$\boldsymbol{\mu}$に対する事前分布$p(\boldsymbol{\mu})$として、多次元ガウス分布$\mathcal{N}(\boldsymbol{\mu} | \mathbf{m}, \boldsymbol{\Lambda}_{\boldsymbol{\mu}}^{-1})$を設定します。
$\boldsymbol{\mu}$の事前分布のパラメータ(超パラメータ)を設定します。
# muの事前分布の平均パラメータを指定 m_d <- c(0, 0) # muの事前分布の分散共分散行列を指定 sigma2_mu_dd <- matrix(c(1000, 0, 0, 1000), nrow = 2, ncol = 2) # muの事前分布の精度行列を計算 lambda_mu_dd <- solve(sigma2_mu_dd)
尤度のときと同様に、多次元ガウス分布の平均パラメータ$\mathbf{m} = (m_1, \cdots, m_D)$をm_d
、分散共分散行列$\boldsymbol{\Sigma}_{\boldsymbol{\mu}} = (\sigma_{(\mu),1,1}^2, \cdots, \sigma_{(\mu),D,D}^2)$をsigma2_mu_dd
として値を指定します。また、sigma2_mu_dd
の逆行列を精度行列lambda_mu_dd
とします。
$\boldsymbol{\mu}$の事前分布もグラフで確認しましょう。作図用の$\boldsymbol{\mu}$の点を作成します。
# 作図用のmuのx軸の値を作成 mu_1_vec <- seq( mu_truth_d[1] - 2 * sqrt(sigma2_mu_dd[1, 1]), mu_truth_d[1] + 2 * sqrt(sigma2_mu_dd[1, 1]), length.out = 500 ) # 作図用のmuのy軸の値を作成 mu_2_vec <- seq( mu_truth_d[2] - 2 * sqrt(sigma2_mu_dd[2, 2]), mu_truth_d[2] + 2 * sqrt(sigma2_mu_dd[2, 2]), length.out = 500 ) # 作図用のmuの点を作成 mu_point_mat <- cbind( rep(mu_1_vec, times = length(mu_2_vec)), rep(mu_2_vec, each = length(mu_1_vec)) )
$\mathbf{x}$の点のときと同様に、2次元ガウス分布に従う変数$\boldsymbol{\mu} = (\mu_1, \mu_2)$がとり得る値を作成してmu_point_mat
とします。この例では、真の値を中心に$\boldsymbol{\mu}$の事前分布の標準偏差の2倍を範囲とします。
$\boldsymbol{\mu}$の事前分布の確率密度を計算します。
# muの事前分布を計算:式(2.72) prior_df <- tibble( mu_1 = mu_point_mat[, 1], mu_2 = mu_point_mat[, 2], density = mvnfast::dmvn( X = mu_point_mat, mu = m_d, sigma = solve(lambda_mu_dd) ) )
尤度のときと同様にして、mu_point_mat
の値ごとに確率密度を計算します。
計算結果は次のようになります。
# 確認 head(prior_df)
## # A tibble: 6 x 3 ## mu_1 mu_2 density ## <dbl> <dbl> <dbl> ## 1 -38.2 -13.2 0.0000702 ## 2 -38.0 -13.2 0.0000708 ## 3 -37.7 -13.2 0.0000715 ## 4 -37.5 -13.2 0.0000722 ## 5 -37.2 -13.2 0.0000729 ## 6 -37.0 -13.2 0.0000736
$\boldsymbol{\mu}$の事前分布を作図します。
# 作図用に真のmuのデータフレームを作成 mu_df <- tibble( mu_1 = mu_truth_d[1], mu_2 = mu_truth_d[2] ) # muの事前分布を作図 ggplot() + geom_contour(data = prior_df, aes(x = mu_1, y = mu_2, z = density, color = ..level..)) + # muの事前分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("m=(", paste(round(m_d, 1), collapse = ", "), ")", ", lambda_mu=(", paste(round(lambda_mu_dd, 5), collapse = ", "), ")"), x = expression(mu[1]), y = expression(mu[2]), color = "density")
$\boldsymbol{\mu}$の真の値をデータフレームに格納しておき、確率密度のグラフと重ねて表示します。
・事後分布の計算
観測データ$\mathbf{X}$から平均パラメータ$\boldsymbol{\mu}$の事後分布$p(\boldsymbol{\mu} | \mathbf{X})$を求めます(平均パラメータ$\boldsymbol{\mu}$を分布推定します)。事後分布は多次元ガウス分布$\mathcal{N}(\boldsymbol{\mu} | \hat{\mathbf{m}}, \hat{\boldsymbol{\Lambda}}_{\boldsymbol{\mu}}^{-1})$になります。
観測データx_nd
を用いて、$\boldsymbol{\mu}$の事後分布のパラメータを計算します。
# muの事後分布のパラメータを計算:式(3.102),(3.103) lambda_mu_hat_dd <- N * lambda_dd + lambda_mu_dd m_hat_d <- solve(lambda_mu_hat_dd) %*% (lambda_dd %*% colSums(x_nd) + lambda_mu_dd %*% m_d) %>% as.vector()
$\boldsymbol{\mu}$の事後分布のパラメータを
で計算して、結果をlambda_mu_hat_dd
、m_hat_d
とします。
# 確認
m_hat_d
lambda_mu_hat_dd
## [1] 23.62355 50.69758 ## [,1] [,2] ## [1,] 0.08724407 -0.01940492 ## [2,] -0.01940492 0.13036611
求めたパラメータを用いて、$\boldsymbol{\mu}$の事後分布の確率密度を計算します。
# muの事後分布を計算:式(2.72) posterior_df <- tibble( mu_1 = mu_point_mat[, 1], mu_2 = mu_point_mat[, 2], density = mvnfast::dmvn( X = , mu_point_mat, mu = m_hat_d, sigma = solve(lambda_mu_hat_dd) ) )
更新した超パラメータm_hat_d, lambda_mu_hat_dd
を用いて、事前分布のときと同様にして計算します。
計算結果は次のようになります。
# 確認 head(posterior_df)
## # A tibble: 6 x 3 ## mu_1 mu_2 density ## <dbl> <dbl> <dbl> ## 1 -38.2 -13.2 1.99e-157 ## 2 -38.0 -13.2 5.70e-157 ## 3 -37.7 -13.2 1.62e-156 ## 4 -37.5 -13.2 4.59e-156 ## 5 -37.2 -13.2 1.29e-155 ## 6 -37.0 -13.2 3.61e-155
$\boldsymbol{\mu}$の事後分布を作図します。
# muの事後分布を作図 ggplot() + geom_contour(data = posterior_df, aes(x = mu_1, y = mu_2, z = density, color = ..level..)) + # muの事後分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("N=", N, ", m_hat=(", paste(round(m_hat_d, 1), collapse = ", "), ")", ", lambda_mu_hat=(", paste(round(lambda_mu_hat_dd, 5), collapse = ", "), ")"), x = expression(mu[1]), y = expression(mu[2]), color = "density")
$\boldsymbol{\mu}$の真の値付近をピークとする分布を推定できています。
x軸とy軸の値に注目すると、事前分布と比べて範囲が狭くなっています。グラフの表示範囲を広げてみましょう。
# muの事後分布を作図 ggplot() + geom_contour(data = posterior_df, aes(x = mu_1, y = mu_2, z = density, color = ..level..)) + # muの事後分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu xlim(c(min(mu_point_mat[, 1]), max(mu_point_mat[, 1]))) + # x軸の表示範囲 ylim(c(min(mu_point_mat[, 2]), max(mu_point_mat[, 2]))) + # y軸の表示範囲 labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("N=", N, ", m_hat=(", paste(round(m_hat_d, 1), collapse = ", "), ")", ", lambda_mu_hat=(", paste(round(lambda_mu_hat_dd, 5), collapse = ", "), ")"), x = expression(mu[1]), y = expression(mu[2]), color = "density")
先のとがった分布になっているのが分かります。
・予測分布の計算
最後に、観測データ$\mathbf{X}$から未観測のデータ$\mathbf{x}_{*}$の予測分布$p(\mathbf{x}_{*} | \mathbf{X})$を求めます。予測分布は多次元ガウス分布$\mathcal{N}(\mathbf{x}_{*} | \hat{\boldsymbol{\mu}}_{*}, \hat{\boldsymbol{\Lambda}}_{*}^{-1})$になります。
$\boldsymbol{\mu}$の事後分布のパラメータを用いて、予測分布のパラメータを計算します。
# 予測分布のパラメータを計算:式(3.109'),(3.110') lambda_star_hat_dd <- solve(solve(lambda_dd) + solve(lambda_mu_hat_dd)) mu_star_hat_d <- m_hat_d
予測分布のパラメータを
で計算して、結果をlambda_star_hat_dd
、m_star_hat_d
とします。
# 確認
mu_star_hat_d
lambda_star_hat_dd
## [1] 23.62355 50.69758 ## [,1] [,2] ## [1,] 0.0016914402 -0.0003804892 ## [2,] -0.0003804892 0.0025369718
$\mathbf{X}$から$\hat{\boldsymbol{\mu}}_{*},\ \hat{\boldsymbol{\Lambda}}_{*}$を学習しているのが式からも分かります。
求めたパラメータを用いて、予測分布の確率密度を計算します。
# 予測分布を計算:式(2.72) predict_df <- tibble( x_1 = x_point_mat[, 1], x_2 = x_point_mat[, 2], density = mvnfast::dmvn( X = x_point_mat, mu = mu_star_hat_d, sigma = solve(lambda_star_hat_dd) ) )
尤度のときと同様に、x_point_mat
の値ごとに確率密度を計算します。
計算結果は次のようになります。
# 確認 head(predict_df)
## # A tibble: 6 x 3 ## x_1 x_2 density ## <dbl> <dbl> <dbl> ## 1 -48.5 -10 0.000000197 ## 2 -48.2 -10 0.000000203 ## 3 -47.9 -10 0.000000209 ## 4 -47.6 -10 0.000000215 ## 5 -47.3 -10 0.000000221 ## 6 -47.0 -10 0.000000227
予測分布を尤度と重ねて作図します。
# 予測分布を作図 ggplot() + geom_contour(data = predict_df, aes(x = x_1, y = x_2, z = density, color = ..level..)) + # 予測分布 geom_contour(data = model_df, aes(x = x_1, y = x_2, z = density, color = ..level..), alpha = 0.5, linetype = "dashed") + # 真の分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu geom_point(data = x_df, aes(x = x_n1, y = x_n2)) + # 観測データ labs(title = "Multivariate Gaussian Distribution", subtitle = paste0("N=", N, ", mu_star_hat=(", paste(round(mu_star_hat_d, 1), collapse = ", "), ")", ", lambda_star_hat=(", paste(round(lambda_star_hat_dd, 5), collapse = ", "), ")"), x = expression(x[1]), y = expression(x[2]), color = "density")
観測データが増えると、予測分布が真の分布に近づきます。
・おまけ:アニメーションで推移の確認
gganimate
パッケージを利用して、事後分布と予測分布の推移をアニメーション(gif画像)で確認するためのコードです。
・コード(クリックで展開)
異なる点のみを簡単に解説します。
・ライブラリの読み込み
# 3.4.1項で利用パッケージ library(tidyverse) library(mvnfast) library(gganimate)
・モデルの設定
# 真の平均パラメータを指定 mu_truth_d <- c(25, 50) # (既知の)分散共分散行列を指定 sigma2_dd <- matrix(c(600, 90, 90, 400), nrow = 2, ncol = 2) # (既知の)精度行列を計算 lambda_dd <- solve(sigma2_dd) # 精度行列 # muの事前分布の平均パラメータを指定 m_d <- c(0, 0) # muの事前分布の分散共分散行列を指定 sigma2_mu_dd <- matrix(c(1000, 0, 0, 1000), nrow = 2, ncol = 2) # muの事前分布の精度行列を計算 lambda_mu_dd <- solve(sigma2_mu_dd) # 初期値による予測分布のパラメータを計算:式(3.109),(3.110) lambda_star_dd <- solve(solve(lambda_dd) + solve(lambda_mu_dd)) mu_star_d <- m_d
事前分布のパラメータを用いて、予測分布のパラメータを計算しておきます。
・作図用のデータフレームの初期化
# 作図用のmuのx軸の値を作成 mu_1_vec <- seq( mu_truth_d[1] - 1 * sqrt(sigma2_mu_dd[1, 1]), mu_truth_d[1] + 1 * sqrt(sigma2_mu_dd[1, 1]), length.out = 250 ) # 作図用のmuのy軸の値を作成 mu_2_vec <- seq( mu_truth_d[2] - 1 * sqrt(sigma2_mu_dd[2, 2]), mu_truth_d[2] + 1 * sqrt(sigma2_mu_dd[2, 2]), length.out = 250 ) # 作図用のmuの点を作成 mu_point_mat <- cbind( rep(mu_1_vec, times = length(mu_2_vec)), rep(mu_2_vec, each = length(mu_1_vec)) ) # muの事前分布を計算:式(2.72) posterior_df <- tibble( mu_1 = mu_point_mat[, 1], mu_2 = mu_point_mat[, 2], density = mvnfast::dmvn( X = mu_point_mat, mu = m_d, sigma = solve(lambda_mu_dd) ), label = as.factor( paste0( "N=", 0, ", m=(", paste(round(m_d, 1), collapse = ", "), ")", ", lambda_mu=(", paste(round(lambda_mu_dd, 5), collapse = ", "), ")" ) ) ) # 作図用のxのx軸の値を作成 x_1_vec <- seq( mu_truth_d[1] - 3 * sqrt(sigma2_dd[1, 1]), mu_truth_d[1] + 3 * sqrt(sigma2_dd[1, 1]), length.out = 250 ) # 作図用のxのy軸の値を作成 x_2_vec <- seq( mu_truth_d[2] - 3 * sqrt(sigma2_dd[2, 2]), mu_truth_d[2] + 3 * sqrt(sigma2_dd[2, 2]), length.out = 250 ) # 作図用のxの点を作成 x_point_mat <- cbind( rep(x_1_vec, times = length(x_2_vec)), rep(x_2_vec, each = length(x_1_vec)) ) # 予測分布を計算:式(2.72) predict_df <- tibble( x_1 = x_point_mat[, 1], x_2 = x_point_mat[, 2], density = mvnfast::dmvn( X = x_point_mat, mu = mu_star_d, sigma = solve(lambda_star_dd) ), label = as.factor( paste0( "N=", 0, ", mu_star=(", paste(round(mu_star_d, 1), collapse = ", "), ")", ", lambda_star=(", paste(round(lambda_star_dd, 5), collapse = ", "), ")" ) ) )
各試行の結果を同じデータフレームに格納していく必要があります。事後分布をposterior_df
、予測分布をpredict_df
として、初期値の結果を持つように作成しておきます。分布の推移と共にパラメータの値を表示するために、label
列にパラメータの値を記録しておきます。ややこしければas.factor(paste0("iter=", n))
として試行回数だけ表示するだけでもそれっぽくなります。
作成したデータフレームは次のようになります。
# 確認 head(posterior_df) head(predict_df)
## # A tibble: 6 x 4 ## mu_1 mu_2 density label ## <dbl> <dbl> <dbl> <fct> ## 1 -6.62 18.4 0.000132 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## 2 -6.37 18.4 0.000132 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## 3 -6.11 18.4 0.000132 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## 4 -5.86 18.4 0.000132 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## 5 -5.61 18.4 0.000132 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## 6 -5.35 18.4 0.000133 N=0, m=(0, 0), lambda_mu=(0.001, 0, 0, 0.001) ## # A tibble: 6 x 4 ## x_1 x_2 density label ## <dbl> <dbl> <dbl> <fct> ## 1 -48.5 -10 0.0000501 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~ ## 2 -47.9 -10 0.0000510 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~ ## 3 -47.3 -10 0.0000519 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~ ## 4 -46.7 -10 0.0000528 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~ ## 5 -46.1 -10 0.0000537 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~ ## 6 -45.5 -10 0.0000546 N=0, mu_star=(0, 0), lambda_star=(0.00063, -4e-05, -4e-~
・推論処理
# データ数(試行回数)を指定 N <- 100 # 観測データの受け皿を初期化 x_nd <- matrix(0, nrow = N, ncol = 2) # ベイズ推論 for(n in 1:N) { # 多次元ガウス分布に従うデータを生成 x_nd[n, ] <- mvnfast::rmvn(n = 1, mu = mu_truth_d, sigma = solve(lambda_dd)) %>% as.vector() # muの事後分布のパラメータを更新:式(3.102),(3.103) old_lambda_mu_dd <- lambda_mu_dd lambda_mu_dd <- lambda_dd + lambda_mu_dd m_d <- solve(lambda_mu_dd) %*% (lambda_dd %*% x_nd[n, ] + old_lambda_mu_dd %*% m_d) %>% as.vector() # muの事後分布を計算:式(2.72) tmp_posterior_df <- tibble( mu_1 = mu_point_mat[, 1], mu_2 = mu_point_mat[, 2], density = mvnfast::dmvn( X = mu_point_mat, mu = m_d, sigma = solve(lambda_mu_dd) ), label = as.factor( paste0( "N=", n, ", m_hat=(", paste(round(m_d, 1), collapse = ", "), ")", ", lambda_mu_hat=(", paste(round(lambda_mu_dd, 5), collapse = ", "), ")" ) ) ) # 予測分布のパラメータを更新:式(3.109),(3.110) lambda_star_dd <- solve(solve(lambda_dd) + solve(lambda_mu_dd)) mu_star_d <- m_d # 予測分布を計算:式(2.72) tmp_predict_df <- tibble( x_1 = x_point_mat[, 1], x_2 = x_point_mat[, 2], density = mvnfast::dmvn( X = x_point_mat, mu = mu_star_d, sigma = solve(lambda_star_dd) ), label = as.factor( paste0( "N=", n, ", mu_star_hat=(", paste(round(mu_star_d, 1), collapse = ", "), ")", ", lambda_star_hat=(", paste(round(lambda_star_dd, 5), collapse = ", "), ")" ) ) ) # 推論結果を結合 posterior_df <- rbind(posterior_df, tmp_posterior_df) predict_df <- rbind(predict_df, tmp_predict_df) # 動作確認 #print(paste0("n=", n, " (", round(n / N * 100, 1), "%)")) }
観測された各データによってどのように学習する(分布が変化する)のかを確認するため、for()
で1データずつ処理します。よって、データ数N
がイタレーション数になります。
一度の処理で事後分布のパラメータを計算するのではなく、事前分布(1ステップ前の事後分布)に対して繰り返し観測データの情報を与えることでパラメータを更新(上書き)していきます。
それに伴い、事後分布のパラメータの計算式(3.102-103)の$N$と$\sum_{n=1}^N \mathbf{x}_n$の計算は、ループ処理によって$N$回繰り返し1
とx_nd[n, ]
を加えることで行います。$n$回目のループ処理のときには、$n-1$回分の1
とx_nd[n, ]
が既にm_d
とlambda_mu_dd
に加えられているわけです。
ただし、事後分布のパラメータの計算において、更新前(1ステップ前)のパラメータを使うため、old_lambda_mu_dd
として値を一時的に保存しておきます。
・事後分布の推移
# 作図用に真のmuのデータフレームを作成 mu_df <- tibble( mu_1 = mu_truth_d[1], mu_2 = mu_truth_d[2] ) # muの事後分布を作図 posterior_graph <- ggplot() + geom_contour(data = posterior_df, aes(x = mu_1, y = mu_2, z = density, color = ..level..)) + # muの事後分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu gganimate::transition_manual(label) + # フレーム labs(title = "Multivariate Gaussian Distribution", subtitle = "{current_frame}", x = expression(mu[1]), y = expression(mu[2]), color = "density") # gif画像を作成 gganimate::animate(posterior_graph, nframes = N + 1, fps = 10)
各フレームの順番を示す列(label
)をgganimate::transition_manual()
に指定します。
初期値(事前分布)を含むため、フレーム数の引数nframes
はN + 1
です。
・予測分布の推移
# 尤度を計算:式(2.72) model_df <- tibble( x_1 = x_point_mat[, 1], x_2 = x_point_mat[, 2], density = mvnfast::dmvn( X = x_point_mat, mu = mu_truth_d, sigma = solve(lambda_dd) ) ) # 作図用に観測データのデータフレームを作成 label_vec <- unique(predict_df[["label"]]) # ラベルを抽出 x_df <- tibble(x_n1 = NA, x_n2 = NA, label = label_vec[1]) # 初期値用 for(n in 1:N) { # n個目までのデータフレームを作成 tmp_x_df <- tibble( x_n1 = x_nd[1:n, 1], x_n2 = x_nd[1:n, 2], label = label_vec[n + 1] ) # 結合 x_df <- rbind(x_df, tmp_x_df) } # 予測分布を作図 predict_graph <- ggplot() + geom_contour(data = predict_df, aes(x = x_1, y = x_2, z = density, color = ..level..)) + # 予測分布 geom_point(data = x_df, aes(x = x_n1, y = x_n2)) + # 観測データ geom_contour(data = model_df, aes(x = x_1, y = x_2, z = density, color = ..level..), alpha = 0.5, linetype = "dashed") + # 真の分布 geom_point(data = mu_df, aes(x = mu_1, y = mu_2), color = "red", shape = 4, size = 5) + # 真のmu gganimate::transition_manual(label) + # フレーム labs(title = "Multivariate Gaussian Distribution", subtitle = "{current_frame}", x = expression(x[1]), y = expression(x[2]), color = "density") # gif画像を作成 gganimate::animate(predict_graph, nframes = N + 1, fps = 10)
n
番目のフレームのグラフにおいて、n
回目までに生成(観測)されたデータ$\mathbf{x}_1, \cdots, \mathbf{x}_n$を表示するために、N
フレーム分までを順番に取り出してx_df
に格納していきます。このときlabel
列がpredict_df
と一致している必要があります。
(始めの方は精度が低い(分散が大きい)ため確率密度が低く等高線が表示されていません。)
新たなデータによって平均(分布の中心)が推移しているのを確認できます。
参考文献
- 須山敦志『ベイズ推論による機械学習入門』(機械学習スタートアップシリーズ)杉山将監修,講談社,2017年.
おわりに
その時々では理解したつもりで記事にしているのですが、勉強を続けていると勘違いしていたことに気付くことも多々あります。適宜修正していきたいですけど結構大変。独学あるある?
【次節の内容】
- 2021/04/10:加筆修正しました。その際に数式読解編と記事を分割しました。