atlas

Research map / Learning theory and algorithms / Prediction and decision losses

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

AAAI 2024 · Learning theory and algorithms › Prediction and decision losses · approach: Distributionally robust optimization · 14 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Analyzes evidence contraction in Deep Evidential Regression caused by non-negativity activation constraints on NIG parameters, and proposes a Non-saturating Uncertainty Regularization term to keep gradients nonzero and improve regression error.

Problem

Deep Evidential Regression (DER) has higher prediction error than standard regression; its non-negativity activation constraints on NIG prior parameters cause evidence contraction that stalls optimization of the prediction.

Approach

Theoretically links evidence contraction (virtual observation nu driven to zero under activation functions) to vanishing gradients of the NLL loss w.r.t. the prediction, and adds a regularizer L_U = (y-γ)^2 · ν(α-1)/(β(ν+1)) whose gradient w.r.t. γ is non-zero everywhere to keep optimizing the prediction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add Non-saturating Uncertainty Regularization to DERDER · UCI regression benchmarkRMSE: Boston: 3.06 ( 0.16) -> 2.67(0.17); Concrete: 5.85 ( 0.15) -> 5.82(0.21)Mixed
Add Non-saturating Uncertainty Regularization to DERDER · UCI regression benchmarkNLL: Power: 2.81 ( 0.07) -> 2.58(0.01); Concrete: 3.01 ( 0.02) -> 3.11(0.03)Mixed
Add Non-saturating Uncertainty Regularization to DERDER · DavisMSE: 0.28(0.00) -> 0.26(0.01)Helped
Add Non-saturating Uncertainty Regularization to DERDER · DavisCI: 0.86(0.02) -> 0.87(0.03)Helped
Add Non-saturating Uncertainty Regularization to DERDER · DavisECE: 0.15(0.02) -> 0.14(0.10)Helped
Add Non-saturating Uncertainty Regularization to DERDER · DavisNLL: -2.34(0.42) -> -2.37(0.08)Hurt
Add Non-saturating Uncertainty Regularization to DERDER · KibaMSE: 0.19(0.00) -> 0.18(0.01)Helped
Add Non-saturating Uncertainty Regularization to DERDER · KibaECE: 0.08(0.03) -> 0.05(0.02)Helped
Add Non-saturating Uncertainty Regularization to DERDER · SST-5MSE: 0.54(0.02) -> 0.52(0.01)Helped
Add Non-saturating Uncertainty Regularization to DERDER · SST-5ECE: 0.09(0.02) -> 0.11(0.02)Hurt
Use Exp instead of ReLU activation in DER (ablation)DER · SST-5MSEHelped
Use ReLU activation instead of Exp activation in DER (ablation)DER · SST-5MSEHurt

Surprising findings

Limitations

Experimental setup

Datasets: UCI regression benchmark (Boston, Concrete, Energy, Kin8nm, Naval, Power, Protein, Wine, Yacht) Davis Kiba SST-5

Benchmarks: UCI regression benchmark Davis Kiba SST-5

Metrics: RMSE NLL MSE CI ECE

Baselines: MC dropout Vanilla DER MT-DER ReLU-DER Exp-DER Standard Regression

Builds on: Deep Evidential Regression multi-task deep evidential regression evidence misdirection regularization

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy