Uncertainty Regularized Evidential Regression

Kai Ye, Tiejin Chen, Hua Wei, Liang Zhan
1/3/2024
cs.LGcs.AI

No official code found. Best implementation: FlynnYe/UR-ERN (21 stars, last commit 2024-06-30).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

FlynnYe/UR-ERNCommunity100%

Repository for "Uncertainty Regularized Evidential Regression" published in AAAI 2024

211Python, Jupyter NotebookDec 19, 20232 years ago

Similar Papers

Cite this paper

@article{ye2024uncertainty,
  title  = {Uncertainty Regularized Evidential Regression},
  author = {Kai Ye and Tiejin Chen and Hua Wei and Liang Zhan},
  year   = {2024},
  eprint = {2401.01484},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.01484v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.01484.svg)](https://codeofpaper.com/paper/2401.01484)

Discussion