🕌

多変量正規分布の積率母関数

2024/04/18に公開

現代数理統計学の多変量正規分布の積率母関数を求める式変形についての仕組みがわからなかったので、行間を埋めた。


積率母関数の定義

n次元の確率ベクトルをY = (Y_1, \dots, Y_n)^\mathsf{T}とするとき、その積率母関数は以下で定義される。ただし、\theta = (\theta_1, \dots, \theta_n)^\mathsf{T}とする。

\gdef\transpose#1{#1^{\mathsf{T}}} \phi(\theta) = \mathrm{E}(e^{\transpose{\theta}Y})

多変量正規分布の確率密度関数

多変量正規分布の確率密度関数は以下で定義される。

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} f(y) = \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ -\frac{1}{2} \qform{(y-\mu)}{\Sigma^{-1}}{(y-\mu)} \right]}

ここで、\mu=(\mu_1, \dots, \mu_n)^\mathsf{T}Yの平均ベクトル、\Sigmaは分散共分散行列を表す。
また、この分布を\mathrm{N}_n(\mu, \Sigma)と表す。

多変量正規分布の積率母関数

多変量正規分布の積率母関数を定義に従って求める。

\gdef\t#1{#1^{\mathsf{T}}} \gdef\nint#1{ \overbrace{ \int\dots\int }^n #1 dy_1 \cdots dy_n } \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} \phi(\theta) &= \mathrm{E}(e^{\t{\theta}Y}) \\ &= \nint{ e^{\t{\theta}y} \cdot \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ -\frac{1}{2} \qform{(y-\mu)}{\Sigma^{-1}}{(y-\mu)} \right]} } \end{align*}

指数関数を指数法則に従ってまとめると

\gdef\t#1{#1^{\mathsf{T}}} \gdef\nint#1{ \overbrace{ \int\dots\int }^n #1 dy_1 \cdots dy_n } \gdef\qform#1#2#3{\t{#1}#2#3} = \nint{ \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ \t{\theta} y -\frac{1}{2} \qform{(y-\mu)}{\Sigma^{-1}}{(y-\mu)} \right]} }

指数関数の中身の計算

準備として以下を展開する。

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} \qform{(y-\mu)}{\Sigma^{-1}}{(y-\mu)} &= {(\t{y}-\t{\mu})}{\Sigma^{-1}}{(y-\mu)} \\ &= \qform{y}{\Sigma^{-1}}{y} - \qform{y}{\Sigma^{-1}}{\mu} - \qform{\mu}{\Sigma^{-1}}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \\ &= \qform{y}{\Sigma^{-1}}{y} - 2 \qform{\mu}{\Sigma^{-1}}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \end{align*}

よって

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} \t{\theta} y -\frac{1}{2} \t{(y-\mu)} \Sigma^{-1} (y-\mu) &= -\frac{1}{2} \left[ \qform{y}{\Sigma^{-1}}{y} - 2 \qform{\mu}{\Sigma^{-1}}{y} -2\qform{\theta}{}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \right] \end{align*}

ここで括弧内を平方完成することを考える。

平方完成の準備のためにyの項をまとめる。

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} & \qform{y}{\Sigma^{-1}}{y} - 2 \qform{\mu}{\Sigma^{-1}}{y} - 2\qform{\theta}{}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \\ =& \qform{y}{\Sigma^{-1}}{y} + 2(- \qform{\mu}{\Sigma^{-1}}{} - \qform{\theta}{}{})y + \qform{\mu}{\Sigma^{-1}}{\mu} \\ \end{align*}

平方完成で目指す形は以下のようになる。(1)

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \qform{y}{\Sigma^{-1}}{y} + 2(- \qform{\mu}{\Sigma^{-1}}{} - \qform{\theta}{}{})y + \qform{\mu}{\Sigma^{-1}}{\mu} = \qform{\left(y + \bigcirc \right)}{\Sigma^{-1}}{\left(y + \bigcirc \right)} + \square

ここで、右辺を展開すると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} & \qform{\left(y - \bigcirc \right)}{\Sigma^{-1}}{\left(y - \bigcirc \right)} + \square \\ =& \qform{y}{\Sigma^{-1}}{y} + \qform{\bigcirc}{\Sigma^{-1}}{y} + \qform{y}{\Sigma^{-1}}{\bigcirc} + \qform{\bigcirc}{\Sigma^{-1}}{\bigcirc} + \square \\ =& \qform{y}{\Sigma^{-1}}{y} + 2 \qform{\bigcirc}{\Sigma^{-1}}{y} + \qform{\bigcirc}{\Sigma^{-1}}{\bigcirc} + \square \\ \end{align*}

となる。

よって(1)のyの1次の係数を比較すると

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} 2\qform{\bigcirc}{\Sigma^{-1}}{} &= 2(-\qform{\mu}{\Sigma^{-1}}{} - \qform{\theta}{}{}) \\ \qform{\bigcirc}{}{} &= -\qform{\mu}{}{} - \qform{\theta}{\Sigma}{} \\ \bigcirc &= -\mu - {\Sigma}{\theta} \\ \end{align*}

yの0次の項を比較すると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \gdef\msm{\qform{\mu}{\Sigma^{-1}}{\mu}} \begin{align*} \square + \qform{\bigcirc}{\Sigma^{-1}}{\bigcirc} &= \msm \\ \square &= - \qform{\bigcirc}{\Sigma^{-1}}{\bigcirc} &+ \msm \\ \square &= - \qform{(-\mu -\Sigma\theta)}{\Sigma^{-1}}{(-\mu - \Sigma\theta)} &+ \msm \\ \square &= - \qform{(-\mu -\Sigma\theta)}{}{(-\Sigma^{-1}\mu - \theta)} &+ \msm \\ \square &= - (-\t{\mu} -\t{\theta}\Sigma){(-\Sigma^{-1}\mu - \theta)} &+ \msm \\ \square &= - (+\t{\mu} +\t{\theta}\Sigma){(+\Sigma^{-1}\mu + \theta)} &+ \msm \\ \square &= - (\msm + \qform{\mu}{}{\theta} + \qform{\theta}{}{\mu} + \qform{\theta}{\Sigma}{\theta} ) &+ \msm \\ \square &= - \qform{\mu}{}{\theta} - \qform{\theta}{}{\mu} - \qform{\theta}{\Sigma}{\theta} \\ \square &= -2 \qform{\theta}{}{\mu} - \qform{\theta}{\Sigma}{\theta} \end{align*}

以上をまとめると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} & \qform{y}{\Sigma^{-1}}{y} - 2 \qform{\mu}{\Sigma^{-1}}{y} - 2\qform{\theta}{}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \\ =& \qform{\left(y - (\mu + \Sigma\theta) \right)}{\Sigma^{-1}}{\left(y - (\mu +\Sigma\theta) \right)} -2 \qform{\theta}{}{\mu} - \qform{\theta}{\Sigma}{\theta} \end{align*}

以上の結果より、指数関数の中身を改めて計算すると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} & \t{\theta} y -\frac{1}{2} \t{(y-\mu)} \Sigma^{-1} (y-\mu) \\ =& -\frac{1}{2} \left[ \qform{y}{\Sigma^{-1}}{y} - 2 \qform{\mu}{\Sigma^{-1}}{y} -2\qform{\theta}{}{y} + \qform{\mu}{\Sigma^{-1}}{\mu} \right] \\ =& -\frac{1}{2} \left[ \qform{\left(y - (\mu + \Sigma\theta) \right)}{\Sigma^{-1}}{\left(y - (\mu +\Sigma\theta) \right)} -2 \qform{\theta}{}{\mu} - \qform{\theta}{\Sigma}{\theta} \right] \\ =& -\frac{1}{2} \left[ \qform{\left(y - (\mu + \Sigma\theta) \right)}{\Sigma^{-1}}{\left(y - (\mu +\Sigma\theta) \right)} \right] + \qform{\theta}{}{\mu} + \frac{\qform{\theta}{\Sigma}{\theta}}{2} \end{align*}

となる。

積分に戻る

以上の計算をもとの積分に代入する。

\gdef\nint#1{ \overbrace{ \int\dots\int }^n #1 dy_1 \cdots dy_n } \gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} \phi(\theta) = & \nint{ \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ \t{\theta} y -\frac{1}{2} \qform{(y-\mu)}{\Sigma^{-1}}{(y-\mu)} \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ -\frac{1}{2} \left[ \qform{\left(y - (\mu + \Sigma\theta) \right)}{\Sigma^{-1}}{\left(y - (\mu +\Sigma\theta) \right)} \right] + \qform{\theta}{}{\mu} + \frac{\qform{\theta}{\Sigma}{\theta}}{2} \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2} |\Sigma|^{1/2}} \exp{\left[ -\frac{1}{2} \left[ \qform{\left(y - (\mu + \Sigma\theta) \right)}{\Sigma^{-1}}{\left(y - (\mu +\Sigma\theta) \right)} \right] \right]} } \cdot \exp{\left[ \qform{\theta}{}{\mu} + \frac{\qform{\theta}{\Sigma}{\theta}}{2} \right] } \end{align*}

ここで、積分される関数の形をよく見ると、多変正規分布の確率密度関数と同じ形をしている。
具体的には、\mathrm{N}_n(\mu + \Sigma\theta, \Sigma)の確率密度関数とみなせる。
確率分布関数を全域で積分すると1になるので、多次元正規分布の積率母関数は、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \phi(\theta) = \exp{\left[ \qform{\theta}{}{\mu} + \frac{\qform{\theta}{\Sigma}{\theta}}{2} \right]}

となる。

行列の性質をもっとうまく使えば簡単になる気がする…

もっと簡単な方法

積率母関数の性質を使う。

積率母関数の性質

確率変数ベクトルXに対して、確率変数ベクトルYY=AX+bで定義されているとする。
このときYの積率母関数を計算すると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \begin{align*} \phi_Y(\theta) = E\left[\exp{\left(\t{\theta}Y\right)}\right] =& E\left[\exp(\t{\theta}(AX+b))\right] \\ =& \exp{(\t{\theta}b)} \cdot E[\exp{\left( \t{\theta}AX \right)}] \\ =& \exp{(\t{\theta}b)} \cdot E[\exp{\left( \t{(\t{A}\theta)}X \right)}] \\ =& \exp{(\t{\theta}b)} \cdot \phi_X(\t{A}\theta) \end{align*}

となる。

標準多変量正規分布の積率母関数

Xを標準多変量正規分布とすると、その確率密度関数は以下のようになる。

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} f(x) = \frac{1}{(2\pi)^{n/2}} \exp{\left[ -\frac{1}{2} \qform{x}{}{x} \right]}

よって、この積率母関数を計算すると、

\gdef\t#1{#1^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}#2#3} \gdef\nint#1{ \overbrace{ \int\dots\int }^n #1 dx_1 \cdots dx_n } \begin{align*} \phi_X(\theta) =& E[e^{\t{\theta}X}] \\ =& \nint{ \exp{(\t{\theta}x)} \cdot \frac{1}{(2\pi)^{n/2}} \exp{\left[ -\frac{1}{2} \qform{x}{}{x} \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2}} \exp{\left[ \t{\theta}x -\frac{1}{2} \qform{x}{}{x} \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2}} \exp{\left[ -\frac{1}{2} (\qform{x}{}{x} -2 \qform{\theta}{}{x}) \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2}} \exp{\left[ -\frac{1}{2} \left(\qform{(x-\theta)}{}{(x-\theta)} -\qform{\theta}{\theta} \right) \right]} } \\ =& \nint{ \frac{1}{(2\pi)^{n/2}} \exp{\left[ -\frac{1}{2} \left(\qform{(x-\theta)}{}{(x-\theta)} \right) \right]} } \cdot \exp{\left[ \frac{1}{2}\qform{\theta}{}{\theta}\right]} \\ =& \exp{\left[ \frac{1}{2}\qform{\theta}{}{\theta}\right]} \end{align*}

となる。最後の積分はN_n(\theta, I)の確率密度関数を全体で積分しているので1になることを利用している。

標準多変量正規分布と多変量正規分布の関係

多変量正規分布をYと標準多変量正規分布をXとすると以下の式が成り立つ。

Y= \mu + \Sigma^{1/2} X

多変量正規分布の積率母関数

以上のことから、多変量正規分布の積率母関数は

\gdef\t#1{{#1}^{\mathsf{T}}} \gdef\qform#1#2#3{\t{#1}{#2}{#3}} \gdef\nint#1{ \overbrace{ \int\dots\int }^n #1 dx_1 \cdots dx_n } \begin{align*} \phi_Y(\theta) =& E[e^{\t{\theta}Y}] \\ =& \exp{(\qform{\theta}{}{\mu})} \cdot \phi_X (\t{\Sigma^{1/2}} \theta) \\ =& \exp{(\qform{\theta}{}{\mu})} \cdot \phi_X (\Sigma^{1/2} \theta) \\ =& \exp{(\qform{\theta}{}{\mu})} \cdot \exp{\left[\frac{1}{2}\qform{(\Sigma^{1/2}\theta)}{}{(\Sigma^{1/2}\theta)} \right]} \\ =& \exp{(\qform{\theta}{}{\mu})} \cdot \exp{\left[\frac{1}{2}\qform{\theta}{\Sigma}{\theta} \right]} \end{align*}

となる。

四行で終わってしまった。
おわり。

GitHubで編集を提案

Discussion