LIME

Local Interpretable Model-agnostic Explanations

LIMEとは

LIME(Local Interpretable Model-agnostic Explanations)とは、任意の機械学習モデルの個々の予測結果を局所的に解釈可能なモデルで近似することで、予測の説明を生成する手法です。2016年にRibeiroらによって提案され、XAI(説明可能なAI)の代表的な手法のひとつとして広く利用されています。

表:LIMEの要点まとめ
ひとことで言うと1件の予測の周辺だけを単純なモデルで近似して、理由を説明する手法。
何がうれしいかどんなモデルにも使え、計算がSHAPより軽い。
注意点近似なので、毎回少し結果が変わる。安定性に注意が必要。

LIMEの仕組み

LIMEは、説明対象のデータポイント周辺でランダムに摂動を加えたデータを生成し、元のモデルでそれらの予測を行います。次に、元のデータポイントとの距離に基づく重み付けを行いながら、線形モデルや決定木などの解釈可能なモデルでこれらの予測を近似します。この局所的な近似モデルの係数や構造から、各特徴量が予測にどの程度寄与しているかを読み取ることができます。

図:LIMEのしくみ
1説明したい1件この予測はなぜ?
▶
2周辺データを作る少しずつ変えた事例
▶
3元モデルで予測変化を観察
▶
4単純モデルで近似その付近だけ線形に
▶
5係数で説明どの要素が効いたか

LIMEの適用例

テキスト分類タスクでは、どの単語が分類結果に影響を与えたかを可視化できます。画像分類タスクでは、画像のどの領域が分類判断の根拠となったかを示すことができます。表形式データでは、各特徴量の寄与度を棒グラフなどで直感的に表示できます。

LIMEの利点と限界

LIMEの最大の利点は、モデルに依存せず(Model-Agnostic)あらゆるモデルに適用できることです。一方で、局所的な近似であるため大域的なモデルの振る舞いは捉えられないこと、摂動の生成方法によって説明が不安定になりうること、高次元データでの近似精度が低下することなどの限界があります。SHAPなどの他手法と組み合わせて使用することが推奨されます。