技術要約: DeDPO: 拡散モデルのための脱バイアス化された直接選好最適化 (Debiased Direct Preference Optimization for Diffusion Models)
1. 問題提起
Direct Preference Optimization (DPO) は、拡散モデルを人間の選好に適合させるための標準的な手法となっており、個別の報酬モデルを必要としないため、強化学習(RLHF)に代わる効率的な選択肢を提供しています。しかし、DPOには重大なスケーラビリティのボトルネックが存在します。それは、大規模かつ高品質な人間の選好ラベルに依存している点であり、これらを取得するには極めて高価で時間がかかります。
合成的な選好(AIモデルによって生成されたもの)を使用してこの問題に対処しようとする最近の試みは、合成ラベルがノイズを含んでいたり、真の人間の判断と一致しなかったり、系統的なバイアスを導入したりすることがあるため、しばしば失敗に終わります。このような不完全なフィードバックで直接学習を行うことは、アライメントの不安定化や性能の低下を招きます。核心となる課題は、少量の高品質な人間によるラベルのみを用いて、安価で大量の合成データを活用しながら、その固有のノイズとバイアスをいかに補正するかという点にあります。
2. 手法: 脱バイアス化されたDPO (DeDPO)
著者らは、因果推論からの脱バイアス推定技術をDPOの目的関数に統合した半教師あり学習フレームワークである DeDPO を提案しています。この手法は、少量のデータが真の人間による選好ラベルでラベル付けされ、より大規模なデータセットがコスト効率の高いAIフィードバック(例:視覚言語モデルや自己学習)によって生成された合成選好でラベル付けされている設定下で動作します。
2.1. 二値分類としてのDPOの再定式化
著者らはまず、標準的なDiffusion-DPOの損失を二値分類問題として再解釈しています。画像ペア (x0,x1) とプロンプト c に対して、選好ラベル z はどちらの画像が好ましいかを示します。DPOの損失は、モデルの暗黙的なマージン予測器 Gθ を用いた二値クロスエントロピー損失 L(Gθ(y),z) として書き換えられます。
2.2. 脱バイアス化された損失関数
素朴な半教師ありアプローチでは、欠落している人間ラベルを合成ラベル G^(y) で単純に補完することになりますが、これはバイアスのあるOutcome Regression (OR) 推定量を導きます。これに対し、DeDPOは、ラベル付きデータとラベルなしデータを組み合わせた脱バイアス損失 (式8) を提案します。
LDeDPO(θ)=Enl+nu[L(Gθ(y),G^(y))]+Enl[L(Gθ(yl),zl)−L(Gθ(yl),G^(yl))]
ここで:
- nl および nu は、ラベル付きおよびラベルなしのサンプル数です。
- zl は、真の人間によるラベルです。
- G^(y) は、合成ラベルです。
補正メカニズム:
- ラベルなしデータ: 損失は、監督信号として合成疑似ラベル G^(y) を直接使用します。
- ラベル付きデータ: 損失は、疑似ラベルを真の人間によるラベルへとシフトさせる補正項 (zl−G^(yl)) を導入します。この補正は、ラベル付きデータの希少性を反映した重要度重み nlnl+nu によって増幅されます。
- 理論的特性:
- 不偏性: DeDPOの期待値は、合成ラベルが正しいかどうかにかかわらず、母集団レベルのDPOの損失と一致します。合成ラベルが完璧であれば全データを利用し、完全に誤っている場合は漸近的にラベル付きデータのみを使用する状態に戻ります。
- 合成モデルの収束に対する堅牢性: 著者らは、ターゲットとなる拡散モデル θ が、合成選好モデル G^ がるい比べてるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるるる