The Evidence Contraction Issue in Deep Evidential Regression: Discussion and Solution

Authors

  • Yuefei Wu School of Computer Science and Technology, Xi’an Jiaotong University Shaanxi Provincial Key Laboratory of Big Data Knowledge Engineering, Xi’an Jiaotong University
  • Bin Shi School of Computer Science and Technology, Xi’an Jiaotong University Shaanxi Provincial Key Laboratory of Big Data Knowledge Engineering, Xi’an Jiaotong University
  • Bo Dong Shaanxi Provincial Key Laboratory of Big Data Knowledge Engineering, Xi’an Jiaotong University School of Distance Education, Xi’an Jiaotong University
  • Qinghua Zheng School of Computer Science and Technology, Xi’an Jiaotong University Shaanxi Provincial Key Laboratory of Big Data Knowledge Engineering, Xi’an Jiaotong University
  • Hua Wei Arizona State University

DOI:

https://doi.org/10.1609/aaai.v38i19.30172

Keywords:

General

Abstract

Deep Evidential Regression (DER) places a prior on the original Gaussian likelihood and treats learning as an evidence acquisition process to quantify uncertainty. For the validity of the evidence theory, DER requires specialized activation functions to ensure that the prior parameters remain non-negative. However, such constraints will trigger evidence contraction, causing sub-optimal performance. In this paper, we analyse DER theoretically, revealing the intrinsic limitations for sub-optimal performance: the non-negativity constraints on the Normal Inverse-Gamma (NIG) prior parameter trigger the evidence contraction under the specialized activation function, which hinders the optimization of DER performance. On this basis, we design a Non-saturating Uncertainty Regularization term, which effectively ensures that the performance is further optimized in the right direction. Experiments on real-world datasets show that our proposed approach improves the performance of DER while maintaining the ability to quantify uncertainty.

Published

2024-03-24

How to Cite

Wu, Y., Shi, B., Dong, B., Zheng, Q., & Wei, H. (2024). The Evidence Contraction Issue in Deep Evidential Regression: Discussion and Solution. Proceedings of the AAAI Conference on Artificial Intelligence, 38(19), 21726-21734. https://doi.org/10.1609/aaai.v38i19.30172

Issue

Section

AAAI Technical Track on Safe, Robust and Responsible AI Track