mlsanityData to 14 Oct 2026

Resilience of Vision Transformers for Domain Generalisation in the Presence of Out-of-Distribution Noisy Images

arXiv 2504.04225 ↗First posted 5 Apr 2025cs.CVcited by 0

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

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
  3. 2023Segment Anythingcited by 8,234
  4. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  5. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  6. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
  7. 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
  8. 2023Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detectioncited by 3,332
  9. 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
  10. 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
  11. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
  12. 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
33 more
  1. 2019Invariant Risk Minimizationcited by 1,854
  2. 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,850
  3. 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,628
  4. 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
  5. 2019Natural Adversarial Examplescited by 1,408
  6. 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,331
  7. 2017Random Erasing Data Augmentationcited by 1,239
  8. 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,200
  9. 2020In Search of Lost Domain Generalizationcited by 1,054
  10. 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,013
  11. 2017Deeper, Broader and Artier Domain Generalizationcited by 986
  12. 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 742
  13. 2020Self-supervised Learning: Generative or Contrastivecited by 694
  14. 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
  15. 2018Recognition in Terra Incognitacited by 658
  16. 2021LeViT: a Vision Transformer in ConvNet's Clothing for Faster Inferencecited by 475
  17. 2018Generalizing to Unseen Domains via Adversarial Data Augmentationcited by 451
  18. 2019Domain Generalization by Solving Jigsaw Puzzlescited by 450
  19. 2018DropBlock: A regularization method for convolutional networkscited by 262
  20. 2015Domain Generalization for Object Recognition with Multi-task Autoencoderscited by 253
  21. 2017Hide-and-Seek: Forcing a Network to be Meticulous for Weakly-supervised Object and Action Localizationcited by 207
  22. 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 170
  23. 2018Domain Generalization via Conditional Invariant Representationcited by 146
  24. 2018Strike (with) a Pose: Neural Networks Are Easily Fooled by Strange Poses of Familiar Objectscited by 143
  25. 2018Dark Model Adaptation: Semantic Image Segmentation from Daytime to Nighttimecited by 122
  26. 2021The Distracting Control Suite -- A Challenging Benchmark for Reinforcement Learning from Pixelscited by 100
  27. 2022Prompt Vision Transformer for Domain Generalizationcited by 52
  28. 2019Adversarial Invariant Feature Learning with Accuracy Constraint for Domain Generalizationcited by 43
  29. 2022Self-Distilled Vision Transformer for Domain Generalizationcited by 35
  30. 2020Improving out-of-distribution generalization via multi-task self-supervised pretrainingcited by 26
  31. 2019Towards Shape Biased Unsupervised Representation Learning for Domain Generalizationcited by 10
  32. 2023Vision Based Machine Learning Algorithms for Out-of-Distribution Generalisationcited by 3
  33. 2023Domain Generalisation with Bidirectional Encoder Representations from Vision Transformerscited by 2

Built on by

Not cited yet.