[2D6-GS-2] 機械学習：ベイス推定

[2D6-GS-2-03] Variational Bayesian Logistic Factorization Machines

〇北原 洋一¹ (1. 株式会社リブセンス)

キーワード：協調フィルタリング、ベイズ推定

Factorization Machines(FM)はMatrix Factorization(MF)を一般化したモデルで、協調フィルタリングにおいて副次的な情報を利用可能にさせる。評価情報が2値の場合ロジスティック回帰タイプのモデルが使われることが多い。FMは多くのMF派生モデルを表現可能な表現能力の高いモデルであるが、何も工夫をしないと学習データにオーバーフィットしやすく、計算量が多いという問題がある。本研究では、変分ベイズ推論によってオーバーフィットを抑制するとともに、事前計算を利用して計算量を削減した、ロジスティック回帰タイプのFMのパラメータ推定方法を提案する。