arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.06136cs.CVcs.LG

用于纹理分析中特征提取的视觉Transformer比较综述

A Comparative Survey of Vision Transformers for Feature Extraction in Texture Analysis

São Carlos Institute of Physics, University of São Paulo(圣保罗大学圣卡洛斯物理学院);Institute of Mathematics and Computer Sciences, University of São Paulo(圣保罗大学数学与计算机科学研究所);Institute of Biosciences, Humanities and Exact Sciences, São Paulo State University(圣保罗州立大学生物科学、人文与精确科学研究所);Ghent University(根特大学)

Leonardo Scabini, Andre Sacilotti, Kallil M. Zielinski, Lucas C. Ribas, Bernard De Baets, Odemir M. Bruno

AI总结:

本文系统比较了21种预训练视觉Transformer在纹理特征提取上的表现,发现ViTs在鲁棒性、颜色/材料纹理和属性识别等任务中普遍优于CNNs和手工模型,并指出ViT-B with DINO、BeiTv2、Swin及EfficientFormer为有前景的模型。

摘要:

纹理是图像中一种重要的视觉属性,已在各种图像识别应用中得到广泛研究。卷积神经网络(CNNs)在许多计算机视觉任务中取得了成功,目前属于最佳的纹理分析方法之一。另一方面,视觉Transformer(ViTs)在目标识别等任务上的表现已超越CNNs,引发了该领域的范式转变。然而,ViTs迄今尚未在纹理识别方面得到系统审视,这阻碍了人们对其在这一特定场景下潜力的正确认识。为此,本文探索了多种预训练ViT架构在迁移到依赖纹理的任务时的表现。我们回顾了21种不同的ViT变体,并在多个任务上进行了广泛的评估,并与CNNs和手工设计模型进行比较,这些任务包括评估对纹理旋转、尺度和光照变化的鲁棒性,以及区分颜色纹理、材料纹理和纹理属性。目标是在将预训练ViTs主要用于特征提取并采用线性分类器进行评估的情况下,理解这些模型在直接应用于纹理识别时的潜力和差异。我们还评估了它们的效率,而效率是与其他方法相比的主要缺点之一。我们的结果表明,ViTs通常优于CNNs和手工设计模型,尤其是在使用更强的预训练以及涉及自然场景纹理(来自互联网的图像)的任务中。我们重点指出了以下有前景的模型:采用DINO预训练的ViT-B、BeiTv2和Swin架构,以及作为低成本替代方案的EfficientFormer。在效率方面,尽管ViT-B和BeiT(v2)具有更高的GFLOPs和参数量,但与ResNet50相比,它们在GPU上可以实现更低的特征提取时间。

英文摘要:

Texture, a significant visual attribute in images, has been extensively investigated across various image recognition applications. Convolutional Neural Networks (CNNs), which have been successful in many computer vision tasks, are currently among the best texture analysis approaches. On the other hand, Vision Transformers (ViTs) have been surpassing the performance of CNNs on tasks such as object recognition, causing a paradigm shift in the field. However, ViTs have so far not been scrutinized for texture recognition, hindering a proper appreciation of their potential in this specific setting. For this reason, this work explores various pre-trained ViT architectures when transferred to tasks that rely on textures. We review 21 different ViT variants and perform an extensive evaluation and comparison with CNNs and hand-engineered models on several tasks, such as assessing robustness to changes in texture rotation, scale, and illumination, and distinguishing color textures, material textures, and texture attributes. The goal is to understand the potential and differences among these models when directly applied to texture recognition, using pre-trained ViTs primarily for feature extraction and employing linear classifiers for evaluation. We also evaluate their efficiency, which is one of the main drawbacks in contrast to other methods. Our results show that ViTs generally outperform both CNNs and hand-engineered models, especially when using stronger pre-training and tasks involving in-the-wild textures (images from the internet). We highlight the following promising models: ViT-B with DINO pre-training, BeiTv2, and the Swin architecture, as well as the EfficientFormer as a low-cost alternative. In terms of efficiency, although having a higher number of GFLOPs and parameters, ViT-B and BeiT(v2) can achieve a lower feature extraction time on GPUs compared to ResNet50.

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

↑