表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

長い動画への質問応答で、Mambaの記憶状態を踏まえて視覚トークンを段階的に減らす

Stateful Token Reduction

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

Stateful Token Reductionの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 6 : Token Importance Heatmaps W/ and W/O the Decay Term on Two Mamba Layers. Left: using only the attention score | 𝒃 ¯ j ⊤ ​ 𝒄 t | |\bar{{\bm{b}}}_{j}^{\top}{\bm{c}}_{t}| without decay ( Equation   3 ) produces a more distributed importance pattern across the entire sequence. Right: following Mamba’s full implicit attention pattern | w t , j | |w_{t,j}| ( Equation   2 ), these two layers concentrate the importance heavily on the most recent tokens. A logarithm color space is used in this figure.

出典:Stateful Token Reduction / arXiv ↗

概要

  • Mambaを含む動画向け視覚言語モデルで、質問との関連度を使い、視覚トークンを後の層ほど強く削減する。
  • 著者らは、Mambaの内部状態に以前の情報が残り得ることと、トークンの重要度順位が層ごとに変わることを分析した。
  • 削減を伴う追加学習後、全注意層で視覚トークンを25.2%残す条件では、三つのベンチマークの平均スコアが61.97から63.27、TTFTが4.65秒から1.14秒になった。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Stateful Token Reductionの仕組み

左から右へ読む
  1. 01動画と質問
  2. 02層ごとに関連度採点
  3. 03視覚トークンを段階削減
  4. 04質問への回答

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

長い動画を扱う視覚言語モデルは、動画を小さな情報単位である「視覚トークン」に変えて質問とともに処理する。フレームが増えるほどトークンも増え、回答を作り始める前の計算が重くなる。

従来の削減法は主にTransformer型モデルを対象としていた。Mambaを含むモデルには、前の情報を内部状態に蓄える層があり、トークンを減らした後も情報の一部が残り得るため、削減する位置と時点を改めて調べる必要がある。

手法

動画を視覚トークンに変え、質問文とともにモデルへ入れる。各層で質問との関連度が高い視覚トークンを選び、元の時間順に残す。Transformerの注意層では、文章が各画像情報に置く重みを採点に使う。「注意」は、処理中に参照する情報へ重みを割り当てる仕組みである。

Mamba層は、それまでに読んだ情報を内部状態に蓄えながら順に処理する。著者らは状態の更新式から画像情報と質問の対応を取り出し、距離による減衰を採点から除いた。例えば長い動画の特定の作業を問う場合、序盤は多くの視覚トークンを通し、後の層ほど残す数を減らして回答につなげる。

新規性

著者らは、MambaとTransformerを組み合わせた動画向け視覚言語モデルで、両方の層に質問を条件とする視覚トークンの重要度を定義した。Mamba層では内部状態の更新を展開し、画像情報と質問の対応を測る。

Mamba層の採点では、離れた位置の情報を弱める減衰項を除く。さらに、重要なトークンの順位が層ごとに変わるという分析を基に、序盤には多く残し、後半で削減を強める方式を提案した。

従来手法との違い

著者らは、削減なし、最初の注意層だけでの削減、全注意層での削減、Mamba層を加えた削減などを、同じハイブリッドモデルで比較した。約25%を残す条件では、追加学習の有無も分けて報告している。

別の実験では、Mamba併用のNemotron-Nano-V2 VL 12BとTransformer型のQwen3-VL 8Bを、全注意層で削減する設定で比べた。モデル自体が異なるため、この差を削減方式だけの効果とは解釈できない。

実験結果

著者らはNemotron-Nano-V2 VL 12Bを、VideoMMEの字幕なし条件、LongVideoBench、LVBenchの動画質問応答で評価した。全注意層で削減し、削減を伴う追加学習を行った表2の条件では、視覚トークンを25.2%残したとき、三つのベンチマークの平均スコアは削減なしの61.97に対して63.27だった。

同表のTTFT(回答の最初のトークンが出るまでの時間)は4.65秒から1.14秒になった。TTFTは単一のNVIDIA A100に256フレームを入力して測った値である。著者らは、追加学習なしで全注意層を削減した別条件では、平均スコア61.70とも報告している。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:設備点検の長い動画に対し、「点検員は警告灯を確認した後、何を操作したか」という質問への回答を作る。入力は点検動画と質問、出力は回答とし、担当者が元動画を見て内容を確認する流れを想定する。

この用途への適応には、対象設備の動画と質問・正解回答のラベルが必要である。回答の根拠時刻も表示するなら時刻ラベルと表示機能を別途用意する。追加のセンサーは前提とせず、既存動画で必要な場面が残るかを検証する。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:まず削減しないモデルを基準に、同じ点検動画と質問で回答の正しさとTTFTを記録する。次に質問に応じた段階的削減を適用し、質問別の正答率、重要場面の見落とし、TTFTを比べる。動画の長さごとの結果も分けて確認する。

必要なデータは点検動画、質問、正解回答のラベルで、追加のセンサーは前提としない。対象業務へ追加学習で適応させる場合は、学習用と評価用の動画を分け、削減ありの条件で学習する。根拠時刻を出すなら、時刻ラベルと出力機能を追加する。

限界と課題

著者らは、視覚トークンの重要度順位が層をまたいで安定せず、序盤の強い削減は信頼しにくいと分析している。報告値は対象モデル、三つの動画ベンチマーク、削減率、削減を伴う追加学習の条件に限られる。

編集上の検証事項として、設備点検動画で質問に必要な場面が削減後も使えるか、回答の誤りがどの場面で起きるかを調べる必要がある。確認できた本文の範囲では、根拠時刻を出す機能や、その時刻の正確さを測る結果は示されていない。

研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Stateful Token Reduction for Long-Video Hybrid VLMs

Token reduction accelerates long-video vision--language models (VLMs), but existing methods target Transformers, where reduction is treated as token pruning. We study token reduction in hybrid Mamba--Transformer VLMs and find that it is \emph{stateful}: Mamba layers maintain a recurrent state that accumulates information from earlier tokens, allowing discarded tokens to persist, so reduction behaves more like compression than dropping.We support this view with a representation-based probing method measuring how much information from discarded tokens is retained, and analyze layer-wise sparsity and cross-layer importance stability. Our findings show importance is sparse within layers but unstable across layers, making aggressive early pruning unreliable while hybrids remain robust to later reduction.Motivated by this, we propose a hybrid-aware token reduction framework with a low-to-high progressive schedule and a unified query-conditioned importance score for attention and Mamba layers. For Mamba, excluding the position-dependent decay from the recurrence produces a stronger selection signal. Across long-video benchmarks, our method achieves $3.8{\times}$--$4.2{\times}$ prefilling speedups at a 25% token budget while maintaining near-baseline accuracy and improving with light finetuning. Hybrid models benefit from aggressive reduction, improving both efficiency and accuracy, whereas Transformers exhibit the standard trade-off. Our method also outperforms prior baselines on the same hybrid backbone and combines effectively with visual redundancy reduction methods.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01