TL;DR

SepPruneは、MLLMの入力シーケンス内にある「モダリティセパレータトークン」が視覚とテキストの橋渡しをしていることに着目。このセパレータをクエリとして全視覚トークンの重要度を一括評価し、不要なトークンを事前に刈り込む。位置エンコーディングを外すことで位置バイアスを排除し、FlashAttentionとも互換。Qwen2.5-VL-7Bで80.2%のトークンを削減しても元の精度の96.3%を維持。

解説

AMI HAPPY

ねえ智也くん、この「SepPrune」って論文、タイトルだけ見ると「セパレータトークンで視覚トークンを刈り込む」って書いてあるけど、どういうこと?

TOMOYA NEUTRAL

ああ、MLLMっていう画像とテキストを両方扱うモデルがあるんだけど、入力に視覚トークンが大量にあって処理が重いんだ。そこで、モダリティセパレータトークンっていう特別なトークンが視覚とテキストの橋渡しをしてることに注目して、そのセパレータを使って重要じゃない視覚トークンを事前に削除する手法だよ。

AMI SURPRISED

なるほど!つまり、セパレータトークンが「この視覚トークンは大事だよ」って教えてくれるってこと?

TOMOYA NEUTRAL

そう。セパレータをクエリにして、全視覚トークンの重要度を一括で評価するんだ。しかも位置エンコーディングを外すことで、位置バイアスを排除してるから、FlashAttentionとも互換性がある。

AMI HAPPY

位置バイアスって何?

TOMOYA NEUTRAL

例えば、画像の左上のトークンが常に重要と判断されやすいみたいな偏りのこと。それをなくすことで、より公平に重要度を測れるんだ。

AMI HAPPY

へえ~。で、実際どれくらい効果あるの?

TOMOYA NEUTRAL

Qwen2.5-VL-7Bっていうモデルで試した結果、80.2%のトークンを削減しても、元の精度の96.3%を維持できたんだ。つまり、かなり高速化できる上に精度もほとんど落ちない。

AMI SURPRISED

すごい!じゃあもう全部これでいいんじゃない?

TOMOYA NEUTRAL

ただ、限界もあるよ。例えば、セパレータトークン自体がモデルに組み込まれている必要があるから、すべてのMLLMにそのまま適用できるわけじゃない。あと、極端に視覚情報が重要なタスクだと、削りすぎると精度が落ちる可能性もある。

AMI HAPPY

なるほどね。でも、これで画像認識が爆速になったら、スマホのカメラで「これは何?」って聞いたときの返事も一瞬になるのかな?

TOMOYA NEUTRAL

まあ、理論上はね。でも、君の質問が長すぎると、その分だけトークンが増えて逆効果かもよ。