We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Quantifying Point-Prediction Uncertainty in Neural Networks via Residual Estimation with an I/O Kernel

Abstract: Neural Networks (NNs) have been extensively used for a wide spectrum of real-world regression tasks, where the goal is to predict a numerical outcome such as revenue, effectiveness, or a quantitative result. In many such tasks, the point prediction is not enough: the uncertainty (i.e. risk or confidence) of that prediction must also be estimated. Standard NNs, which are most often used in such tasks, do not provide uncertainty information. Existing approaches address this issue by combining Bayesian models with NNs, but these models are hard to implement, more expensive to train, and usually do not predict as accurately as standard NNs. In this paper, a new framework (RIO) is developed that makes it possible to estimate uncertainty in any pretrained standard NN. The behavior of the NN is captured by modeling its prediction residuals with a Gaussian Process, whose kernel includes both the NN's input and its output. The framework is evaluated in twelve real-world datasets, where it is found to (1) provide reliable estimates of uncertainty, (2) reduce the error of the point predictions, and (3) scale well to large datasets. Given that RIO can be applied to any standard NN without modifications to model architecture or training pipeline, it provides an important ingredient for building real-world NN applications.
Comments: Fixed technical typos in log marginal likelihood calculation and Lemma A.2 proof
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Journal reference: Published at ICLR 2020
Cite as: arXiv:1906.00588 [cs.LG]
  (or arXiv:1906.00588v5 [cs.LG] for this version)

Submission history

From: Xin Qiu [view email]
[v1] Mon, 3 Jun 2019 06:08:57 GMT (315kb,D)
[v2] Wed, 25 Sep 2019 13:05:22 GMT (2132kb,D)
[v3] Mon, 17 Feb 2020 02:20:00 GMT (5840kb,D)
[v4] Thu, 28 May 2020 10:51:17 GMT (5840kb,D)
[v5] Thu, 4 Jun 2020 15:26:07 GMT (5840kb,D)

Link back to: arXiv, form interface, contact.