Timezone: »
Crucial for building trust in deep learning models for critical real-world applications is efficient and theoretically sound uncertainty quantification, a task that continues to be challenging. Useful uncertainty information is expected to have two key properties: It should be valid (guaranteeing coverage) and discriminative (more uncertain when the expected risk is high). Moreover, when combined with deep learning (DL) methods, it should be scalable and affect the DL model performance minimally. Most existing Bayesian methods lack frequentist coverage guarantees and usually affect model performance. The few available frequentist methods are rarely discriminative and/or violate coverage guarantees due to unrealistic assumptions. Moreover, many methods are expensive or require substantial modifications to the base neural network. Building upon recent advances in conformal prediction [13, 33] and leveraging the classical idea of kernel regression, we propose Locally Valid and Discriminative prediction intervals (LVD), a simple, efficient, and lightweight method to construct discriminative prediction intervals (PIs) for almost any DL model. With no assumptions on the data distribution, such PIs also offer finite-sample local coverage guarantees (contrasted to the simpler marginal coverage). We empirically verify, using diverse datasets, that besides being the only locally valid method for DL, LVD also exceeds or matches the performance (including coverage rate and prediction accuracy) of existing uncertainty quantification methods, while offering additional benefits in scalability and flexibility.
Author Information
Zhen Lin (University of Illinois, Urbana Champaign)
Shubhendu Trivedi (MIT)
Jimeng Sun (Georgia Tech)
deep learning for healthcare tensor factorization
More from the Same Authors
-
2018 Poster: MiME: Multilevel Medical Embedding of Electronic Health Records for Predictive Healthcare »
Edward Choi · Cao Xiao · Walter Stewart · Jimeng Sun -
2018 Poster: Clebsch–Gordan Nets: a Fully Fourier Space Spherical Convolutional Neural Network »
Risi Kondor · Zhen Lin · Shubhendu Trivedi -
2016 Poster: RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanism »
Edward Choi · Mohammad Taha Bahadori · Jimeng Sun · Joshua Kulas · Andy Schuetz · Walter Stewart -
2015 Poster: Time-Sensitive Recommendation From Recurrent User Activities »
Nan Du · Yichen Wang · Niao He · Jimeng Sun · Le Song -
2014 Poster: Discriminative Metric Learning by Neighborhood Gerrymandering »
Shubhendu Trivedi · David Mcallester · Greg Shakhnarovich