Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Courtroom Analogy: New Perspective on Uncertainty-Aware Classification

About

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.

Taeseong Yoon, Heeyoung Kim• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 (test)
Accuracy92.46
882
Image ClassificationCIFAR-10--
875
Image ClassificationCIFAR-100
Accuracy75.91
357
OOD DetectionCIFAR-10 (IND) SVHN (OOD)
AUROC0.9397
138
Image ClassificationCIFAR-10-LT
Top-1 Accuracy71.53
127
OOD DetectionCIFAR-100 IND SVHN OOD
AUROC (%)84.42
81
Out-of-Distribution DetectionCIFAR-10 ID CIFAR-100 OOD--
66
OOD DetectionCIFAR100 ID TImageNet OOD
AUROC0.8022
38
Misclassification DetectionCIFAR-10--
31
Misclassification DetectionCIFAR-100--
27
Showing 10 of 32 rows

Other info

Follow for update