AIのウソを見抜く新技術、計算の手間を大幅削減し小型モデルでも高精度に判定
要点 3行
- AIが文脈に基づかない誤った回答を出力する現象を、小型モデルで効率的に検知する手法が提案された。
- 注意データの行列を直接展開せず計算することで、計算の遅延増加を最大1.4%まで抑えることに成功した。
- 3つの性能評価において、提案手法を用いた小型モデルがより大規模な判定用AIを上回る精度を達成した。
大規模言語モデルが文脈に基づかない誤った回答を生成する課題に対し、凍結した小型の代替モデルを用いて検知する新たな手法が発表された。従来の注意機構に基づく検知手法では、系列長に応じてメモリ使用量が二乗で増大する注意行列を展開する必要があり、計算負荷の高さが実用上の課題となっていた。
提案手法では、注意行列を展開することなく、高速化技術のFlashAttentionの処理過程での統計量累積やスパース参照などを組み合わせて特徴量を算出する。H100環境での検証では、Qwen3-8Bの前向き計算において、2,000トークン時の遅延増加が16%だったのに対し、3万5,000トークンでは1.4%に抑えられ、応答トークンあたりのメモリ消費も22.5KiBにとどまった。
3つのベンチマークによる検証では、40億パラメータの代替モデルが、40億〜90億パラメータの評価用モデル9種類のうち最も高いF1スコアを持つモデルを2つのベンチマークで上回る精度を示した。
この記事は arXiv cs.AIの発表を AI が日本語に要約したものです。正確な内容は原文をご確認ください。
アプリで開く