arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.01484cs.LGcs.AI
收藏

不确定性正则化证据回归

Uncertainty Regularized Evidential Regression

University of Pittsburgh(匹兹堡大学);Arizona State University(亚利桑那州立大学)

Kai Ye, Tiejin Chen, Hua Wei, Liang Zhan

AI总结:

本文分析了证据回归网络因非负值约束导致的性能局限,提出一种新颖的正则化项使其能从整个训练集学习,实验验证了该方法的有效性。

摘要:

证据回归网络(Evidential Regression Network, ERN)代表了一种将深度学习与Dempster-Shafer理论相结合以预测目标并量化相关不确定性的新方法。在基础理论的指导下,必须采用特定的激活函数来强制实现非负值,这一约束限制了模型从所有样本中学习的能力,从而损害了模型性能。本文对这一局限性进行了理论分析,并提出了一项改进以克服该问题。首先,我们定义了模型无法有效从中学习样本的区域。随后,我们深入分析了ERN并研究了这一约束。利用分析得出的见解,我们通过引入一种新颖的正则化项来解决这一局限性,使ERN能够从整个训练集中学习。我们的大量实验证实了我们的理论发现,并证明了所提解决方案的有效性。

英文摘要:

The Evidential Regression Network (ERN) represents a novel approach that integrates deep learning with Dempster-Shafer's theory to predict a target and quantify the associated uncertainty. Guided by the underlying theory, specific activation functions must be employed to enforce non-negative values, which is a constraint that compromises model performance by limiting its ability to learn from all samples. This paper provides a theoretical analysis of this limitation and introduces an improvement to overcome it. Initially, we define the region where the models can't effectively learn from the samples. Following this, we thoroughly analyze the ERN and investigate this constraint. Leveraging the insights from our analysis, we address the limitation by introducing a novel regularization term that empowers the ERN to learn from the whole training set. Our extensive experiments substantiate our theoretical findings and demonstrate the effectiveness of the proposed solution.

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

↑