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

状態空間エンコーディングによる線形時間のコンテキスト圧縮

記事の要約

本論文は、Mamba状態空間モデルに基づくプラグアンドプレイのオンライン型コンテキスト圧縮器「Boa Compressor」を提案する。任意長のコンテキストを1回の線形時間パスで符号化し、凍結したターゲットLLMの条件付けに用いる。同一圧縮率で既存の最先端オフライン手法と同等以上、Transformerベース実装より約1桁高速で、最大16倍の圧縮率まで対応する。

ソースarXiv Computational Linguistics著者: Marc Serra Ortega, Andrey Barsky, Carlos Boned Riera, Tom\'as Ockier, Dimosthenis Karatzas
状態空間エンコーディングによる線形時間のコンテキスト圧縮
誤りを報告

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

訂正案内
本文へ

Transformerベースの大規模言語モデル(LLM)は、コンテキスト長が増えるにつれて計算コストとメモリコストが大きく膨らむ。この課題に対処するため、推論前にコンテキストをより短い表現へ圧縮する手法が注目されている。しかし既存のソフトプロンプト圧縮手法は、一般にターゲットモデルと同じTransformerバックボーンを用いて圧縮表現を生成するため、その生成に大きなレイテンシが生じる。その結果、コンテキストを一度だけ圧縮し、多数のクエリで再利用するオフライン設定に限定されがちだった。

Marc Serra Ortega氏ら5名による論文は、この制約を緩和する「Boa Compressor」を提案する。これはMamba状態空間モデル(SSM)を土台にしたプラグアンドプレイのオンライン型コンテキスト圧縮器であり、任意長のコンテキストを1回の線形時間パスで符号化し、その表現を凍結したターゲットLLMの条件付けに用いる。

Boaの訓練には蒸留が用いられる。完全なコンテキストを入力するLLMを教師、圧縮表現を入力する同じLLMを生徒とし、両者の次トークンlogit分布を一致させるようにSSMを学習する。これにより圧縮器は、ターゲットモデルと同じバックボーンに依存せず、予測に重要な信号を保持する表現を学ぶ。

論文によれば、同一の圧縮率において、Boaは既存の最先端オフライン圧縮ベースラインと同等かそれ以上の性能を示す。さらにTransformerベースの実装より約1桁高速であり、圧縮処理の追加ではなくエンドツーエンドのレイテンシ削減につながる。より長いコンテキストへの拡張性や、最大16倍という積極的な圧縮率にも対応することも示されている。

圧縮がクエリごとに推論時で実行できるほど低コストである点を踏まえ、著者らはBoaを効率的な長文コンテキスト推論のための実用的かつ汎用的なツールと位置づけている。論文のメタデータによると、2026年10月7日に投稿され、Computation and Language(cs.CL)に分類されつつMachine Learning(cs.LG)にもまたがり、全文は15ページ、図7点で構成される。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • BoaはMamba状態空間モデルにより、任意長のコンテキストを1回の線形時間パスで圧縮し、Transformerベース圧縮器の遅延を回避する。
  • 完全なコンテキストを入力する教師と圧縮表現を入力する生徒の次トークンlogit分布を一致させる蒸留で訓練される。
  • 同一圧縮率で最先端のオフラインベースラインと同等以上、Transformerベース実装より約1桁高速である。
  • より長いコンテキストや最大16倍の積極的な圧縮率にも対応し、クエリごとのオンライン圧縮を実用的にする。

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