Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces the "courtroom analogy" for uncertainty-aware classification, framing the problem as a debate between class-specific advocates whose opinions are aggregated using input-dependent plausibility weights. Each advocate's opinion is modeled as a Dirichlet distribution, decomposed into shared evidence and class-specific advocacy, resulting in a structured mixture of Dirichlet distributions. The authors instantiate this framework as Mixture of Dirichlet EXperts (MoDEX), a neural architecture that predicts courtroom parameters, achieving state-of-the-art uncertainty quantification performance with interpretable uncertainty estimates.
Uncertainty-aware classification gets a compelling new metaphor: imagine class probabilities as courtroom debates, where expert "advocates" argue their case and input data determines the jury's weighting of their opinions.
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.