Improving Reliability Estimation for Individual Numeric Predictions: A Machine Learning Approach

被引:7
作者
Adomavicius, Gediminas [1 ]
Wang, Yaqiong [2 ]
机构
[1] Univ Minnesota, Carlson Sch Management, Dept Informat & Decis Sci, Minneapolis, MN 55455 USA
[2] Santa Clara Univ, Leavey Sch Business, Informat Syst & Analyt Dept, Santa Clara, CA 95050 USA
关键词
numeric prediction; reliability of individual predictions; machine learning; NEURAL NETWORKS; CONFIDENCE; INTERVALS; MODEL;
D O I
10.1287/ijoc.2020.1019
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Numerical predictive modeling is widely used in different application domains. Although many modeling techniques have been proposed, and a number of different aggregate accuracy metrics exist for evaluating the overall performance of predictive models, other important aspects, such as the reliability (or confidence and uncertainty) of individual predictions, have been underexplored. We propose to use estimated absolute prediction error as the indicator of individual prediction reliability, which has the benefits of being intuitive and providing highly interpretable information to decision makers, as well as allowing for more precise evaluation of reliability estimation quality. As importantly, the proposed reliability indicator allows the reframing of reliability estimation itself as a canonical numeric prediction problem, which makes the proposed approach general-purpose (i.e., it can work in conjunction with any outcome prediction model), alleviates the need for distributional assumptions, and enables the use of advanced, state-of-the-art machine learning techniques to learn individual prediction reliability patterns directly from data. Extensive experimental results on multiple real-world data sets show that the proposed machine learning-based approach can significantly improve individual prediction reliability estimation as compared with a number of baselines from prior work, especially in more complex predictive scenarios.
引用
收藏
页码:503 / 521
页数:20
相关论文
共 58 条
[1]   Estimation of individual prediction reliability using the local sensitivity analysis [J].
Bosnic, Zoran ;
Kononenko, Igor .
APPLIED INTELLIGENCE, 2008, 29 (03) :187-203
[2]   Comparison of approaches for estimating reliability of individual regression predictions [J].
Bosnic, Zoran ;
Kononenko, Igor .
DATA & KNOWLEDGE ENGINEERING, 2008, 67 (03) :504-516
[3]   An overview of advances in reliability estimation of individual predictions in machine learning [J].
Bosnic, Zoran ;
Kononenko, Igor .
INTELLIGENT DATA ANALYSIS, 2009, 13 (02) :385-401
[4]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[5]  
Brier G. W., 1950, MON WEATHER REV, V78, P1, DOI [DOI 10.1175/1520-0493(1950)0782.0.CO
[6]  
2, 10.1175/1520-0493(1950)078%3C0001:vofeit%3E2.0.co
[7]  
2]
[8]   No Longer Confidential: Estimating the Confidence of Individual Regression Predictions [J].
Briesemeister, Sebastian ;
Rahnenfuehrer, Joerg ;
Kohlbacher, Oliver .
PLOS ONE, 2012, 7 (11)
[9]  
Carney J. G., 1999, IJCNN'99. International Joint Conference on Neural Networks. Proceedings (Cat. No.99CH36339), P1215, DOI 10.1109/IJCNN.1999.831133
[10]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794