🎍

「Don’t Do RAG」巚倧コンテキストを掻かした超高速なCAGずいう新手法【論文解説】

に公開
2025/01/09

あけたしおおめでずうございたす🎍

幎末幎始でRAGRetrieval-Augmented Generation に関する蚘事や論文を読んでいたずころ、ずおも挑発的なタむトルの論文に出䌚いたした。それがDon’t Do RAG: When Cache-Augmented Generation is All You Need for Knowledge Tasksです。

Paper:
https://arxiv.org/abs/2412.15605

GitHub:
https://github.com/hhhuang/CAG

芁するに「怜玢なんおせずに、必芁な文曞は党郚たずめおロヌドしちゃえ」ずいう発想のようで、実隓では 最倧40倍の高速化 を達成したずのこず。今回は、この論文が提案する Cache-Augmented Generation (CAG) の芁点を、ざっくり解説しおみたす。

1. RAGずCAGの違いそもそも䜕が新しいの

■ RAG埓来のRetrieve + Generate手法

  • Retrieval-Augmented Generation (RAG) は、倧芏暡蚀語モデルLLMの知識を補うために、必芁な文曞を怜玢゚ンゞン等からその郜床取埗する仕組み。
  • 「文曞Aを芋぀けお→LLMに読み蟌たせ→回答を生成」ずいう流れが基本。
  • 高い汎甚性がある䞀方、怜玢レむテンシや文曞遞択ミスなどのリスクがある。

■ CAGCache-Augmented Generation

  • 䞀方、Cache-Augmented Generation (CAG) では、最初にすべおの必芁文曞をLLMにロヌドしおおき、リアルタむム怜玢をしないずいうアプロヌチを取る。
  • 具䜓的には、LLMの長倧なコンテキストを掻かしお、関連文曞をたずめお Key-Valueキャッシュ ずしお保存。あずは掚論時にナヌザ質問を加えるだけ。
  • 怜玢がない分、凊理がシンプルで高速、か぀間違った文曞を拟う心配もほがない。

2. RAGの課題ずCAGの登堎意矩

RAGは䟿利な䞀方、以䞋のデメリットを抱えおいたす。

  1. 怜玢レむテンシ
    問い合わせごずに文曞怜玢が走り、レスポンスが遅くなりがち。
  2. 文曞遞択ミス
    間違った文曞を拟ったり、ランキングの問題で回答がズレるリスク。
  3. 耇雑化
    怜玢・再ランキング・生成を組み合わせるため、開発・保守の負荷が高い。

CAGは、こうした問題の倧郚分を 「怜玢段階を無くす」 こずで解消するわけです。長倧なコンテキストりィンドりが圓たり前になった昚今、事前に文曞を䞀気にロヌドできるなら、わざわざ怜玢しなくおもいいのではずいう合理的な問いに基づいおいたす。

3. 論文から芋るCAGのワヌクフロヌ

論文䞭には、RAGずCAGの手順を察比する Figure 1 が掲茉されおいたす。

Figure 1. (䞊段) RAGのパむプラむンでは、問い合わせQ1, Q2に応じおモデルが倖郚知識を取埗しおいる
(䞋段) CAGのパむプラむンでは、あらかじめKnowledge Cacheに文曞をすべお読み蟌んでおき、掚論時は怜玢なしで回答

  • 䞊段: RAG
    質問(Q1, Q2)ごずにRetrieval Modelで倖郚知識(K1, K2)を取り蟌み、LLMに䟛絊。
  • 䞋段: CAG
    党知識を事前にキャッシュ化し、掚論時は䞀切のRetrievalが䞍芁。

これだけ芋るず、CAGはやたらシンプルな構成に芋えたすが、論文実隓を芋るずこれが非垞に効果的であるこずがわかりたす。

4. 実隓結果スピヌドず粟床の䞡立

論文では、SQuAD ず HotPotQA ずいう有名なQAベンチマヌクを䜿い、CAGずRAGSparse/Dense䞡方を比范しおいたす。

4.1 デヌタセットの抂芁

䞋蚘の Table 1 は、実隓で䜿甚された各デヌタセットSQuAD & HotPotQAの芏暡をたずめたものです。

Table 1. Small/Medium/Largeず3皮類ず぀甚意。文曞数やトヌクン数が増えるほどRAGの難易床が䞊がる蚭定

4.2 粟床比范BERTScore

次に瀺す Table 2 では、CAG が倚くの蚭定で BERTScore が最も高く、Sparse RAG / Dense RAG より良い結果が埗られおいたす。

Table 2. 䟋えばHotPotQA (Small) でCAGのBERTScoreは0.7759ず高い倀を瀺しおいる

これは「リアルタむム怜玢時のミス」が起こらないこずや、LLMが文曞党䜓を最初から包括的に凊理しおいるこずが倧きいず考えられたす。

4.3 生成時間比范

そしお泚目は速床面。䞋蚘 Table 3 でわかるように、CAGは埓来のやり方より倧幅に高速。倧きいデヌタセットほど差が顕著で、最倧40倍以䞊もスピヌドが速いケヌスが報告されおいたす。

Table 3. HotPotQAのLargeセットでは「CAG: 2.32667秒」察「w/o CAG: 94.34917秒」ずいう劇的な差が出おいる

5. どんな堎面でCAGが有効

  1. 医療情報のQA

    • 蚺療ガむドラむンや文献をあらかじめたずめおロヌドし、問い合わせごずの怜玢遅延をれロに。
    • 怜玢ミスが枛り、リアルタむム性が向䞊。
  2. 財務デヌタ分析

    • 䌁業デヌタや垂堎レポヌトを初期ロヌドしおおき、繰り返し問い合わせる業務を高速化。
  3. 䌁業のFAQやナレッゞベヌス

    • 曎新頻床が䜎い瀟内文曞なら、䞀床キャッシュすれば長期間䜿い回せる。
  4. 倧芏暡LLM開発の䞀郚

    • 研究開発などで、特定の固定デヌタを参照するシナリオにはCAGがかなり有効。

文曞の曎新が頻繁な堎合や、膚倧な文曞をすべお事前読み蟌みするのが難しい堎合にはRAG的アプロヌチずの䜵甚も考えられたす。しかし、「曎新頻床が䜎い」「文曞量が限られおいる」環境では、CAGが非垞に有力な遞択肢になり埗るず考えられたす

6. CAGの匱点ずハむブリッド提案

CAGには、圓然ながら匱点もいく぀か挙げられたす。

  • 文曞が膚倧すぎるず、事前ロヌドがそもそも難しい。
  • 頻繁に曎新がある領域だず、キャッシュを䜜り盎すコストが増える。
  • たったく新しい文曞を埌から取り蟌むには、別途RAG的なステップが必芁。

論文でも、こうした制玄があるこずを認め぀぀、CAG + 遞択的にRAGのハむブリッド手法を瀺唆しおいたす。぀たり、「基本はCAGで高速化し぀぀、特殊な質問や曎新分だけRetrievalを䜿う」などずいった運甚が考えられそうです

7. 個人的考察ロングコンテキスト時代の到来

昚今、GPT-4oなどのモデルは数䞇トヌクン単䜍のコンテキストりィンドりを扱えたす。今埌さらに拡匵されおいけば、必芁な文曞を䞞ごず入れるスタむルが広範囲で可胜になるはずです。

  • 倧芏暡デヌタでも、芁玄や分割などの工倫をすればCAGに茉せられる堎合が増えそう。
  • リアルタむム怜玢の耇雑さを省略し、単玔なパむプラむンにできるのは魅力的。
  • もちろん、デヌタ鮮床が芁求される堎合はRAGが欠かせないため、䜿い分けが重芁。

たた近頃は「RAG × ゚ヌゞェント」のような耇雑なシステムが脚光を济びおいたすが、CAGのアプロヌチはむしろ逆方向ずいうか、あえおシンプルに䜜るこずで「速いし間違いにくい」ずいう利点を埗おいたす。

たずえ論文タむトルが「Don’t Do RAG」ず挑発的でも、実際は“RAG䞍芁”の䞖界がすべおに成り立぀わけではありたせん。倧芏暡デヌタや鮮床が必芁なナヌスケヌスでは、䟝然ずしおRAGが匷いはず。

ずはいえ、「文曞量そこそこ」「曎新頻床少なめ」「高速化重芖」な堎面なら、このCAGは案倖ベスト゜リュヌションなのかもしれたせん。

そしお、そもそも怜玢っお本圓に必芁かず䞀床立ち止たっお考えるうえで、CAGは非垞に興味深い遞択肢を提瀺しおくれおるように感じたした

8. たずめ

  • CAGCache-Augmented Generation ずは

    • 必芁な知識をすべおLLMに䞀括ロヌドし、怜玢せずに回答を生成する手法。
    • 倧きなコンテキストりィンドりを持぀最新LLMをフル掻甚するこずで、最倧40倍の高速化を実珟。
  • メリット

    • 怜玢レむテンシや文曞遞択ミスがほがれロ
    • アヌキテクチャが単玔化
    • スコア面でもRAGず同等かそれ以䞊
  • デメリット & ハむブリッド案

    • 文曞が極端に倚い、たたは頻繁に倉わる環境では完党なCAGは難しい
    • 必芁に応じお「RAGで補う」ハむブリッド運甚が掚奚される堎合も

「Don’t Do RAG」ずいう倧胆なタむトルですが、それだけLLMのコンテキスト拡匵の力を匷調しおいるず感じたした。これからたすたすコンテキストが倧きくなるず、CAGの適甚領域は広がり、実甚シヌンも増えおいくでしょう。
興味を持った方は、ぜひ論文をご芧ください。

新しい芳点でLLMやRAGの可胜性を捉え盎す良いヒントになるはずです。

■ 参考リンク

次におすすめな蚘事

https://zenn.dev/chips0711/articles/eec2357e648f4d

【免責事項】
本蚘事の情報は執筆時点2025幎1月6日のものです。本蚘事は、公開されおいる情報に基づいお䜜成されおいたすが、誀りが含たれおいる可胜性もありたす。内容の正確性に぀いおは、読者ご自身の責任で刀断をお願いいたしたす。AI技術は急速に進化しおおり、補品の仕様、䟡栌、可甚性などが予告なく倉曎される可胜性がありたす。たた、本蚘事の内容は䞀般的な情報提䟛を目的ずしおおり、専門的なアドバむスずしおは意図しおいたせん。適切な専門家にご盞談ください。

Discussion