AI・機械学習
上級
Flora 低ランクファインチューニング(フローラテイランクファインチューニング)
LLMのフルファインチューニングをランダム射影で低ランク勾配更新に変換するメモリ効率化手法。Han et al. 2024提案。LoRAと異なりウェイト行列の構造を変えず、オプティマイザ状態のメモリを削減しながらフルFTに近い精度を達成する。
0 回閲覧
0 いいね
2026/6/6 更新
関連タグ
Flora
PEFT
低ランク勾配
メモリ効率化
オプティマイザ圧縮
フルファインチューニング
LLM
Flora 低ランクファインチューニング
Flora(Full-Rank Low-Rank Adaptation)は、2024年にHan et al.が提案したLLMファインチューニングのメモリ効率化手法。LoRAとは異なるアプローチで、ウェイト行列自体は低ランク化せずにオプティマイザ状態を圧縮することでGPUメモリを削減する。
LoRAとFloraの根本的な違い
| 比較項目 | LoRA | Flora |
|---|---|---|
| ウェイト構造 | 低ランク行列(B×A)に変更 | フルランクを維持 |
| 学習パラメータ | B, Aのみ | 全パラメータ |
| 圧縮対象 | ウェイト行列 | オプティマイザ状態 |
| 推論コスト | LoRAマージ後は同等 | 変化なし |
| 精度 | Full FTより劣る | Full FTに近い |
Adamオプティマイザ状態の問題
Adamオプティマイザはウェイトごとに2つの状態(1次モーメント m・2次モーメント v)を保持する。大規模LLMでこれはウェイト自体の2倍のメモリを消費する:
LLaMA-2 70B フルFTのメモリ要件:
- ウェイト: 140GB (FP16)
- Adamオプティマイザ状態: 280GB (FP32)
- 合計: 420GB以上
Floraの解決策
FloraはAdamのオプティマイザ状態をランダム射影で低ランク近似:
- 勾配更新時にランダム射影を適用
- 低次元空間で状態を更新後に射影で戻す
実験結果
| 設定 | GPUメモリ | MMLU精度 |
|---|---|---|
| Full FT (70B) | 420GB+ | 68.1 |
| LoRA (r=4, 70B) | 160GB | 63.4 |
| Flora (70B) | 220GB | 67.2 |
まとめ
Floraは「フルファインチューニングに近い精度を保ちながらGPUメモリを削減したい」場合の選択肢。LoRAより精度が高い分、メモリ削減効率はLoRAほど極端ではないが、精度とメモリのバランスが優秀。