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

Evidential Transformation Network: Turning Pretrained Models into Evidential Models for Post-hoc Uncertainty Estimation

About

Pretrained models have become standard in both vision and language, yet they typically do not provide reliable measures of confidence. Existing uncertainty estimation methods, such as deep ensembles and MC dropout, are often too computationally expensive to deploy in practice. Evidential Deep Learning (EDL) offers a more efficient alternative, but it requires models to be trained to output evidential quantities from the start, which is rarely true for pretrained networks. To enable EDL-style uncertainty estimation in pretrained models, we propose the Evidential Transformation Network (ETN), a lightweight post-hoc module that converts a pretrained predictor into an evidential model. ETN operates in logit space: it learns a sample-dependent affine transformation of the logits and interprets the transformed outputs as parameters of a Dirichlet distribution for uncertainty estimation. We evaluate ETN on image classification and large language model question-answering benchmarks under both in-distribution and out-of-distribution settings. ETN consistently improves uncertainty estimation over post-hoc baselines while preserving accuracy and adding only minimal computational overhead.

Yongchan Chun, Chanhee Park, Jeongho Yoon, Jaehyung Seo, Heuiseok Lim• 2026

Related benchmarks

TaskDatasetResultRank
Out-of-Distribution DetectionCIFAR-10
AUROC91.8
121
Out-of-Distribution DetectionImageNet--
108
Out-of-Distribution DetectionCIFAR-10 vs CIFAR-100
AUROC84.84
70
Multiple-choice Question AnsweringOBQA
Accuracy93.2
69
Multiple-choice Question AnsweringRACE
Accuracy89.48
54
Out-of-Distribution DetectionRACE to MMLU
AUROC87.57
41
Out-of-Distribution DetectionOBQA to MMLU
AUROC87.09
41
Out-of-Distribution DetectionCIFAR10 vs. SVHN
AUROC88.79
31
Out-of-Distribution DetectionImageNet-R
ROC AUC0.7942
28
Uncertainty EstimationRACE Llama-3.1-8B and Gemma-2-9B backbones (test)
AUROC91.3
24
Showing 10 of 31 rows

Other info

Follow for update