Beyond the Performance Illusion: Structure-Aware Stratified Partitioning and Curriculum Distributionally Robust Optimization for Spatially Correlated Domains
Not on the map. Why?
Abstract (arXiv)
Performance evaluation in AI systems commonly assumes that random dataset splits produce independent and identically distributed (i.i.d.) subsets. We show that this assumption often breaks down in spatiotemporally correlated domains such as aerial surveillance, precision agriculture, and medical imaging, leading to two systematic failures: data leakage, where correlated samples span training and validation splits and inflate performance estimates, and hidden stratification, where errors on minority subpopulations are obscured by aggregate metrics. To address these issues, we propose a unified evaluation and training framework for spatially correlated data. We introduce Structure-Aware Stratified Partitioning (SASP), which constructs validation splits that reduce spatiotemporal leakage while preserving meaningful class balance, and Curriculum Distributionally Robust Optimization (CDRO), a curriculum-based relaxation of distributionally robust training that stabilizes optimization under these stricter splits. Across multiple benchmarks, this combination yields consistently improved generalization, more reliable confidence calibration, and exposes failure modes that remain hidden under conventional random-split evaluation.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 32 references that are in the map. First posted on arXiv 2 Jul 2026.
Builds on
22 of 32 references.
References · 22
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
- 2019Scalability in Perception for Autonomous Driving: Waymo Open Datasetcited by 2,451
- 2019Invariant Risk Minimizationcited by 1,854
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2020In Search of Lost Domain Generalizationcited by 1,054
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
- 2020Measuring Robustness to Natural Distribution Shifts in Image Classificationcited by 454
- 2021Open-Set Recognition: a Good Closed-Set Classifier is All You Need?cited by 353
10 more
- 2020Environment Inference for Invariant Learningcited by 306
- 2020The Risks of Invariant Risk Minimizationcited by 262
- 2019Hidden Stratification Causes Clinically Meaningful Failures in Machine Learning for Medical Imagingcited by 216
- 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
- 2020Large-Scale Methods for Distributionally Robust Optimizationcited by 179
- 2022On Feature Learning in the Presence of Spurious Correlationscited by 145
- 2018Vision Meets Drones: A Challengecited by 81
- 2021Examining and Combating Spurious Features under Distribution Shiftcited by 65
- 2020Global Wheat Head Detection (GWHD) dataset: a large and diverse dataset of high resolution RGB labelled images to develop and benchmark wheat head detection methodscited by 42
- 2021Failure Modes of Domain Generalization Algorithmscited by 8
Built on by
Citing papers per year: 2026: 1.
1 paper.