8 ガウス過程回帰のハイパーパラメータ推定
前回、ガウス過程回帰によって観測値から自由度が高く柔軟な関数の確率モデルが導かれ、複数のテスト入力に対する回帰値とバラツキを表す分散が推定できることを紹介した。
ガウス過程回帰は、カーネル行列を共分散行列とするガウス分布が主役を務めていることも紹介した。そして、カーネル行列を構成するカーネル関数は、ハイパーパラメータをもつ関数でもあった。前回までは、固定したハイパーパラメータによって推定したが、より良い推定を行うためのハイパーパラメータをチューニングすることが求められる。このチューニングをハイパーパラメータ推定という。ここでは、動径基底関数を例に、最適なハイパーパラメータを推定する方法を紹介する。
8.1 尤度関数の最大化
次式で表すように、3つのハイパーパラメータをまとめてとおくと、カーネル関数はに依存するので、明示的に
と表すことができる。さらに、このカーネルから計算されるカーネル行列もに依存することになるのでと表すこととする。
観測データが与えられたときの出力ベクトルの尤度関数は、
となる。最適なハイパーパラメータ推定は、この尤度関数を最大化するを求めることと等価であるので、式 (8.2) の対数を取った対数尤度関数の最大化問題を考えることとする。
の5個の観測値が与えられたとき、式 (8.1) のカーネル関数で定義されるガウス過程回帰モデルを考えることとする。ハイパーパラメータを固定した場合と、3つのハイパーパラメータの対数を等間隔のグリッド上に配置し、式 (8.3) を直接計算してが最大になるグリッドを推定した場合を比較する。ハイパーパラメータの推定結果は、で、であった。図 8.1(左上)にとしたときの対数尤度を2次元平面上に等高線図としてプロットした。また、図 8.1(右下)にはと固定した場合、図 8.1(左下)にはの場合に推定したガウス回帰過程を示した。対数尤度での差(尤度ではおよそ25倍に相当)があることが示された。参考までに、固定したハイパーパラメータ近傍の対数尤度を図 8.1(右上)に示す。
図 8.1 対数尤度関数と推定したガウス過程回帰
ハイパーパラメータ空間内のグリッドごとに式 (8.3) によって直接対数尤度を求めるのは計算負荷が大きく、この最大化問題を解く方法として様々な提案がなされているが、代表的な手法として
- MCMC法(Markov chain Monte Carlo method, マルコフ連鎖モンテカルロ法)
- SCG法やL-BFGS法などの勾配法(gradient method)
が挙げられる。ここでは、勾配法についてその概要を紹介する。なお、最適なハイパーパラメータの推定のために、MATLABやPythonのプラットフォームで利用できるライブラリが用意されているので改めて紹介の機会を設けることにする。
8.2 勾配法
勾配法は、図 8.1(右上)の青色+を出発点とし、最良推定値である図 8.1(左上)の赤色+である頂上へ辿り着く過程に例えることができる。定性的で直感的ではあるが、出発点における勾配の方向へ進み、峠に到達したら改めて勾配を評価することを繰り返すことで頂上に到達することができる。
そこで、式 (8.3) で定義する対数尤度の勾配をハイパーパラメータによる微分で評価することを考える。式 (8.3) の係数と定数項を省略した次式を微分する。
式 (8.4) に以下の公式を適用することで、のの成分による微分は、
[公式]行列式と逆行列の微分
正方行列がスカラーの関数であるとき
と表すことができる。式 (8.5) は任意のカーネル関数について成立し、カーネル関数のハイパーパラメータの数にも制約がない重要な表現である。
式 (8.1) をカーネル関数とするカーネル行列の微分を式 (8.5) に適用するとき、であるので、
と変換すると、カーネル関数は
さらに、カーネル関数の微分は
となる。1次微分で求めた式 (8.5) による勾配は最適解ではなく、2次微分を含めたより効率的なアルゴリズムである
共役勾配法(CG法、conjugate gradient method)が提案されている。
8.3 ガウス過程回帰のまとめ
ハイパーパラメータ推定については概要のみの紹介となったが、カーネル関数を異なるカーネルの線形結合として定義することもできる。たとえば、線形カーネルとガウスカーネルによって、長期間にわたる気温変動をモデル化することもできる。
これまで線形回帰モデルから出発してガウス過程回帰を紹介してきたが、ガウス過程回帰は「観測値集合から柔軟な回帰モデルの関数を推定する確率モデル」と位置付けることができる。
モノづくりのライフサイクルにおいて、設計プロセスは「性能・品質・コスト」を決定する重要なプロセスである。図 8.2 に示すようにCAEツールと連携することで、部品・アセンブリ・システムそれぞれの階層において試作モデルを作成することなく、「性能・品質・コスト」のバラツキを予測することができる強力なツールを入手したことになる。
ガウス過程回帰を実装した代理モデル(Emulator)をより効果的に運用するためには、以下の課題が残されている。
- 設計空間が3次元を超える場合のバラツキを可視化するツール
- 効率的な設計条件を探索する実験計画法(design of experiments, DOE)
前者については名案はなく、頭の中で想像するか、主成分分析(principal component analysis, PCA)など次元削減手法の適用が現実的な対応である。
次回は後者について紹介することとする。
図 8.2 代理モデルを実現するためのガウス過程回帰