Exploring uncertainty in regression neural networks for construction of prediction intervals

被引:20
作者
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
相关论文
共 60 条
  • [1] Blundell C, 2015, PR MACH LEARN RES, V37, P1613
  • [2] Bojarski Mariusz, 2016, arXiv
  • [3] Breiman L, 1996, MACH LEARN, V24, P123, DOI 10.1023/A:1018054314350
  • [4] Damianou A., 2013, ARTIF INTELL, V31, P207
  • [5] Depeweg S., 2017, ARXIV PREPRINT ARXIV
  • [6] Detlefsen N.S., 2019, ARXIV PREPRINT ARXIV
  • [7] Efron B., 1993, An introduction to the bootstrap, DOI 10.1007/978-1-4899-4541-9
  • [8] Dermatologist-level classification of skin cancer with deep neural networks
    Esteva, Andre
    Kuprel, Brett
    Novoa, Roberto A.
    Ko, Justin
    Swetter, Susan M.
    Blau, Helen M.
    Thrun, Sebastian
    [J]. NATURE, 2017, 542 (7639) : 115 - +
  • [9] Gal Y, 2016, PR MACH LEARN RES, V48
  • [10] Geifman Y, 2017, ADV NEUR IN, V30