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 条
  • [51] Shui CJ, 2021, PR MACH LEARN RES, V139
  • [52] On the benefits of representation regularization in invariance based domain generalization
    Shui, Changjian
    Wang, Boyu
    Gagne, Christian
    [J]. MACHINE LEARNING, 2022, 111 (03) : 895 - 915
  • [53] Somavarapu N, 2020, Arxiv, DOI arXiv:2006.11207
  • [54] Szegedy C, 2015, PROC CVPR IEEE, P1, DOI 10.1109/CVPR.2015.7298594
  • [55] Neuron Coverage-Guided Domain Generalization
    Tian, Chris Xing
    Li, Haoliang
    Xie, Xiaofei
    Liu, Yang
    Wang, Shiqi
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (01) : 1302 - 1311
  • [56] Tobin J, 2017, IEEE INT C INT ROBOT, P23
  • [57] Torralba A, 2011, PROC CVPR IEEE, P1521, DOI 10.1109/CVPR.2011.5995347
  • [58] Vaswani A, 2017, ADV NEUR IN, V30
  • [59] Deep Hashing Network for Unsupervised Domain Adaptation
    Venkateswara, Hemanth
    Eusebio, Jose
    Chakraborty, Shayok
    Panchanathan, Sethuraman
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 5385 - 5394
  • [60] Generalizing to Unseen Domains: A Survey on Domain Generalization
    Wang, Jindong
    Lan, Cuiling
    Liu, Chang
    Ouyang, Yidong
    Qin, Tao
    Lu, Wang
    Chen, Yiqiang
    Zeng, Wenjun
    Yu, Philip S.
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (08) : 8052 - 8072