本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

UCCI:コスト最適なLLMカスケードルーティングのための校正済み不確実性

記事の要約

UCCIは「校正優先」のLLMカスケードルーターを提案する。トークン単位のマージン不確実性を等張回帰でクエリごとの誤り確率に写像し、制約付きコスト最小化によってエスカレーション閾値を選ぶ。本番のNERワークロードでは、micro-F1=0.91を維持しつつ推論コストを31%削減し、ECEを0.12から0.03へ低減。エントロピー閾値法、split-conformalルーティング、FrugalGPT型の学習閾値を上回る。

ソースarXiv Machine Learning著者: Varun Kotte
UCCI:コスト最適なLLMカスケードルーティングのための校正済み不確実性
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

Varun Kotte氏による本論文(arXiv:2605.18796、2026年5月11日投稿)は、大規模言語モデル運用におけるコスト課題に焦点を当てている。LLMカスケードやモデルルーティングは、容易なクエリを小型モデルに、難しいクエリを大型モデルに振り分けることで推論コストを下げる手法だが、著者は、実運用されているルーターの多くが未校正の信頼度スコアに依存し、ワークロードごとに閾値を調整する必要がある点を問題視している。

UCCIの中心は「まず校正し、その後にルーティングする」という設計である。トークン単位のマージン不確実性を等張回帰(isotonic regression)によってクエリごとの誤り確率へ変換し、校正済みスコアを得たうえで、制約付きコスト最小化によりエスカレーション閾値を決定する。人手による試行錯誤の閾値調整を不要にする狙いである。理論面では、3つの明示的な仮定の下で校正済みスコアに基づく閾値ポリシーがコスト最適であることを示し、等張校正が期待校正誤差(ECE)に対してO(n^{-1/3})のサンプル複雑度を達成することも示している。

評価には本番環境の固有表現認識(NER)ワークロードを用い、75,000件のクエリをH100 GPU上の4Bおよび12Bの命令チューニング済みモデルで処理した。結果として、micro-F1=0.91を維持しながら推論コストを31%(95%信頼区間:27%〜35%)削減し、ECEを0.12から0.03へ低減した。信頼度スコアが実際の誤り率に近づいたことを示す結果である。

同一の動作点において、UCCIはエントロピー閾値法、split-conformalルーティング、FrugalGPT型の学習閾値を上回った。著者は、すべてのカスケード結果が実際のモデル出力と実測H100遅延に基づくエンドツーエンドのルーティングで得られたものであり、グローバルな精度や名目上のAPI価格によるシミュレーションではない点を強調している。論文は9ページ、図2点、表4点で構成され、コードも公開されている。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • トークン単位のマージン不確実性を等張回帰で校正し、ECEを0.12から0.03へ改善
  • 制約付きコスト最小化でエスカレーション閾値を決定し、3つの明示的仮定の下でコスト最適性を示す
  • 4B・12Bモデルと実測H100遅延を用いた75,000クエリの本番NERワークロードで推論コストを31%削減

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。