Domain-Adaptive Vision Transformers for Generalizing Across Visual Domains

被引:2
作者
Cho, Yunsung [1 ]
Yun, Jungmin [2 ]
Kwon, Junehyoung [1 ]
Kim, Youngbin [1 ]
机构
[1] Chung Ang Univ, Grad Sch Adv Imaging Sci Multimedia & Film, Seoul 06974, South Korea
[2] Chung Ang Univ, Dept Artificial Intelligence, Seoul 06974, South Korea
基金
新加坡国家研究基金会;
关键词
Domain generalization; ViT; masked ViT; cross-attention-based ViT; glance and gaze; human cognitive approach; STYLE;
D O I
10.1109/ACCESS.2023.3324545
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep-learning models often struggle to generalize well to unseen domains because of the distribution shift between the training and real-world data. Domain generalization aims to train models that can acquire general features from data across different domains, thereby improving the performance on unseen domains. Inspired by the glance-and-gaze approach, which mimics the way humans perceive the real world, we introduce the domain-adaptive vision transformer (DA-ViT) model, which adopts a human cognitive perspective for domain generalization. We merge glance and gaze blocks to initially capture general information from each block and subsequently acquire more detailed and focused information. Unlike previous methods that predominantly employ convolutional neural networks, we adapted the ViT model to learn features that are robust across different visual domains. DA-ViT is pretrained on the ImageNet 1K dataset and designed to adaptively learn features that are generalizable across various visual domains. We evaluated our adapted model for domain generalization and demonstrated that it outperforms the ResNet50 model based on non-ensemble algorithms by 0.7\%p on the VLCS benchmark dataset. Our proposed model introduces a new approach for domain generalization that leverages the capabilities of vision transformers to adapt effectively to diverse visual domains.
引用
收藏
页码:115644 / 115653
页数:10
相关论文
共 79 条
  • [1] Ba J.L., 2016, arXiv
  • [2] Bao HB, 2020, PR MACH LEARN RES, V119
  • [3] Brown T., 2020, ADV NEURAL INF PROCE, V33, P1877
  • [4] Carion Nicolas, 2020, Computer Vision - ECCV 2020. 16th European Conference. Proceedings. Lecture Notes in Computer Science (LNCS 12346), P213, DOI 10.1007/978-3-030-58452-8_13
  • [5] CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification
    Chen, Chun-Fu
    Fan, Quanfu
    Panda, Rameswar
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 347 - 356
  • [6] Pre-Trained Image Processing Transformer
    Chen, Hanting
    Wang, Yunhe
    Guo, Tianyu
    Xu, Chang
    Deng, Yiping
    Liu, Zhenhua
    Ma, Siwei
    Xu, Chunjing
    Xu, Chao
    Gao, Wen
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 12294 - 12305
  • [7] Discriminative adversarial domain generalization with meta-learning based cross-domain validation
    Chen, Keyu
    Zhuang, Di
    Chang, J. Morris
    [J]. NEUROCOMPUTING, 2022, 467 : 418 - 426
  • [8] Chen Q., 2021, P ADV NEUR INF PROC, V34, P25878
  • [9] D'Innocente Antonio, 2019, Pattern Recognition. 40th German Conference, GCPR 2018. Proceedings: Lecture Notes in Computer Science (LNCS 11269), P187, DOI 10.1007/978-3-030-12939-2_14
  • [10] Dai Z, 2021, ADV NEUR IN, V34