Detecting Fine-Grained Differences between Image Datasets
Organizations: Technical University of Munich, Munich Center for Machine Learning (MCML) · Helmholtz Munich · Ludwig Maximilian University of Munich
Abstract
Detecting whether and in particular how two data distributions differ is a fundamental statistical problem. For example, we might be interested in which concepts (expressed in natural language) differ between two classes, the outputs of different generative models, or the same model with different prompt specifications. When working with large and complex image datasets, we intuitively think that first selecting concepts from a broad list using fine-grained patch embeddings, and then optimizing the concepts to better fit the data is an effective workflow. However, prior work instead uses MLLMs to propose concepts and image-level embeddings to select them. Therefore, we systematically compare these techniques and find that selecting concepts from broad lists can replace MLLM-proposals at obviously lower cost. MLLM inference can then be spent on optimizing selected concepts instead to achieve substantial improvements. Additionally, by designing a new benchmark that focuses on low-prevalence and multi-concept distinctions, we find that selecting nonredundant concepts instead of only the most distinguishing concepts matters, and we introduce techniques for how to effectively do this. Finally, we show how our methods address important problems like detecting spurious correlations and biases. Overall, our work presents a fundamental perspective shift on how we analyze interpretable distributional differences between datasets.
Figures & tables
| VDB-Easy | VDB-Med. | VDB-Hard | MetaShift | |||||||||
| Sim. | Sim. | Sim. | Sim. | |||||||||
| MLLM + SigLIP (PTR) | 0.981 | 0.809 | 0.549 | 0.891 | 0.873 | 0.425 | 0.815 | 0.899 | 0.331 | 0.891 | 0.601 | 0.441 |
| MLLM + FG-CLIP (PTR) | 0.983 | 0.853 | 0.523 | 0.897 | 0.907 | 0.424 | 0.816 | 0.940 | 0.353 | 0.884 | 0.663 | 0.473 |
| Distrib. Shift: U-WaX | 0.972 | 0.335 | 0.306 | 0.816 | 0.262 | 0.185 | 0.757 | 0.196 | 0.147 | 0.834 | 0.232 | 0.400 |
| Distrib. Shift: Linear | 0.969 | 0.347 | 0.319 | 0.842 | 0.307 | 0.200 | 0.752 | 0.221 | 0.157 | 0.844 | 0.239 | 0.405 |
| Caption Summary | 0.974 | 0.613 | 0.338 | 0.858 | 0.734 | 0.267 | 0.805 | 0.654 | 0.236 | 0.825 | 0.423 | 0.252 |
| LVIS | MS COCO | SUN RGB-D | DOTA | |||||||||
| MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | |
| RAM++ | 31.5 | 56.9 | 35.9 | 44.7 | 62.8 | 44.1 | 15.1 | 60.5 | 39.7 | 29.4 | 53.6 | 31.4 |
| MLLM + SigLIP (PTR) | 26.4 | 41.9 | 27.1 | 40.2 | 48.5 | 37.3 | 10.1 | 42.6 | 23.5 | 33.8 | 41.7 | 30.2 |
| MLLM + FG-CLIP (PTR) | 22.2 | 39.8 | 23.9 | 27.4 | 40.6 | 25.9 | 15.8 | 42.1 | 27.4 | 31.1 | 40.8 | 27.0 |
| + Multi-Level Masking | 27.5 | 43.1 | 27.5 | 42.4 | 49.4 | 38.2 | 20.6 | 47.7 | 37.3 | 32.4 | 40.0 | 27.6 |
| + Residual-OMP | 32.8 | 45.8 | 33.2 | 53.5 | 54.2 | 48.5 | 22.9 | 49.9 | 44.2 | 45.5 | 47.0 | 41.1 |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| VDB-Easy | VDB-Med. | VDB-Hard | MetaShift | |||||||||
| MLLM + SigLIP (PTR) | 0.978 | 0.756 | 0.549 | 0.880 | 0.807 | 0.425 | 0.806 | 0.845 | 0.331 | 0.863 | 0.561 | 0.441 |
| Distrib. Shift: U-WaX | 0.963 | 0.313 | 0.306 | 0.799 | 0.258 | 0.185 | 0.714 | 0.214 | 0.147 | 0.805 | 0.229 | 0.400 |
| Distrib. Shift: Linear | 0.968 | 0.317 | 0.319 | 0.832 | 0.297 | 0.200 | 0.716 | 0.244 | 0.157 | 0.798 | 0.233 | 0.405 |
| Caption Summary | 0.968 | 0.584 | 0.338 | 0.843 | 0.713 | 0.267 | 0.794 | 0.640 | 0.236 | 0.822 | 0.403 | 0.252 |
| Image Tagging: MLLM | 0.968 | 0.552 | 0.346 | 0.858 | 0.635 | 0.288 | 0.783 | 0.608 | 0.243 | 0.817 | 0.392 | 0.310 |
| Gemma4-31B | Qwen3.8-27B | |||||||
| Dataset | Acc | Prec | Rec | F1 | Acc | Prec | Rec | F1 |
| BDD100K | 0.784 | 0.893 | 0.676 | 0.758 | 0.799 | 0.930 | 0.649 | 0.760 |
| DOTA | 0.778 | 0.930 | 0.619 | 0.721 | 0.788 | 0.948 | 0.610 | 0.730 |
| CelebA | 0.792 | 0.821 | 0.778 | 0.762 | 0.777 | 0.815 | 0.735 | 0.736 |
| SUN RGB-D | 0.810 | 0.818 | 0.828 | 0.813 | 0.808 | 0.839 | 0.785 | 0.802 |
| MS COCO | 0.904 | 0.968 | 0.836 | 0.890 | 0.879 | 0.988 | 0.768 | 0.855 |
| LVIS | MS COCO | SUN RGB-D | DOTA | Avg. | ||||||||||||
| MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | ||
| MLLM | Top- | 27.5 | 43.1 | 27.5 | 42.4 | 49.4 | 38.2 | 20.6 | 47.7 | 37.3 | 32.4 | 40.0 | 27.6 | 30.7 | 45.1 | 32.7 |
| MMR | 32.7 | 45.7 | 32.9 | 52.9 | 54.0 | 48.2 | 22.5 | 49.5 | 43.6 | 42.5 | 45.7 | 37.6 | 37.7 | 48.7 | 40.6 | |
| mRMR | 32.3 | 45.6 | 32.9 | 51.9 | 53.9 | 47.3 | 22.7 | 49.5 | 42.9 | 40.2 | 44.7 | 34.9 | 36.8 | 48.4 | 39.5 | |
| mRMR cross-level | 30.8 | 44.9 | 32.3 | 50.4 | 52.4 | 45.5 | 21.2 | 49.0 | 42.3 | 39.5 | 44.7 | 34.7 | 35.5 | 47.8 | 38.7 | |
| Forward Selection | 29.2 | 44.0 | 29.3 | 46.1 | 50.2 | 40.7 | 21.2 | 49.1 | 41.9 | 38.3 | 42.7 | 32.6 | 33.7 | 46.5 | 36.1 | |
| R@ | Sim. | R J | R@ | Sim. | R J | R@ | Sim. | R J | R@ | Sim. | R J | R@ | Sim. | R J | |
| MLLM + FG-CLIP | 20.3 | 36.1 | 20.9 | 24.1 | 40.8 | 26.0 | 26.7 | 43.7 | 30.6 | 28.7 | 45.9 | 34.0 | 31.7 | 48.9 | 39.3 |
| + Multi-Level Masking | 25.6 | 39.7 | 25.4 | 30.7 | 45.1 | 32.7 | 34.1 | 48.4 | 38.5 | 36.6 | 50.8 | 42.7 | 40.6 | 54.3 | 49.8 |
| + Residual-OMP | 30.5 | 42.1 | 30.0 | 38.6 | 49.2 | 41.8 | 42.4 | 52.8 | 49.2 | 44.6 | 54.9 | 53.6 | 47.6 | 57.9 | 60.3 |
| + MLLM Optimization | 36.1 | 54.4 | 36.8 | 44.5 | 62.3 | 49.5 | 48.3 | 66.1 | 56.5 | 50.7 | 68.5 | 61.4 | 53.3 | 71.6 | 68.0 |
| COCA + SigLIP | 22.0 | 42.4 | 19.4 | 26.6 | 47.1 | 25.1 | 30.0 | 49.8 | 29.7 | 32.6 | 51.8 | 33.3 | 35.9 | 54.7 | 38.9 |
| LVIS | MS COCO | SUN RGB-D | DOTA | |||||||||
| MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | |
| MLLM + Optimization | 39.6 | 61.0 | 42.9 | 68.0 | 75.7 | 66.4 | 24.3 | 58.0 | 46.9 | 46.3 | 55.2 | 43.7 |
| + unexplained images | 39.4 | 60.4 | 42.5 | 67.6 | 75.2 | 65.3 | 24.2 | 57.8 | 47.1 | 46.2 | 54.9 | 43.7 |
| COCA + Optimization | 40.4 | 62.1 | 43.1 | 66.7 | 74.3 | 63.1 | 25.7 | 62.2 | 49.7 | 49.7 | 63.0 | 52.1 |
| + unexplained images | 39.8 | 61.6 | 42.5 | 65.7 | 74.0 | 62.0 | 26.0 | 62.2 | 49.7 | 48.4 | 63.3 | 51.7 |
| LVIS | MS COCO | SUN RGB-D | DOTA | |||||||||
| MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | MCC-R | Sim. | R J | |
| MLLM + FG-CLIP + Multi-Level Masking | ||||||||||||
| Published | 27.5 | 43.1 | 27.5 | 42.4 | 49.4 | 38.2 | 20.6 | 47.7 | 37.3 | 32.4 | 40.0 | 27.6 |
| 3 levels, sampled | 27.4 | 43.1 | 27.7 | 42.1 | 49.3 | 38.2 | 20.6 | 47.6 | 36.7 | 32.5 | 40.3 | 28.0 |
| 4 levels, sampled | 27.3 | 43.0 | 27.6 | 42.1 | 49.3 | 38.0 | 20.5 | 47.6 | 36.8 | 32.4 | 40.1 | 27.9 |
| COCA + FG-CLIP + Multi-Level Masking | ||||||||||||
| LVIS | MS COCO | SUN RGB-D | DOTA | |||||||||||||
| MCC-R | MCC-R | MCC-R | MCC-R | |||||||||||||
| RAM++ | 31.5 | 57.8 | 72.8 | 72.6 | 44.7 | 60.4 | 76.9 | 76.4 | 15.1 | 60.5 | 68.4 | 68.2 | 29.4 | 66.2 | 75.9 | 75.0 |
| MLLM + SigLIP | 26.4 | 54.7 | 80.6 | 68.8 | 40.2 | 57.3 | 87.6 | 73.0 | 10.1 | 58.7 | 74.8 | 66.8 | 33.8 | 70.9 | 87.4 | 74.6 |
| MLLM + FG-CLIP, no masking | 22.2 | 56.5 | 78.6 | 70.3 | 27.4 | 58.8 | 84.1 | 73.4 | 15.8 | 61.5 | 72.8 | 66.5 | 31.1 | 69.3 | 83.3 | 74.1 |
| MLLM + FG-CLIP | 27.5 | 54.9 | 80.5 | 72.2 | 42.4 | 57.5 | 87.8 | 77.6 | 20.6 | 60.8 | 75.2 | 69.9 | 32.4 | 68.8 | 83.3 | 74.3 |
| + Residual-OMP | 32.8 | 56.2 | 82.9 | 78.1 | 53.5 | 58.9 | 90.7 | 84.5 | 22.9 | 61.7 | 77.3 | 73.2 | 45.5 | 70.9 | 86.5 | 82.4 |
Distribution Shift Detection
Distribution shift detection identifies when data or model behavior changes between reference and deployment settings.
See all 63 papers