← Search

Kaizheng Wang

13 accepted papers

2026

Credal Ensemble Distillation for Uncertainty Quantification

AAAI 2026technical

Deep ensembles (DE) have emerged as a powerful approach for quantifying predictive uncertainty and distinguishing its aleatoric and epistemic components, thereby enhancing model robustness and reliability. However, their high computational and memory costs during inference pose significant challenge

Cited by 0SourcePDFScholar
2026

Learning Credal Ensembles via Distributionally Robust Optimization

ICML 2026spotlight

Credal predictors are epistemic-uncertainty-aware models that produce a convex set of probabilistic predictions. They provide a principled framework for quantifying predictive epistemic uncertainty (EU) and have been shown to improve model robustness across a range of settings. However, most state-o…

Cited by 0SourceScholar
2025

A Unified Evaluation Framework for Epistemic Predictions

AISTATS 2025poster

Predictions of uncertainty-aware models are diverse, ranging from single point estimates (often averaged over prediction samples) to predictive distributions, to set-valued or credal-set representations. We propose a novel unified evaluation framework for uncertainty-aware classifiers, applicable to…

Cited by 0SourceScholar
2025

Credal Wrapper of Model Averaging for Uncertainty Estimation in Classification

ICLR 2025spotlight

This paper presents an innovative approach, called credal wrapper, to formulating a credal set representation of model averaging for Bayesian neural networks (BNNs) and deep ensembles (DEs), capable of improving uncertainty estimation in classification tasks. Given a finite collection of single pred…

Cited by 0SourcePDFScholar
2025

Random-Set Neural Networks

ICLR 2025poster

Machine learning is increasingly deployed in safety-critical domains where erroneous predictions may lead to potentially catastrophic consequences, highlighting the need for learning systems to be aware of how confident they are in their own predictions: in other words, 'to know when they do not kno…

Cited by 1SourcePDFScholar
2024

Credal Deep Ensembles for Uncertainty Quantification

NeurIPS 2024poster

This paper introduces an innovative approach to classification called Credal Deep Ensembles (CreDEs), namely, ensembles of novel Credal-Set Neural Networks (CreNets). CreNets are trained to predict a lower and an upper probability bound for each class, which, in turn, determine a convex set of proba…

Cited by 3SourcePDFScholar
2024

Model Assessment and Selection under Temporal Distribution Shift

ICML 2024poster

We investigate model assessment and selection in a changing environment, by synthesizing datasets from both the current time period and historical epochs. To tackle unknown and potentially arbitrary temporal distribution shift, we develop an adaptive rolling window approach to estimate the generaliz…

2018

Implicit Regularization in Nonconvex Statistical Estimation: Gradient Descent Converges Linearly for Phase Retrieval and Matrix Completion

ICML 2018oral

Recent years have seen a flurry of activities in designing provably efficient nonconvex optimization procedures for solving statistical estimation problems. For various problems like phase retrieval or low-rank matrix completion, state-of-the-art nonconvex procedures require proper regularization (e…

Cited by 334SourcePDFScholar