arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.08754cs.LGstat.ML

基于密度感知的证据深度学习的置信度估计

Uncertainty Estimation by Density Aware Evidential Deep Learning

Department of Industrial and Systems Engineering, KAIST, Daejeon, Republic of Korea(工业与系统工程系,韩国科学技术院)

Taeseong Yoon, Heeyoung Kim

AI总结:

针对证据深度学习在OOD检测和分类上的不足,提出密度感知证据深度学习(DAEDL),融合特征密度并改进参数化,理论性质优良,在多项任务上达到最先进性能。

摘要:

证据深度学习(EDL)在置信度估计方面取得了显著成功。然而,仍有改进空间,特别是在分布外(OOD)检测和分类任务中。EDL的OOD检测性能有限,源于其在量化不确定性时无法反映测试样本与训练数据之间的距离;而其分类性能有限,则源于其对浓度参数的参数化方式。为解决这些局限性,我们提出了一种新方法,称为密度感知证据深度学习(DAEDL)。DAEDL在预测阶段将测试样本的特征空间密度与EDL的输出相结合,同时采用一种新的参数化方法,解决了传统参数化中的问题。我们证明了DAEDL具有若干有利的理论性质。DAEDL在与置信度估计和分类相关的多种下游任务中展示了最先进的性能。

英文摘要:

Evidential deep learning (EDL) has shown remarkable success in uncertainty estimation. However, there is still room for improvement, particularly in out-of-distribution (OOD) detection and classification tasks. The limited OOD detection performance of EDL arises from its inability to reflect the distance between the testing example and training data when quantifying uncertainty, while its limited classification performance stems from its parameterization of the concentration parameters. To address these limitations, we propose a novel method called Density Aware Evidential Deep Learning (DAEDL). DAEDL integrates the feature space density of the testing example with the output of EDL during the prediction stage, while using a novel parameterization that resolves the issues in the conventional parameterization. We prove that DAEDL enjoys a number of favorable theoretical properties. DAEDL demonstrates state-of-the-art performance across diverse downstream tasks related to uncertainty estimation and classification

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑