概要
機械学習モデルのパラメーターチューニングにベイズ最適化を使用した例
|
機械学習は、既存の情報から未知の値を予測する手段であるという点で、幅広い範囲において需要があります。
また、機械学習モデルを作成し使用するためには、機械学習アルゴリズムのパラメーターが重要となっています。 しかし、機械学習には難解なパラメーターも存在するために、機械学習はブラックボックスとなる場合も少なくありません。 そこで、ブラックボックス関数において有効な最適化手法の一つである ベイズ最適化 をPipeline Pilotのプロトコルに組み込むことで、学習モデルの改善を目指します。 つまり、本事例ではPipeline Pilotにベイズ最適化を組み込むことで、学習モデルの予測精度向上を図っております。 |
keywordReference |
手法
フローチャートとプロトコル
|
機械学習モデルは、SVMモデルをR言語へアクセスすることで使用しました。 また、ベイズ最適化の最適化対象にRMSE(平均平方二乗誤差)を取り、そして最小化することを目的としました。 RMSEが最小になるモデルを作成することで、モデル精度の向上を図ることが出来ます。 |
keyword |
結果
SVMモデルのパラメーター改善についての考察
|
Iteration number is ... 1 |
SVMモデル構築へ適用するパラメーターをベイズ最適化によって改善させた場合の、モデル精度の変化を上に示しました。 モデル精度は、SVMモデルによる予測値と正解値を比較するプロット、予測値と正解値の誤差のプロットの2種類のプロットから見て取ることが出来ます。 また、ベイズ最適化を通してパラメーター改善を行ったサイクル回数は10回であり、左上画像には初期状態のモデル精度とパラメーター改善後のモデル精度を示しております。 加えて、各サイクル回数におけるそれぞれのプロットは右上に示させていただきました。 このプロットから、ベイズ最適化は機械学習に用いるパラメーターの様な、判断基準が難しくブラックボックスに近いパラメーターの最適化に対して有効であると言うことが出来ます。 |