arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.04801cs.CVcs.LG

通过推理时注意力工程衰减伪影以改进图像聚类

Improving Image Clustering with Artifacts Attenuation via Inference-Time Attention Engineering

Kioxia Corporation(铠侠公司)

Kazumoto Nakamura, Yuji Nozawa, Yu-Chieh Lin, Kengo Nakata, Youyang Ng

AI总结:

针对预训练ViT模型(尤其DINOv2)中多头注意力高范数伪影降低零样本图像聚类准确率的问题,提出推理时注意力工程(ITAE)方法,无需重训练/微调,通过识别并衰减伪影对应注意力值,在多个数据集上提升了聚类准确率。

摘要:

本文旨在提升预训练Vision Transformer(ViT,视觉Transformer)模型——尤其是DINOv2——在图像聚类任务中的性能,且无需重新训练或微调。随着模型规模增大,多头注意力的图像块中会出现高范数伪影异常。我们观察到该异常会导致零样本图像聚类的准确率下降。这类伪影的特征是注意力图中存在与其他图像块令牌相比数值异常偏大的取值。为解决这些伪影,我们提出名为推理时注意力工程(ITAE)的方法,该方法在推理阶段调整注意力函数。具体而言,我们通过研究多头注意力中的查询-键-值(QKV)图像块之一来识别伪影,并衰减预训练模型内其对应的注意力值。ITAE在多个数据集上提升了聚类准确率,在潜在空间中展现出更具表达力的特征。我们的研究结果凸显了ITAE作为实用方案的潜力:它可减少预训练ViT模型中的伪影,且无需重新训练或微调即可提升模型在聚类任务中的性能。

英文摘要:

The goal of this paper is to improve the performance of pretrained Vision Transformer (ViT) models, particularly DINOv2, in image clustering task without requiring re-training or fine-tuning. As model size increases, high-norm artifacts anomaly appears in the patches of multi-head attention. We observe that this anomaly leads to reduced accuracy in zero-shot image clustering. These artifacts are characterized by disproportionately large values in the attention map compared to other patch tokens. To address these artifacts, we propose an approach called Inference-Time Attention Engineering (ITAE), which manipulates attention function during inference. Specifically, we identify the artifacts by investigating one of the Query-Key-Value (QKV) patches in the multi-head attention and attenuate their corresponding attention values inside the pretrained models. ITAE shows improved clustering accuracy on multiple datasets by exhibiting more expressive features in latent space. Our findings highlight the potential of ITAE as a practical solution for reducing artifacts in pretrained ViT models and improving model performance in clustering tasks without the need for re-training or fine-tuning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑