Combating Textual Noise and Redundancy: Entropy-Aware Dense Visual Token Pruning
本文提出了一种熵感知密集剪枝(Entropy-Aware Dense Pruning, EADP)框架,该框架通过利用统计熵过滤文本噪声,并利用次模最大化优化视觉标记选择,从而在密集指令下保留细粒度线索,进而提升视觉语言模型的效率。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
想象一下,你有一个非常聪明但有点不知所措的助手(一个视觉-语言模型),它正试图回答一个关于图片的问题。这张图片是由数千个微小的拼图碎片(即 Token)组成的。助手会观察每一个碎片来寻找答案。
问题在于?观察每一个碎片太慢了,而且助手会感到疲劳。所以,我们通常尝试扔掉那些“无聊”的碎片,只保留重要的部分。这被称为Token Pruning(Token 剪枝)。
然而,论文的作者发现目前这种做法在两个特定的、有趣的方式上出了问题。他们构建了一个名为 EADP(熵感知密集剪枝,Entropy-Aware Dense Pruning)的新系统来修复这些问题。
以下是他们如何修复这些问题的,使用了简单的类比:
两个大问题
1. “静态噪声”问题(文本噪声)
想象一下,你正根据一段描述试图在一个拥挤的房间里找到一个特定的人,描述是:“图中是否有狗?”
- 旧方法: 助手将整个句子作为一个大块来倾听。它听到了“狗”,但也听到了“是”、“有”、“一只”和“在”。因为它把整个句子当作一个整体,这些无聊的词(“是”、“一个”)产生了大量的背景静态噪声。这种静态噪声淹没了重要的词(“狗”),导致助手看向了房间里错误的地方(比如空地板),而不是看向狗。
- 论文的解决方法: 他们意识到有些词(如“是”或标点符号)之所以具有“噪声”,是因为它们并不指向图像中的任何具体事物。他们使用了一种叫做熵(Entropy)(衡量混乱或随机性的指标)的数学技巧来检测这些有噪声的词。如果一个词的注意力像房间里的静态噪声一样散乱,他们就会将其过滤掉。如果一个词很集中(比如“狗”),他们就会保留它。这清理了指令,让助手明确知道该寻找什么。
2. “单点聚焦”问题(冗余)
想象一下,助手终于知道要看狗了。
- 旧方法: 它使用了一个“Top-K”规则。它找到了看起来最像狗鼻子的那块拼图碎片,并保留了它。然后它又找到了另一块看起来像鼻子的碎片,并再次保留。结果它得到了一堆 10 块看起来都只是狗鼻子的拼图碎片,却把耳朵、尾巴和身体都扔掉了。助手的鼻子很灵,但完全不知道狗的其他部分长什么样。
- 论文的解决方法: 他们意识到我们需要的是一个平衡的视角,而不只是“最好的”视角。他们将选择过程改为了一个名为**“设施选址”(Facility Location)**的游戏。
- 类比: 想象一下你在城市中布置消防站。你不会仅仅把 10 个站点都放在火灾最多的那个社区(那是“Top-K”的错误)。相反,你会布置这些站点,使得城市的每一个部分都靠近某个站点。
- EADP 对图像也这样做。它挑选出的 Token 能够覆盖整个狗(鼻子、耳朵、尾巴),而不会在重复内容上浪费空间。它确保最终缩减后的 Token 集合能讲述完整的故事,而不仅仅是一个微小的细节。
EADP 是如何工作的(分步说明)
- 清理指令: 它扫描文本提示,识别出“有噪声”的词(如“是”或“?”),并将它们移除。它保留那些“响亮”的词(如“烤箱”或“大象”),这些词真正指向图像中的物体。
- 制作地图: 它创建一张图像的热力图,显示重要事物所在的位置,但它会对热力图进行平滑处理,以免它只强调某一个微小的像素点。
- 智能选择: 它不再仅仅挑选地图上最热的点,而是玩起了“消防站”游戏。它挑选出一组分布足够广泛的 Token,以覆盖整个图像,确保没有重要的部分被遗漏。
结果
作者在许多不同的 AI 模型和困难问题(例如寻找图像中的微小文字或回答关于视频的复杂问题)上测试了该方法。
- 速度: 通过扔掉 80-90% 无用的图片碎片,AI 的运行速度提升了 2 到 5 倍。
- 聪明程度: 与其他在面对复杂图片或棘手问题时会感到困惑的方法不同,EADP 保持了 AI 的聪明才智。在某些情况下,它的得分甚至比原始的、未经过剪枝的 AI 还要高,因为这减少了 AI 被噪声干扰的情况。
简而言之: EADP 就像是 AI 视觉领域的一个智能编辑师。它剪掉了指令中无聊的词汇,并确保 AI 保留一组多样化且完整的图片碎片,从而在不损失细节观察能力的前提下,让运行速度变得更快。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。