MMDiff:マルチモーダル大規模言語モデルの内部特徴を発見し操作する方法
研究の主題は、画像と言葉の両方を扱うマルチモーダル大規模言語モデル(MLLM)の内部で働く「特徴」を見つけ出し、操作できるようにすることです。著者らはMMDiffという枠組みを提案しました。これは隠れ層の表現を疎な成分に分解する「疎自己符号化器(Sparse Autoencoder、SAE)」をマルチモーダル用に学習させ、その差分を使ってどの特徴がマルチモーダル学習で変わったかを見つけたり、特定の振る舞いを制御したりします。要点を平たく言えば、「何がモデルの視覚的な振る舞いを生んでいるか」を特定して、それを消したり強めたりできるようにする手法です。
具体的には三つの使い方を示します。一つ目は特徴の分離で、元の言語モデルに対応するSAEとマルチモーダル適応後のSAEを比較して、どの特徴が変化したかを識別します。二つ目はタスク特化の特徴検出で、トークンごとの対照的な発火解析(どの特徴がその出力に強く関係しているかを見る解析)により因果的に重要な特徴を見つけます。三つ目は特徴レベルの制御で、発見した特徴方向を因果的に取り除いたり、逆に操作して結果を変えることができます。簡単に言えば、SAEで見える「方向」を手がかりに、モデルの内側をのぞき、操作する仕組みです。
実験は三つのモデル系列で行われました:LLaVA-MORE、PaliGemma 2、InternVL3.5。評価タスクは視覚的な空間理解、マルチモーダルの安全性テスト、OCR(光学式文字認識)などです。結果として、発見した疎な特徴を取り除くと、空間タスクで平均12%性能が低下し、OCRでは平均17%低下しました。安全性に関する攻撃に対しては、攻撃成功率が平均24%減少しました。一方で、VQA(Visual Question Answering、視覚質問応答)の性能には影響が見られませんでした。また、見つけた特徴を「操作して強める」ことで、空間理解とOCRの精度が標準的な単層ステアリング基準よりそれぞれ平均+3.6%と+1.8%改善しました。これらは論文で示された平均的な効果です。
この研究が重要な理由は二つあります。第一に、SAEを使った差分解析は、単に結果を観察するだけでなく、内部の原因につながる特徴を明示的に示す手段を提供する点です。第二に、その発見を使ってモデルの振る舞いを監査(どの特徴が危険な応答を生んでいるか調べる)したり、実際に望む方向へ制御したりできる点です。これにより、マルチモーダルモデルの安全性向上や能力の微調整がより直接的に行える可能性が示唆されます。
重要な注意点もあります。結果は三つのモデル系列といくつかのタスクに限定した評価に基づいており、すべてのモデルや応用にそのまま当てはまるとは限りません。SAEによる「疎な方向」に分解できることが前提であり、すべての因果的要素が捕捉されるとは保証されません。また、示された数字は平均的な効果であり、個々のケースでは異なる影響が出る可能性があります。一般化や長期的な安全性への影響を確かめるには、より広いモデル群と多様なタスクでの追加検証が必要です。