mlsanityData to 14 Oct 2026

Beyond ImageNet: Understanding Cross-Dataset Robustness of Lightweight Vision Models

arXiv 2511.00335 ↗First posted 1 Nov 2025cs.CV cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Lightweight vision classification models such as MobileNet, ShuffleNet, and EfficientNet are increasingly deployed in mobile and embedded systems, yet their performance has been predominantly benchmarked on ImageNet. This raises critical questions: Do models that excel on ImageNet also generalize across other domains? How can cross-dataset robustness be systematically quantified? And which architectural elements consistently drive generalization under tight resource constraints? Here, we present the first systematic evaluation of 11 lightweight vision models (2.5M parameters), trained under a fixed 100-epoch schedule across 7 diverse datasets. We introduce the Cross-Dataset Score (xScore), a unified metric that quantifies the consistency and robustness of model performance across diverse visual domains. Our results show that (1) ImageNet accuracy does not reliably predict performance on fine-grained or medical datasets, (2) xScore provides a scalable predictor of mobile model performance that can be estimated from just four datasets, and (3) certain architectural components--such as isotropic convolutions with higher spatial resolution and channel-wise attention--promote broader generalization, while Transformer-based blocks yield little additional benefit, despite incurring higher parameter overhead. This study provides a reproducible framework for evaluating lightweight vision models beyond ImageNet, highlights key design principles for mobile-friendly architectures, and guides the development of future models that generalize robustly across diverse application domains.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 37 of its 52 references that are in the map. First posted on arXiv 1 Nov 2025.

Builds on

37 of 52 references.

References · 37

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
  3. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  4. 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
  5. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
  6. 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
  7. 2019Searching for MobileNetV3cited by 1,994
  8. 2017ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devicescited by 1,200
  9. 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
  10. 2021EfficientNetV2: Smaller Models and Faster Trainingcited by 1,135
  11. 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,060
  12. 2018MnasNet: Platform-Aware Neural Architecture Search for Mobilecited by 943
25 more
  1. 2021DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsificationcited by 864
  2. 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 642
  3. 2018Do Better ImageNet Models Transfer Better?cited by 565
  4. 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 497
  5. 2018FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Searchcited by 485
  6. 2019Single Path One-Shot Neural Architecture Search with Uniform Samplingcited by 414
  7. 2019GhostNet: More Features from Cheap Operationscited by 410
  8. 2022EfficientFormer: Vision Transformers at MobileNet Speedcited by 241
  9. 2021Efficient Deep Learning: A Survey on Making Deep Learning Models Smaller, Faster, and Bettercited by 154
  10. 2020FBNetV2: Differentiable Neural Architecture Search for Spatial and Channel Dimensionscited by 152
  11. 2022MobileOne: An Improved One millisecond Mobile Backbonecited by 109
  12. 2017Shift: A Zero FLOP, Zero Parameter Alternative to Spatial Convolutionscited by 93
  13. 2019AdderNet: Do We Really Need Multiplications in Deep Learning?cited by 82
  14. 2023SwiftFormer: Efficient Additive Attention for Transformer-based Real-time Mobile Vision Applicationscited by 66
  15. 2018SqueezeNext: Hardware-Aware Neural Network Designcited by 61
  16. 2020Rethinking Bottleneck Structure for Efficient Mobile Network Designcited by 54
  17. 2025Empowering Edge Intelligence: A Comprehensive Survey on On-Device AI Modelscited by 46
  18. 2022GhostNetV2: Enhance Cheap Operation with Long-Range Attentioncited by 44
  19. 2024Tiny Machine Learning: Progress and Futurescited by 40
  20. 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
  21. 2023Does progress on ImageNet transfer to real-world datasets?cited by 35
  22. 2023ConvNet vs Transformer, Supervised vs CLIP: Beyond ImageNet Accuracycited by 19
  23. 2022CDFKD-MFS: Collaborative Data-free Knowledge Distillation via Multi-level Feature Sharingcited by 15
  24. 2021Is it enough to optimize CNN architectures on ImageNet?cited by 9
  25. 2023From MNIST to ImageNet and Back: Benchmarking Continual Curriculum Learningcited by 6

Built on by

Not cited yet.