Resilience of Vision Transformers for Domain Generalisation in the Presence of Out-of-Distribution Noisy Images
Not on the map. Why?
Abstract (arXiv)
Modern AI models excel in controlled settings but often fail in real-world scenarios where data distributions shift unpredictably - a challenge known as domain generalisation (DG). This paper tackles this limitation by rigorously evaluating vision tramsformers, specifically the BEIT architecture which is a model pre-trained with masked image modelling (MIM), against synthetic out-of-distribution (OOD) benchmarks designed to mimic real-world noise and occlusions. We introduce a novel framework to generate OOD test cases by strategically masking object regions in images using grid patterns (25%, 50%, 75% occlusion) and leveraging cutting-edge zero-shot segmentation via Segment Anything and Grounding DINO to ensure precise object localisation. Experiments across three benchmarks (PACS, Office-Home, DomainNet) demonstrate BEIT's known robustness while maintaining 94% accuracy on PACS and 87% on Office-Home, despite significant occlusions, outperforming CNNs and other vision transformers by margins of up to 37%. Analysis of self-attention distances reveals that the BEIT dependence on global features correlates with its resilience. Furthermore, our synthetic benchmarks expose critical failure modes: performance degrades sharply when occlusions disrupt object shapes e.g. 68% drop for external grid masking vs. 22% for internal masking. This work provides two key advances (1) a scalable method to generate OOD benchmarks using controllable noise, and (2) empirical evidence that MIM and self-attention mechanism in vision transformers enhance DG by learning invariant features. These insights bridge the gap between lab-trained models and real-world deployment that offer a blueprint for building AI systems that generalise reliably under uncertainty.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 45 of its 55 references that are in the map. First posted on arXiv 5 Apr 2025.
Builds on
45 of 55 references.
References · 45
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2023Segment Anythingcited by 8,234
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2023Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detectioncited by 3,332
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
33 more
- 2019Invariant Risk Minimizationcited by 1,854
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,850
- 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,628
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
- 2019Natural Adversarial Examplescited by 1,408
- 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,331
- 2017Random Erasing Data Augmentationcited by 1,239
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,200
- 2020In Search of Lost Domain Generalizationcited by 1,054
- 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,013
- 2017Deeper, Broader and Artier Domain Generalizationcited by 986
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 742
- 2020Self-supervised Learning: Generative or Contrastivecited by 694
- 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
- 2018Recognition in Terra Incognitacited by 658
- 2021LeViT: a Vision Transformer in ConvNet's Clothing for Faster Inferencecited by 475
- 2018Generalizing to Unseen Domains via Adversarial Data Augmentationcited by 451
- 2019Domain Generalization by Solving Jigsaw Puzzlescited by 450
- 2018DropBlock: A regularization method for convolutional networkscited by 262
- 2015Domain Generalization for Object Recognition with Multi-task Autoencoderscited by 253
- 2017Hide-and-Seek: Forcing a Network to be Meticulous for Weakly-supervised Object and Action Localizationcited by 207
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 170
- 2018Domain Generalization via Conditional Invariant Representationcited by 146
- 2018Strike (with) a Pose: Neural Networks Are Easily Fooled by Strange Poses of Familiar Objectscited by 143
- 2018Dark Model Adaptation: Semantic Image Segmentation from Daytime to Nighttimecited by 122
- 2021The Distracting Control Suite -- A Challenging Benchmark for Reinforcement Learning from Pixelscited by 100
- 2022Prompt Vision Transformer for Domain Generalizationcited by 52
- 2019Adversarial Invariant Feature Learning with Accuracy Constraint for Domain Generalizationcited by 43
- 2022Self-Distilled Vision Transformer for Domain Generalizationcited by 35
- 2020Improving out-of-distribution generalization via multi-task self-supervised pretrainingcited by 26
- 2019Towards Shape Biased Unsupervised Representation Learning for Domain Generalizationcited by 10
- 2023Vision Based Machine Learning Algorithms for Out-of-Distribution Generalisationcited by 3
- 2023Domain Generalisation with Bidirectional Encoder Representations from Vision Transformerscited by 2
Built on by
Not cited yet.