クレジットの無駄遣いを防ぐ:GPTCacheとRedisによるLLMコスト最適化

AI tutorial - IT technology blog
AI tutorial - IT technology blog

LLM予算を蝕む目に見えないコスト

LLMアプリケーションを立ち上げた後の「ハネムーン期間」は、通常、最初の請求サイクルで終わりを迎えます。多くの場合、APIコストはユーザーベースよりも速いスピードで増加することに気づくでしょう。ドキュメントアシスタントであれテクニカルサポートボットであれ、現実としてユーザーの質問が常に独創的であることは稀です。彼らは、表現こそ違えど同じ質問を繰り返します。従来のキャッシュは100%の文字列一致を必要とするため、ここでは役に立ちません。

考えてみてください。ユーザーAが「パスワードをリセットするにはどうすればいいですか?」と尋ね、ユーザーBが「パスワードを忘れた際の手順は?」と尋ねたとします。従来のRedisキャッシュにとって、これらは全く別のものです。しかし、あなたの銀行口座にとっては、OpenAIから全く同じ回答を得るために2回分の料金が発生することを意味します。1日5万件のクエリを処理するプロダクション環境において、この非効率性は単なる迷惑ではなく、毎月数千ドルもの損失を招く「資金漏洩」なのです。

クイックスタート:5分でセマンティックキャッシュを導入する

セマンティックキャッシュはテキストそのものではなく、「意図」に注目します。クエリの数学的な「意味」(エンベディング)を保存することで、GPTCacheのようなツールは、新しい質問が以前に回答されたものと機能的に同一であるかどうかを識別できます。これは、コードとLLMの間のスマートなレイヤーとして機能します。

まず、ライブラリをインストールします:

pip install gptcache openai

わずか数行のPythonコードで、実用的なローカルキャッシュを構築できます:

from gptcache import cache
from gptcache.adapter import openai
from gptcache.embedding import Onnx
from gptcache.similarity_evaluation.distance import SearchDistanceEvaluation
from gptcache.processor.pre import get_prompt
from gptcache.manager import get_data_manager

# エンベディングエンジンとベクトルストレージのセットアップ
data_manager = get_data_manager(cache_base="sqlite", vector_base="faiss", max_size=1000)

cache.init(
    pre_embedding_func=get_prompt,
    embedding_func=Onnx().to_embeddings,
    data_manager=data_manager,
    similarity_evaluation=SearchDistanceEvaluation(),
)

# 通常通りOpenAIを呼び出す
response = openai.ChatCompletion.create(
  model="gpt-3.5-turbo",
  messages=[{"role": "user", "content": "セマンティックキャッシュとは何ですか?"}]
)

最初のリクエストはAPIに到達し、約2秒かかります。2回目のリクエストは、たとえ言い回しが異なっていても、ローカルストレージから50ミリ秒未満で返されます。これにより、トークンコストを100%節約し、レイテンシを97%削減できました。

魔法の裏側:ベクトルと閾値

LLMは、多次元座標系で言語を処理します。このベクトル空間では、「天気はどうですか?」と「外は雨が降っていますか?」はすぐ隣に位置します。セマンティックキャッシュはこの地理的特性を利用します。

セマンティックキャッシュの3本の柱

  • エンベッダー(Embedder): ユーザーの複雑な言語を数値の羅列に変換します。
  • ベクトルストア(Vector Store): キーワード検索ではなく、「最近傍」計算に特化した特殊なデータベースです。
  • エバリュエーター(Evaluator): 0.08という距離スコアが一致と見なすのに十分近いかどうかを判断する門番です。

新しいクエリとキャッシュされたクエリの距離が、選択した閾値(厳密な精度を求める場合は通常0.1)を下回る場合、システムはキャッシュされた結果を返します。それを上回る場合は「キャッシュミス」となり、クエリをLLMに送信し、次回の利用に備えて新しい結果を保存します。

スケーリング:本番環境に向けたRedisへの移行

ローカルのSQLiteファイルはデモには適していますが、分散クラウド環境では機能しません。複数のAPIノードでキャッシュを共有するには、Redisが必要です。具体的には、外部データベースを必要とせずにネイティブでベクトル検索を処理できるRedis Stackを使用します。

最近の導入事例では、中央集中型のRedisキャッシュに切り替えたことで、ノードAがノードBで回答された質問の恩恵を即座に受けられるようになりました。この統一されたメモリは、スケールアップに伴い高いヒット率を維持するために不可欠です。

以下は、本番環境向けのRedisバックエンドの設定例です:

from gptcache.manager import get_data_manager

# ベクトル検索をサポートするRedisインスタンスに接続
data_manager = get_data_manager(
    data_path="redis://localhost:6379",
    vector_path="redis://localhost:6379",
    vector_params={"dimension": 384} # モデル의 次元数に合わせる
)

cache.init(data_manager=data_manager, ...)

スマートエビクション:データの鮮度を保つ

キャッシュを無制限に増大させないでください。RedisのTTL(Time To Live)を使用して、回答の有効期限を設定します。一般的なFAQであれば48時間のTTLが適しています。頻繁に更新されるドキュメントの場合は、ユーザーが古いテクニカルアドバイスを受け取らないよう、12時間ごとにキャッシュを循環させるとよいでしょう。

最高のパフォーマンスを引き出すためのプロのコツ

実装は簡単ですが、最適化が難しい部分です。トラフィックの多いAIアプリを管理してきた経験から学んだことを紹介します:

1. 閾値を緩くしすぎない

まずは厳格な閾値である0.1から始めてください。0.25に設定すると、「Javaプログラミング」に関する質問に対して「コーヒー豆(Java)」に関するキャッシュ回答を返してしまう可能性があります。ログを毎日監視し、これらの「誤ヒット」をチェックしてください。

2. ノイズを除去する

株価や天気、ユーザー固有の個人情報(PII)などのリアルタイムデータを含むクエリは決してキャッシュしないでください。シンプルな正規表現プロセッサを使用してこれらを特定し、キャッシュを完全にバイパスすることで、セキュリティ漏洩や古いデータの提供を防げます。

3. 40%の「スイートスポット」を目指す

適切に調整されたサポートボットのキャッシュヒット率は、通常30%から50%の間になります。10%を下回る場合は閾値が厳しすぎるか、ユーザーが非常にユニークな質問をしている可能性があります。80%に達している場合は、おそらく一般的で不正確な回答を提供してしまっています。

4. サイズよりも速度を重視する

キャッシュのために巨大なエンベディングモデルは必要ありません。ONNX経由でall-MiniLM-L6-v2を使用すれば、外部APIを呼び出すよりも大幅に高速です。検索プロセス全体をインフラ内で完結させ、レイテンシを100ミリ秒未満に抑えましょう。

セマンティックキャッシュは、赤字のAI実験を収益性の高いプロダクションツールに変えるための最も効果的な方法です。GPTCacheとRedisを組み合わせることで、利益率を守りつつ、ユーザーが期待する瞬時のレスポンスタイムを提供できます。

Share: