Calibrated Reliable Regression using Maximum Mean Discrepancy
Not on the map. Why?
Abstract (arXiv)
Accurate quantification of uncertainty is crucial for real-world applications of machine learning. However, modern deep neural networks still produce unreliable predictive uncertainty, often yielding over-confident predictions. In this paper, we are concerned with getting well-calibrated predictions in regression tasks. We propose the calibrated regression method using the maximum mean discrepancy by minimizing the kernel embedding measure. Theoretically, the calibration error of our method asymptotically converges to zero when the sample size is large enough. Experiments on non-trivial real datasets show that our method can produce well-calibrated and sharp prediction intervals, which outperforms the related state-of-the-art methods.
Key facts. Cited by 36 papers in the map (data through 7 Oct 2026). First posted on arXiv 18 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 7; 2023: 10; 2024: 2; 2025: 8; 2026: 4.
36 papers.
Sort:
By area:
- 2022Uncertainty Quantification with Pre-trained Language Models: A Large-Scale Empirical Analysiscited by 104
- 2021Classifier Calibration: A survey on how to assess and improve predicted class probabilitiescited by 71
- 2021Uncertainty Toolbox: an Open-Source Library for Assessing, Visualizing, and Improving Uncertainty Quantificationcited by 57
- 2023How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?cited by 14
- 2022Calibration and generalizability of probabilistic models on low-data chemical datasets with DIONYSUScited by 10
- 2022Evaluating Aleatoric Uncertainty via Conditional Generative Modelscited by 8
- 2022Parametric and Multivariate Uncertainty Calibration for Regression and Object Detectioncited by 8
- 2023Client: Cross-variable Linear Integrated Enhanced Transformer for Multivariate Long-Term Time Series Forecastingcited by 8
- 2022Estimating Regression Predictive Distributions with Sample Networkscited by 4
- 2021f-Cal: Calibrated aleatoric uncertainty estimation from neural networks for robot perceptioncited by 3
- 2023Likelihood Annealing: Fast Calibrated Uncertainty for Regressioncited by 3
- 2023Distribution-Free Model-Agnostic Regression Calibration via Nonparametric Methodscited by 3
24 more
- 2023Uncertainty Calibration for Counterfactual Propensity Estimation in Recommendationcited by 2
- 2023Parity Calibrationcited by 2
- 2024Conformal Depression Predictioncited by 2
- 2025FlowECG: Using Flow Matching to Create a More Efficient ECG Signal Generatorcited by 2
- 2021Accurate and Reliable Forecasting using Stochastic Differential Equationscited by 1
- 2022Ensemble Multi-Quantiles: Adaptively Flexible Distribution Prediction for Uncertainty Quantificationcited by 1
- 2022Improving Uncertainty Quantification of Variance Networks by Tree-Structured Learningcited by 1
- 2023Measuring and Modeling Uncertainty Degree for Monocular Depth Estimationcited by 1
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time seriescited by 1
- 2025Localization Meets Uncertainty: Uncertainty-Aware Multi-Modal Localizationcited by 1
- 2025Cooperative Variance Estimation and Bayesian Neural Networks for Disentangling Aleatoric and Epistemic Uncertaintiescited by 1
- 2025Deep Hidden Cognition Facilitates Reliable Chain-of-Thought Reasoningcited by 1
- 2025Evaluating the Quality of the Quantified Uncertainty for (Re)Calibration of Data-Driven Regression Modelscited by 1
- 2025Microstructure-based Variational Neural Networks for Robust Uncertainty Quantification in Materials Digital Twinscited by 1
- 2026Transfer Learning Through Conditional Quantile Matchingcited by 1
- 2026Pointwise Metrics Mislead: An Evaluation Protocol for Multimodal Inverse Problemscited by 1
- 2026Decision-Aligned Evaluation of Uncertainty Quantificationcited by 1
- 2021Diffusion Tensor Estimation with Uncertainty Calibrationcited by 0
- 2023A Hierarchical Bayesian Model for Deep Few-Shot Meta Learningcited by 0
- 2023Meta-learning to Calibrate Gaussian Processes with Deep Kernels for Regression Uncertainty Estimationcited by 0
- 2024Causally-Aware Unsupervised Feature Selection Learningcited by 0
- 2025Beyond Correlation: Causal Multi-View Unsupervised Feature Selection Learningcited by 0
- 2025ProTerrain: Probabilistic Physics-Informed Rough Terrain World Modelingcited by 0
- 2026Learning Context-conditioned Gaussian Overbounds for Convolution-Based Uncertainty Propagationcited by 0
- 2026Decision-Aligned Evaluation of Uncertainty Quantification25 Jun 2026
- 2026Pointwise Metrics Mislead: An Evaluation Protocol for Multimodal Inverse Problems21 May 2026
- 2026Learning Context-conditioned Gaussian Overbounds for Convolution-Based Uncertainty Propagation15 May 2026
- 2026Transfer Learning Through Conditional Quantile Matching2 Feb 2026
- 2025Microstructure-based Variational Neural Networks for Robust Uncertainty Quantification in Materials Digital Twins19 Dec 2025
- 2025ProTerrain: Probabilistic Physics-Informed Rough Terrain World Modeling22 Oct 2025
- 2025Beyond Correlation: Causal Multi-View Unsupervised Feature Selection Learning17 Sep 2025
- 2025FlowECG: Using Flow Matching to Create a More Efficient ECG Signal Generator31 Aug 2025
- 2025Evaluating the Quality of the Quantified Uncertainty for (Re)Calibration of Data-Driven Regression Models25 Aug 2025
- 2025Deep Hidden Cognition Facilitates Reliable Chain-of-Thought Reasoning14 Jul 2025
- 2025Cooperative Variance Estimation and Bayesian Neural Networks for Disentangling Aleatoric and Epistemic Uncertainties5 May 2025
- 2025Localization Meets Uncertainty: Uncertainty-Aware Multi-Modal Localization10 Apr 2025
24 more
- 2024Causally-Aware Unsupervised Feature Selection Learning16 Oct 2024
- 2024Conformal Depression Prediction29 May 2024
- 2023Meta-learning to Calibrate Gaussian Processes with Deep Kernels for Regression Uncertainty Estimation13 Dec 2023
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time series3 Dec 2023
- 2023Measuring and Modeling Uncertainty Degree for Monocular Depth Estimation19 Jul 2023
- 2023A Hierarchical Bayesian Model for Deep Few-Shot Meta Learning16 Jun 2023
- 2023Client: Cross-variable Linear Integrated Enhanced Transformer for Multivariate Long-Term Time Series Forecasting30 May 2023
- 2023Parity Calibration29 May 2023
- 2023Distribution-Free Model-Agnostic Regression Calibration via Nonparametric Methods20 May 2023
- 2023Uncertainty Calibration for Counterfactual Propensity Estimation in Recommendation23 Mar 2023
- 2023Likelihood Annealing: Fast Calibrated Uncertainty for Regression21 Feb 2023
- 2023How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?7 Feb 2023
- 2022Improving Uncertainty Quantification of Variance Networks by Tree-Structured Learning24 Dec 2022
- 2022Calibration and generalizability of probabilistic models on low-data chemical datasets with DIONYSUS3 Dec 2022
- 2022Ensemble Multi-Quantiles: Adaptively Flexible Distribution Prediction for Uncertainty Quantification26 Nov 2022
- 2022Estimating Regression Predictive Distributions with Sample Networks24 Nov 2022
- 2022Uncertainty Quantification with Pre-trained Language Models: A Large-Scale Empirical Analysis10 Oct 2022
- 2022Parametric and Multivariate Uncertainty Calibration for Regression and Object Detection4 Jul 2022
- 2022Evaluating Aleatoric Uncertainty via Conditional Generative Models9 Jun 2022
- 2021Classifier Calibration: A survey on how to assess and improve predicted class probabilities20 Dec 2021
- 2021Diffusion Tensor Estimation with Uncertainty Calibration21 Nov 2021
- 2021f-Cal: Calibrated aleatoric uncertainty estimation from neural networks for robot perception28 Sep 2021
- 2021Uncertainty Toolbox: an Open-Source Library for Assessing, Visualizing, and Improving Uncertainty Quantification21 Sep 2021
- 2021Accurate and Reliable Forecasting using Stochastic Differential Equations28 Mar 2021
- 2022Uncertainty Quantification with Pre-trained Language Models: A Large-Scale Empirical Analysiscited by 104
- 2021Classifier Calibration: A survey on how to assess and improve predicted class probabilitiescited by 71
- 2021Uncertainty Toolbox: an Open-Source Library for Assessing, Visualizing, and Improving Uncertainty Quantificationcited by 57
- 2023How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?cited by 14
- 2022Calibration and generalizability of probabilistic models on low-data chemical datasets with DIONYSUScited by 10
- 2022Evaluating Aleatoric Uncertainty via Conditional Generative Modelscited by 8
- 2022Parametric and Multivariate Uncertainty Calibration for Regression and Object Detectioncited by 8
- 2023Client: Cross-variable Linear Integrated Enhanced Transformer for Multivariate Long-Term Time Series Forecastingcited by 8
- 2022Estimating Regression Predictive Distributions with Sample Networkscited by 4
- 2021f-Cal: Calibrated aleatoric uncertainty estimation from neural networks for robot perceptioncited by 3
- 2023Likelihood Annealing: Fast Calibrated Uncertainty for Regressioncited by 3
- 2023Distribution-Free Model-Agnostic Regression Calibration via Nonparametric Methodscited by 3
12 most cited of 34 All 34 (JSON)
- 2026Decision-Aligned Evaluation of Uncertainty Quantification25 Jun 2026
- 2026Pointwise Metrics Mislead: An Evaluation Protocol for Multimodal Inverse Problems21 May 2026
- 2026Learning Context-conditioned Gaussian Overbounds for Convolution-Based Uncertainty Propagation15 May 2026
- 2026Transfer Learning Through Conditional Quantile Matching2 Feb 2026
- 2025Microstructure-based Variational Neural Networks for Robust Uncertainty Quantification in Materials Digital Twins19 Dec 2025
- 2025Beyond Correlation: Causal Multi-View Unsupervised Feature Selection Learning17 Sep 2025
- 2025FlowECG: Using Flow Matching to Create a More Efficient ECG Signal Generator31 Aug 2025
- 2025Evaluating the Quality of the Quantified Uncertainty for (Re)Calibration of Data-Driven Regression Models25 Aug 2025
- 2025Cooperative Variance Estimation and Bayesian Neural Networks for Disentangling Aleatoric and Epistemic Uncertainties5 May 2025
- 2025Localization Meets Uncertainty: Uncertainty-Aware Multi-Modal Localization10 Apr 2025
- 2024Causally-Aware Unsupervised Feature Selection Learning16 Oct 2024
- 2024Conformal Depression Prediction29 May 2024
12 newest of 34 All 34 (JSON)