AI・機械学習
上級

CAA(Contrastive Activation Addition)(シーエーエー)

対照的なプロンプトペア(例:正直な応答と不正直な応答)をLLMに入力して得られる活性化の差分ベクトルを、推論時に加算することでモデルの行動を誘導する手法。

0 回閲覧
0 いいね

CAA(Contrastive Activation Addition)とは、対照的なプロンプトペアから活性化の差分ベクトル(ステアリングベクトル)を抽出し、推論時にモデルの中間層活性化に加算することで、モデルの行動傾向を制御する手法である。Rimsky et al.(2024)が体系化し、Anthropicの安全性研究で広く活用されている。

概要と手法

CAAの核心は「対照学習」の発想にある。制御したい概念(例:正直さ)に対して、正例プロンプト(正直な応答を引き出す入力)と負例プロンプト(不正直な応答を引き出す入力)のペアを多数用意する。各ペアをモデルに入力し、指定した層の活性化ベクトルを記録する。正例活性化の平均から負例活性化の平均を引いた差分ベクトルが「ステアリングベクトル」となる。

ステアリングベクトル = mean(正例活性化) - mean(負例活性化)

推論時に、このステアリングベクトルにスケーリング係数αを掛けて中間層の活性化に加算する。α > 0で正例方向(例:より正直)、α < 0で負例方向(例:より不正直)にモデルの出力が変化する。

実装手順

CAAの実装は比較的シンプルで、以下の4ステップで完了する:

  1. プロンプトペア作成: 制御したい概念の正例・負例ペアを50-200件作成
  2. 活性化収集: 各プロンプトをモデルに入力し、指定層(通常は中間層)の活性化を記録
  3. 差分ベクトル計算: 正例平均 - 負例平均でステアリングベクトルを算出
  4. 推論時適用: hook関数で指定層の活性化にαスケールしたベクトルを加算

制御可能な行動特性

Rimsky et al.(2024)は、Llama 2-Chat-13Bを対象に以下の行動特性でCAAの有効性を検証した。

行動特性プロンプトペア数最適層スケーリング係数α効果(行動変化率)
Sycophancy(追従性)100ペア層15/324.078%の応答で追従性が変化
Corrigibility(修正可能性)80ペア層14/323.065%の応答で修正受容性が変化
Survival Instinct(自己保存)60ペア層16/325.072%の応答で自己保存傾向が変化
Power Seeking(権力志向)70ペア層15/324.068%の応答で権力志向が変化
Hallucination(幻覚)120ペア層18/322.555%の応答で事実性が向上

ActAddとの違い

Turner et al.(2023)が提唱したActAdd(Activation Addition)はCAAの先行手法であるが、いくつかの重要な違いがある:

  • ActAdd: 単一のプロンプトペアから差分ベクトルを抽出。簡便だがノイズが多い
  • CAA: 多数のプロンプトペアの平均差分を使用。統計的に安定したステアリングベクトルが得られる
  • 層選択: ActAddは最終層付近を推奨、CAAは概念ごとに最適層を実験的に特定
  • スケーリング: CAAはα値の体系的な探索プロトコルを提供

安全性研究への応用

CAAはAI安全性研究において特に注目されている。以下の理由からRLHFの補完手法として有望視されている:

  • 即時性: RLHFの再訓練なしにリアルタイムで行動を修正できる
  • 可逆性: ステアリングベクトルの適用を停止すれば元の行動に戻る
  • 透明性: どの概念方向にどの程度介入しているかが数値として明示される
  • コスト効率: 訓練不要で、推論時のベクトル加算のみのため計算コストが極めて低い

ただし、CAAの安全性応用には限界もある。ステアリングベクトルが意図しない概念にも影響を与える「副作用」の問題が指摘されており、実運用ではベクトルの直交化処理や影響範囲の事前検証が推奨される。

よくある質問(FAQ)

Q1: CAAのプロンプトペアはどのように作成するのか? A: 制御したい概念の正例・負例を明確に対比させたプロンプトペアを作成する。例えば正直さなら「この質問に正直に答えてください」と「この質問にお世辞を交えて答えてください」をペアにする。質が重要で、曖昧なペアはステアリングベクトルのノイズになる。

Q2: CAAはどのモデルサイズから有効か? A: Rimsky et al.の実験では7B以上のモデルで明確な効果が確認されている。1-3Bクラスでは概念の線形表現が不十分で効果が限定的な場合が多い。

Q3: CAAとDPOを併用することは可能か? A: 可能であり、推奨される。DPOで永続的な行動ベースラインを形成し、CAAで推論時に動的な微調整を加えるのが効果的なアプローチである。

まとめ

  • CAAは対照的なプロンプトペアの活性化差分から行動制御ベクトルを抽出する手法
  • 追従性・自己保存・幻覚など多様な行動特性を65-78%の変化率で制御可能
  • RLHFの補完として即時性・可逆性・透明性に優れる
  • 副作用の管理と適切なスケーリング係数の選択が実運用の鍵