通过自监督学习,让模型自动选择关注区域并高效提取特征。
LookWhere? Efficient Visual Recognition by Learning Where to Look and What to See from Self-Supervision
- 用低分辨率选择器+高分辨率提取器,分步计算减少冗余
- 在高分辨率交通标志识别中节省34倍计算量,提速6倍
- 无需任务微调,适配图像分类与分割等多场景
视觉变换器模型越来越大、精度越来越高,但计算成本也急剧上升,尤其在高分辨率下,令牌数量随图像尺寸呈平方增长。为此,我们提出LookWhere方法,通过自监督方式学习何处计算、关注什么内容,将计算分解为低分辨率选择器和高分辨率提取器两阶段,避免处理全图高分辨率输入。通过蒸馏自监督教师模型,联合预训练选择器与提取器,实现对图像可迁移表征的经济高效选择与提取。相比以往需修剪已计算令牌的方法,或依赖复杂任务优化的令牌选择方案,LookWhere无需额外开销即可实现精准筛选。在高分辨率稀疏识别任务(如交通标志)中,准确率保持不变的同时,计算量最多降低34倍,耗时减少6倍;在标准识别任务(如ImageNet分类、ADE20K分割)中,精度提升且推理时间缩短1.36倍。代码与权重见https://github.com/antofuller/lookwhere。
原文摘要 · Abstract (English)
Vision transformers are ever larger, more accurate, and more expensive to compute. The expense is even more extreme at high resolution as the number of tokens grows quadratically with the image size. We turn to adaptive computation to cope with this cost by learning to predict where to compute. Our LookWhere method divides the computation between a low-resolution selector and a high-resolution extractor without ever processing the full high-resolution input. We jointly pretrain the selector and extractor without task supervision by distillation from a self-supervised teacher, in effect, learning where and what to compute simultaneously. Unlike prior token reduction methods, which pay to save by pruning already-computed tokens, and prior token selection methods, which require complex and expensive per-task optimization, LookWhere economically and accurately selects and extracts transferrable representations of images. We show that LookWhere excels at sparse recognition on high-resolution inputs (Traffic Signs), maintaining accuracy while reducing FLOPs by up to 34x and time by 6x. It also excels at standard recognition tasks that are global (ImageNet classification) or local (ADE20K segmentation), improving accuracy while reducing time by 1.36x. See https://github.com/antofuller/lookwhere for the code and weights.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。