Learning Credal Ensembles via Distributionally Robust Optimization
Kaizheng Wang, Ghifari Adam Faza, Fabio Cuzzolin, Siu Lun Chau, David Moens, Hans Hallez
Abstract
Credal predictors are epistemic-uncertainty-aware models that produce a convex set of probabilistic predictions. They provide a principled framework for quantifying predictive epistemic uncertainty (EU) and have been shown to improve model robustness across a range of settings. However, most state-of-the-art (SOTA) methods primarily define EU as disagreement induced by random training initializations, which mainly reflects sensitivity to optimization randomness rather than uncertainty from more substantive sources. In response, we formulate EU as disagreement between models trained under different degrees of relaxation of the i.i.d. assumption between the training and test distributions. Building on this idea, we propose CreDRO , which learns an ensemble of plausible models via distributionally robust optimization. As a result, CreDRO captures EU arising not only from training randomness but also from informative disagreement due to potential train–test distribution shifts. Empirically, CreDRO consistently outperforms SOTA credal approaches on downstream tasks, including out-of-distribution detection on extensive benchmarks and selective classification in medical settings.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext dbd0aad4-9c9c-469e-80b4-1cd01968decfBuilds on21
- BatchEnsemble: an Alternative Approach to Efficient Ensemble and Lifelong LearningYeming Wen, Dustin Tran, Jimmy BaICLR 2020 · 569 citations
- Learning from Failure: De-biasing Classifier from Biased ClassifierJun Hyun Nam, Hyuntak Cha, Sungsoo Ahn, Jaeho Lee et al.NeurIPS 2020 · 428 citations
- Fairness without Demographics through Adversarially Reweighted LearningPreethi Lahoti, Alex Beutel, Jilin Chen, Kang Lee et al.NeurIPS 2020 · 406 citations
- Large-Scale Methods for Distributionally Robust OptimizationDaniel Levy, Yair Carmon, John C. Duchi, Aaron SidfordNeurIPS 2020 · 281 citations
- Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-CountsBertrand Charpentier, Daniel Zügner, Stephan GünnemannNeurIPS 2020 · 263 citations
Related papers
- Credal Deep Ensembles for Uncertainty QuantificationKaizheng Wang, Fabio Cuzzolin, Shireen Kudukkil Manchingal, Keivan Shariatmadar et al.NeurIPS 2024 · 37 citations
- Credal Ensemble Distillation for Uncertainty QuantificationKaizheng Wang, Fabio Cuzzolin, David Moens, Hans HallezAAAI 2026
- Multi-Expert Distributionally Robust Optimization for Out-of-Distribution GeneralizationJinyong Jeong, Hyungu Kahng, Seoung Bum KimNeurIPS 2025 · 6 citations
- Efficient Credal Prediction through DecalibrationPaul Hofman, Timo Löhr, Maximilian Muschalik, Yusuf Sale et al.ICLR 2026 · 1 citation
- Credal Concept Bottleneck Models for Epistemic-Aleatoric Uncertainty DecompositionTanmoy Mukherjee, Thomas Bailleux, Pierre Marquis, Zied BouraouiACL 2026
