[2Q1-J-2] Machine learning: models for prediction

[2Q1-J-2-01] The Cox hazard model using Factorization Machines

〇Kanata Satake¹, Makoto Yamada^1,2, Kota Matsui², Shigeyuki Matsui^2,3, Hisashi Kashima^1,2 (1. Kyoto University, 2. RIKEN Center, 3. Nagoya University)

Keywords:Cox hazard model, Survival analysis, Factorization Machines

生存時間分析はイベントが発生するまでの時間を分析するための手法であり、多くの分野で使用されている。 この生存時間分析の中で最も良く用いられるモデルとして、Cox比例ハザードモデルがある。Cox比例ハザードモデルはイベントの詳細な分布が分からない場合でも利用でき、また線形式の係数により特徴の重要性を解釈することができる。しかし一方で特徴の線形な関係性しか利用できないという問題がある。非線形なCoxモデルも研究されているが、それらのモデルでは解釈ができない欠点がある。本研究では、CoxハザードモデルにFactorization Machines (FM)を導入したモデルを提案する。このモデルでは共変量間の相互作用を利用でき、また解釈も可能である。提案手法の性能を評価するため、実際の遺伝子データから特徴選択を行い、それによって選択された特徴を用いて実験を行った。その結果、提案手法が既存手法に比べ良い性能を示すことを確認した。