Fine-Grained Visual Classification (FGVC) aims to distinguish images with subtle differences and limited inter-class variation. This study addresses the lack of understanding of dataset intricacies by analyzing the CUB-200-2011 (CUB) dataset. We identify four ultra-fine-grained subsets that significantly impact accuracy. While local visual cues can often distinguish samples, some fine-grained cases require global context for accurate classification. To address this, we propose a novel LSTM Decision Module (LSTM-DM) that combines local and global information flexibly. Additionally, a Region Selection Module (RSM) selects discriminative regions of fine-grained samples. These modules are integrated into the ViT architecture, enhancing its performance on FGVC tasks. A contrastive loss further improves feature representation by increasing the distance between confusing classes. Extensive evaluations show that our Gate-ViT outperforms existing state-of-the-art methods on four benchmark datasets.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Gate-ViT: Gated Vision Transformer for Fine-Grained Visual Classification

  • Xiaowei Lu,
  • Kanqi Wang,
  • Peiyu Wang,
  • Qin Zhang,
  • Yang Zhao,
  • Gang Liu,
  • Xiaohan Yu

摘要

Fine-Grained Visual Classification (FGVC) aims to distinguish images with subtle differences and limited inter-class variation. This study addresses the lack of understanding of dataset intricacies by analyzing the CUB-200-2011 (CUB) dataset. We identify four ultra-fine-grained subsets that significantly impact accuracy. While local visual cues can often distinguish samples, some fine-grained cases require global context for accurate classification. To address this, we propose a novel LSTM Decision Module (LSTM-DM) that combines local and global information flexibly. Additionally, a Region Selection Module (RSM) selects discriminative regions of fine-grained samples. These modules are integrated into the ViT architecture, enhancing its performance on FGVC tasks. A contrastive loss further improves feature representation by increasing the distance between confusing classes. Extensive evaluations show that our Gate-ViT outperforms existing state-of-the-art methods on four benchmark datasets.