arXiv:2503.06368cs.CVcs.AI2025-03

用随机无序编码让ViT在纹理识别上超越CNN

VORTEX: Challenging CNNs at Texture Recognition by using Vision Transformers with Orderless and Randomized Token Encodings

  • 用多尺度特征聚合+无序编码提升ViT纹理表征能力
  • 9个数据集上达到或超越当前最优性能
  • 无需微调模型,适合快速迁移至新任务

纹理识别长期由ImageNet预训练的深度卷积神经网络主导,需专门改进和特征工程才能达到最佳效果。尽管视觉变换器(ViTs)已提出数年,其在纹理识别中的表现仍不明确。为此,本文提出VORTEX(基于无序与随机令牌编码的ViT纹理识别方法),通过从预训练的ViT主干中提取多层级令牌嵌入,并采用轻量级模块聚合层次化特征并进行无序编码,构建更优的图像表示以支持纹理识别。该方法可无缝集成到任意具有通用Transformer结构的ViT中,且不需微调主干网络,仅将其作为冻结特征提取器,输入线性SVM分类器。我们在九个多样化的纹理数据集上评估了VORTEX,证明其在多种纹理分析场景下可实现或超越当前最优性能。此外,相比成本相当的CNN,VORTEX在计算效率上表现出更强的鲁棒性。代码与实验脚本已公开。

原文摘要 · Abstract (English)

Texture recognition has recently been dominated by ImageNet-pre-trained deep Convolutional Neural Networks (CNNs), with specialized modifications and feature engineering required to achieve state-of-the-art (SOTA) performance. However, although Vision Transformers (ViTs) were introduced a few years ago, little is known about their texture recognition ability. Therefore, in this work, we introduce VORTEX (ViTs with Orderless and Randomized Token Encodings for Texture Recognition), a novel method that enables the effective use of ViTs for texture analysis. VORTEX extracts multi-depth token embeddings from pre-trained ViT backbones and employs a lightweight module to aggregate hierarchical features and perform orderless encoding, obtaining a better image representation for texture recognition tasks. This approach allows seamless integration with any ViT with the common transformer architecture. Moreover, no fine-tuning of the backbone is performed, since they are used only as frozen feature extractors, and the features are fed to a linear SVM. We evaluate VORTEX on nine diverse texture datasets, demonstrating its ability to achieve or surpass SOTA performance in a variety of texture analysis scenarios. By bridging the gap between texture recognition with CNNs and transformer-based architectures, VORTEX paves the way for adopting emerging transformer foundation models. Furthermore, VORTEX demonstrates robust computational efficiency when coupled with ViT backbones compared to CNNs with similar costs. The method implementation and experimental scripts are publicly available in our online repository.

纹理识别视觉变换器特征编码SVM

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。