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

TraceQuery:拡散言語モデルの完全ノイズ除去軌跡から再計算すべき位置を学習する

記事の要約

本論文は、拡散大規模言語モデル向けの軽量なスパースデコーディングフレームワーク TraceQuery を提案する。軌跡実験により、キャッシュベース生成の精度は完全ノイズ除去軌跡への近さで決まり、誤りはアンマスク順序のずれに集中することが示された。TraceQuery は完全ノイズ除去軌跡を用いて、未解決位置のうち限られた新規計算を受けるべき場所を教師あり学習し、近未来の軌跡関連位置を少数予測してその候補のみをクエリし、バックボーンが新規予測から確定するトークンを選ぶ。キャッシュベース手法の中で最良の品質と計算量のトレードオフを実現し、タスクやモデル変種、モデル系列をまたいで汎化する。

ソースarXiv Computational Linguistics著者: Yukai Jin, Zhankai Ye, Jiayue Yang, Fangyi Li, Shuoqiu Li, Xu Gao, Shibo Li, Xin Liu, Shangqian Gao
TraceQuery:拡散言語モデルの完全ノイズ除去軌跡から再計算すべき位置を学習する
誤りを報告

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

訂正案内
本文へ

拡散大規模言語モデル(dLLM)は、反復的なマスク付きノイズ除去によって柔軟な生成を可能にする一方、各ノイズ除去ステップでトークン表現を繰り返し再計算するため、推論コストが大きい。既存のキャッシュベース手法は、ノイズ除去ステップ間で中間結果を再利用し、各ステップで一部の位置だけを再計算することでコストを削減する。再計算する位置は、固定スケジュールか、変化を示すヒューリスティックな信号によって選ばれてきた。

Yukai Jin、Zhankai Ye、Jiayue Yang、Fangyi Li、Shuoqiu Li、Xu Gao、Shibo Li、Xin Liu、Shangqian Gao による本論文は、この問題を軌跡の観点から捉え直す。著者らの軌跡実験によれば、より本質的な量はデコーディング軌跡であり、単なる表現誤差ではない。キャッシュを用いた生成でも、完全ノイズ除去軌跡に十分近ければ精度は保たれる。誤りはアンマスク順序の逸脱の周辺に集中し、より長い参照プレフィックスが回復されると減少する。

このことから著者らは、表現誤差を抑えることも有用だが、生成を完全ノイズ除去軌跡上に保つことが正確なデコーディングへのより直接的な道だという原則を導く。そこで効率的な dLLM 推論を「軌跡誘導型クエリ」として定式化する。表現誤差を広く修復するのではなく、完全ノイズ除去軌跡を用いて、未解決の位置のうち限られた新規計算を受けるべき場所を教師あり学習する。

この定式化に基づき、著者らは軽量なスパースデコーディングフレームワーク TRACEQUERY を提案する。近未来の軌跡に関連する位置を少数予測し、その候補だけをクエリし、それらの新規予測からバックボーンが確定するトークンを選ぶ仕組みである。これにより、すべての位置を無差別に再計算することを避けつつ、「どこを再計算するか」の判断をデコーディング過程に結び付ける。

実験では、TRACEQUERY がキャッシュベース手法の中で最も優れた品質と計算量のトレードオフを達成し、大幅に少ない新規トークン計算で高い精度を維持することが示された。また、タスク、モデル変種、モデル系列をまたいで良好に汎化する。

論文は本文9ページ、付録5ページ、図4点、表1点で構成され、計算と言語(cs.CL)および人工知能(cs.AI)に分類されている。2026年10月7日に arXiv へ投稿され、識別子は arXiv:2610.33008v1、コードは GitHub で公開されている。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • キャッシュベースの dLLM 生成は完全ノイズ除去軌跡に近いほど高精度で、誤りはアンマスク順序のずれに集中する
  • TraceQuery は近未来の軌跡関連位置を少数予測し、その候補のみをクエリして、バックボーンに確定トークンを選ばせる
  • キャッシュベース手法の中で最良の品質と計算量のトレードオフを示し、タスク・モデル変種・モデル系列をまたいで汎化する

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