← Search

Shengjia Zhao

24 accepted papers

2026

Multi-dimensional Neural Decoding with Orthogonal Representations for Brain-Computer Interfaces

AAAI 2026technical

Current brain-computer interfaces primarily decode single motor variables, limiting natural control requiring simultaneous multi-dimensional extraction. We introduce Multi-dimensional Neural Decoding (MND), a task that simultaneously extracts multiple motor variables (direction, position, velocity,

Cited by 0SourcePDFScholar
2024

Online Distribution Shift Detection via Recency Prediction

ICRA 2024poster

When deploying modern machine learning-enabled robotic systems in high-stakes applications, detecting distribution shift is critical. However, most existing methods for detecting distribution shift are not well-suited to robotics settings, where data often arrives in a streaming fashion and may be v…

Cited by 8SourceScholar
2022

Comparing Distributions by Measuring Differences that Affect Decision Making

ICLR 2022oral

Measuring the discrepancy between two probability distributions is a fundamental problem in machine learning and statistics. We propose a new class of discrepancies based on the optimal loss for a decision task -- two distributions are different if the optimal decision loss is higher on their mixtur…

Cited by 34SourcePDFScholar
2022

Generalizing Bayesian Optimization with Decision-theoretic Entropies

NeurIPS 2022accept

Bayesian optimization (BO) is a popular method for efficiently inferring optima of an expensive black-box function via a sequence of queries. Existing information-theoretic BO procedures aim to make queries that most reduce the uncertainty about optima, where the uncertainty is captured by Shannon e…

Cited by 14SourcePDFScholar
2022

Local calibration: metrics and recalibration

UAI 2022poster

Probabilistic classifiers output confidence scores along with their predictions, and these confidence scores should be calibrated, i.e., they should reflect the reliability of the prediction. Confidence scores that minimize standard metrics such as the expected calibration error (ECE) accurately mea…

Cited by 23SourcePDFScholar
2021

Calibrating Predictions to Decisions: A Novel Approach to Multi-Class Calibration

NeurIPS 2021poster

When facing uncertainty, decision-makers want predictions they can trust. A machine learning provider can convey confidence to decision-makers by guaranteeing their predictions are distribution calibrated--- amongst the inputs that receive a predicted vector of class probabilities q, the actual dist…

Cited by 85SourcePDFScholar
2021

Improved Autoregressive Modeling with Distribution Smoothing

ICLR 2021oral

While autoregressive models excel at image compression, their sample quality is often lacking. Although not realistic, generated images often have high likelihood according to the model, resembling the case of adversarial examples. Inspired by a successful adversarial defense method, we incorporate…

Cited by 23SourcePDFScholar
2021

Right Decisions from Wrong Predictions: A Mechanism Design Alternative to Individual Calibration

AISTATS 2021poster

Decision makers often need to rely on imperfect probabilistic forecasts. While average performance metrics are typically available, it is difficult to assess the quality of individual forecasts and the corresponding utilities. To convey confidence about individual predictions to decision-makers, we…

2020

A Framework for Sample Efficient Interval Estimation with Control Variates

AISTATS 2020poster

We consider the problem of estimating confidence intervals for the mean of a random variable, where the goal is to produce the smallest possible interval for a given number of samples. While minimax optimal algorithms are known for this problem in the general case, improved performance is possible u…

2020

A Theory of Usable Information under Computational Constraints

ICLR 2020talk

We propose a new framework for reasoning about information in complex systems. Our foundation is based on a variational extension of Shannon’s information theory that takes into account the modeling power and computational constraints of the observer. The resulting predictive V-information encompass…

Cited by 186SourcecodeScholar
2020

Permutation Invariant Graph Generation via Score-Based Generative Modeling

AISTATS 2020poster

Learning generative models for graph-structured data is challenging because graphs are discrete, combinatorial, and the underlying data distribution is invariant to the ordering of nodes. However, most of the existing generative models for graphs are not invariant to the chosen ordering, which might…

2019

Adaptive Hashing for Model Counting

UAI 2019poster

Randomized hashing algorithms have seen recent success in providing bounds on the model count of a propositional formula. These methods repeatedly check the satisfiability of a formula subject to increasingly stringent random constraints. Key to these approaches is the choice of a fixed family of…

2019

Learning Controllable Fair Representations

AISTATS 2019poster

Learning data representations that are transferable and are fair with respect to certain protected attributes is crucial to reducing unfair decisions while preserving the utility of the data. We propose an information-theoretically motivated objective for learning maximally expressive representation…

2019

Learning Neural PDE Solvers with Convergence Guarantees

ICLR 2019poster

Partial differential equations (PDEs) are widely used across the physical and computational sciences. Decades of research and engineering went into designing fast iterative solution methods. Existing solvers are general purpose, but may be sub-optimal for specific classes of problems. In contrast to…

Cited by 157SourcePDFScholar
2018

Bias and Generalization in Deep Generative Models: An Empirical Study

NeurIPS 2018spotlight

In high dimensional settings, density estimation algorithms rely crucially on their inductive bias. Despite recent empirical success, the inductive bias of deep generative models is not well understood. In this paper we propose a framework to systematically investigate bias and generalization in dee…

2016

Adaptive Concentration Inequalities for Sequential Decision Problems

NeurIPS 2016poster

A key challenge in sequential decision problems is to determine how many samples are needed for an agent to make reliable decisions with good probabilistic guarantees. We introduce Hoeffding-like concentration inequalities that hold for a random, adaptively chosen number of samples. Our inequaliti…

Cited by 59SourcePDFScholar