Exploring uncertainty in regression neural networks for construction of prediction intervals

被引:16
|
作者
Lai, Yuandu [1 ,3 ]
Shi, Yucheng [1 ,3 ]
Han, Yahong [1 ,3 ]
Shao, Yunfeng [2 ]
Qi, Meiyu [2 ]
Li, Bingshuai [2 ]
机构
[1] Tianjin Univ, Coll Intelligence & Comp, Tianjin, Peoples R China
[2] Huawei Technol, Huawei Noahs Ark Lab, Hong Kong, Peoples R China
[3] Tianjin Univ, Tianjin Key Lab Machine Learning, Tianjin, Peoples R China
关键词
Prediction intervals; Uncertainty estimation; Neural networks; Deep learning; ASSESSMENTS;
D O I
10.1016/j.neucom.2022.01.084
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning has achieved impressive performance on many tasks in recent years. However, it has been found that it is still not enough for deep neural networks to provide only point estimates. For high-risk tasks, we need to assess the reliability of the model predictions. This requires us to quantify the uncertainty of model prediction and construct prediction intervals. One of the significant advantages of the proposed method is that it simultaneously implements point estimation and uncertainty quantification. In this paper, we explore the uncertainty in regression neural networks to construct the prediction intervals. In general, we comprehensively consider two categories of uncertainties: aleatory uncertainty and epistemic uncertainty. We design a novel loss function, which enables us to learn uncertainty without uncertainty labels. We only need to supervise the learning of regression tasks. In the process of training, the model implicitly learns aleatory uncertainty under the guidance of loss function. And that epistemic uncertainty is accounted for in the ensembled form. Our method correlates the construction of prediction intervals with uncertainty estimation. Experimental results on some publicly available datasets show that the performance of our method is competitive with other state-of-the-art methods. (c) 2022 Elsevier B.V. All rights reserved.
引用
收藏
页码:249 / 257
页数:9
相关论文
共 50 条
  • [1] Reliable prediction intervals with regression neural networks
    Papadopoulos, Harris
    Haralambous, Haris
    NEURAL NETWORKS, 2011, 24 (08) : 842 - 851
  • [2] Prediction intervals for neural networks via nonlinear regression
    De Veaux, RD
    Schumi, J
    Schweinsberg, J
    Ungar, LH
    TECHNOMETRICS, 1998, 40 (04) : 273 - 282
  • [3] On construction of prediction intervals for heteroscedastic regression
    Wu, Yun
    Xiong, Shifeng
    COMMUNICATIONS IN STATISTICS-SIMULATION AND COMPUTATION, 2024, 53 (06) : 2940 - 2965
  • [4] Constructing prediction intervals to explore uncertainty based on deep neural networks
    Yang J.
    Chen L.
    Chen H.
    Liu J.
    Han B.
    Journal of Intelligent and Fuzzy Systems, 2024, 46 (04): : 10441 - 10456
  • [5] Distributed Gaussian Granular Neural Networks Ensemble for Prediction Intervals Construction
    Sheng, Chunyang
    Wang, Haixia
    Lu, Xiao
    Zhang, Zhiguo
    Cui, Wei
    Li, Yuxia
    COMPLEXITY, 2019, 2019
  • [6] Range estimation of construction costs using neural networks with bootstrap prediction intervals
    Sonmez, Rifat
    EXPERT SYSTEMS WITH APPLICATIONS, 2011, 38 (08) : 9913 - 9917
  • [7] Prediction intervals for artificial neural networks
    Hwang, JTG
    Ding, AA
    JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 1997, 92 (438) : 748 - 757
  • [8] Integrated prediction intervals and specific value predictions for regression problems using neural networks
    Simhayev, Eli
    Katz, Gilad
    Rokach, Lior
    KNOWLEDGE-BASED SYSTEMS, 2022, 247
  • [9] Uncertainty quantification in multivariable regression for material property prediction with Bayesian neural networks
    Li, Longze
    Chang, Jiang
    Vakanski, Aleksandar
    Wang, Yachun
    Yao, Tiankai
    Xian, Min
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [10] Ozone prediction on the basis of neural networks, support vector regression and methods with uncertainty
    Hajek, Petr
    Olej, Vladimir
    ECOLOGICAL INFORMATICS, 2012, 12 : 31 - 42