【ラフト(けんさくかくちょうファインチューニング)】

RAFT(検索拡張ファインチューニング) とは?

公開:
💡 答えを見つける訓練を、本番さながらの問題集でやり直す
📌 このページのポイント
RAFT:RAGのためのLLM専用特訓 入力文書群 📄 正解文書 (oracle) 📄 ダミー文書 (distractor) 📄 ダミー文書 (distractor) ❓ 質問 (query) LLM ファインチューニング 中... 学習する能力 正解文書を判別する力 oracleを見極める 引用して回答する力 CoT形式で根拠を示す ノイズを無視する力 distractorに惑わされない 正解文書 (oracle) ダミー文書 (distractor) 質問
RAFTのイメージ:正解とダミーを混ぜた文書群でLLMを特訓する
ひよこ ひよこ
RAGってLLMに検索結果を渡して回答させる仕組みだよね。RAFTって何が違うの?
ペンギン先生 ペンギン先生
RAGはLLMに「検索結果を使いこなす能力」を前提にしているけど、普通のLLMはその訓練を受けていないんだよ。RAFTはその能力をファインチューニングで後付けする手法なんだ。
ひよこ ひよこ
「検索結果を使いこなす能力」ってどういうことなの?
ペンギン先生 ペンギン先生
例えば10件の文書を渡されたとき、どれが本当に答えに関係する文書でどれがノイズかを見極めて、正しい文書から答えを引用する能力だよ。RAFTはこれを意図的に訓練するんだ。
ひよこ ひよこ
じゃあどうやって訓練するの?
ペンギン先生 ペンギン先生
「正解が書かれた文書(oracle)」と「正解がないダミー文書(distractor)」を混ぜて渡して、その中から正しく答えを導くタスクを大量に練習させるんだよ。本番のRAG環境そのままの問題集で特訓する感じだね。
ひよこ ひよこ
なるほど!汎用のファインチューニングとは目的が違うんだね。
ペンギン先生 ペンギン先生
そうだよ。汎用ファインチューニングは「この質問にはこう答えよ」と覚えさせるけど、RAFTは「この文書群からどう答えを探すか」を覚えさせる。RAG前提の設計だから精度の向上幅も大きいんだ。
ひよこ ひよこ
回答に引用元を付けられるって聞いたけど、それも訓練の一部なの?
ペンギン先生 ペンギン先生
そうなんだよ。RAFT訓練では「答えは文書Aの〇〇という部分に書かれています」というCoT(連鎖思考)形式の回答を生成させるから、引用元を明示する習慣も同時に身につくんだ。ハルシネーション対策にもなるよ。
ペンギン
まとめ:ざっくりこれだけ覚えればOK!
「RAFT」って出てきたら「RAGのためのLLM専用特訓」と思えればだいたいOK!
📖 おまけ:英語の意味
「Retrieval Augmented Fine-Tuning」 = 検索拡張ファインチューニング
💬 RAG(Retrieval Augmented Generation)の検索ステップに強くなるようLLMを訓練する手法で、2024年にMicrosoftとUC Berkeleyの研究チームが提案したんだよ。
← 用語集にもどる