Courtroom Analogy: New Perspective on Uncertainty-Aware Classification
Taeseong Yoon, Heeyoung Kim
Abstract
Single-pass uncertainty quantification (UQ) methods for classification represent uncertainty by predicting a tractable distribution over the class probability vector. While existing approaches primarily focus on enhancing the expressiveness of this distribution, they often provide limited insight into how predictive uncertainty is structured and aggregated, resulting in weak interpretability. We introduce the courtroom analogy , which conceptualizes uncertainty-aware classification as a structured debate among class-specific advocates. Each advocate forms a probabilistic opinion, and a final verdict is reached by aggregating these opinions using input-dependent plausibility weights. In this framework, each advocate's opinion is modeled as a Dirichlet distribution whose concentration parameter is decomposed into shared evidence and class-specific advocacy. This yields a structured mixture of Dirichlet distributions with semantically interpretable parameters. To instantiate this formulation, we propose Mixture of Dirichlet EXperts (MoDEX), a single-pass neural architecture that predicts the courtroom parameters, enabling efficient and expressive UQ while explicitly modeling uncertainty aggregation. We demonstrate that MoDEX enjoys strong theoretical properties and achieves state-of-the-art UQ performance across diverse benchmarks, yielding interpretable uncertainty estimates with meaningful semantics.
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 103ed5e9-2d02-4b2d-9a55-b08058b1b557Builds on21
- Long-tail learning via logit adjustmentAditya Krishna Menon, Sadeep Jayasumana, Ankit Singh Rawat, Himanshu Jain et al.ICLR 2021 · 937 citations
- Deep Evidential RegressionAlexander Amini, Wilko Schwarting, Ava Soleimany, Daniela RusNeurIPS 2020 · 777 citations
- Uncertainty Estimation Using a Single Deep Deterministic Neural NetworkJoost van Amersfoort, Lewis Smith, Yee Whye Teh, Yarin GalICML 2020 · 529 citations
- Human Uncertainty Makes Classification More RobustJoshua C. Peterson, Ruairidh M. Battleday, Thomas L. Griffiths, Olga RussakovskyICCV 2019 · 362 citations
- Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU NetworksAgustinus Kristiadi, Matthias Hein, Philipp HennigICML 2020 · 344 citations
Related papers
- Uncertainty Estimation by Flexible Evidential Deep LearningTaeseong Yoon, Heeyoung KimNeurIPS 2025 · 12 citations
- R-EDL: Relaxing Nonessential Settings of Evidential Deep LearningMengyuan Chen, Junyu Gao, Changsheng XuICLR 2024 · 18 citations
- Hyper Evidential Deep Learning to Quantify Composite Classification UncertaintyChangbin Li, Kangshuo Li, Yuzhe Ou, Lance M. Kaplan et al.ICLR 2024 · 10 citations
- Handling Ambiguity in Emotion: From Out-of-Domain Detection to Distribution EstimationWen Wu, Bo Li, Chao Zhang, Chung-Cheng Chiu et al.ACL 2024
- Building Trust in Decision with Conformalized Multi-view Deep ClassificationWei Liu, Yufei Chen, Xiaodong YueACM MM 2024 · 13 citations
