← Search

Bei Jiang

22 accepted papers

2026

A Bayesian Nonparametric Framework for Private, Fair, and Balanced Tabular Data Synthesis

ICLR 2026poster

A fundamental challenge in data synthesis is protecting the fairness and privacy of the individual, particularly in data-scarce environments where underrepresented groups are at risk of further marginalization by reproducing the biases inherent in the data modeling process. We introduce a privacy- a…

Cited by 0SourceScholar
2026

Adaptive Conformal Prediction via Mixture-of-Experts Gating Similarity

ICLR 2026poster

Prediction intervals are essential for applying machine learning models in real applications, yet most conformal prediction (CP) methods provide coverage guarantees that overlook the heterogeneity and domain knowledge that characterize modern multimodal datasets. We introduce Mixture-of-Experts Conf…

Cited by 0SourceScholar
2026

Exponential-Wrapped Mechanisms: Differential Privacy on Hadamard Manifolds Made Practical

ICLR 2026poster

We propose a general and computationally efficient framework for achieving differential privacy (DP) on Hadamard manifolds, which are complete and simply connected Riemannian manifolds with non-positive curvature. Leveraging the Cartan-Hadamard theorem, we introduce Exponential-Wrapped Laplace and G…

Cited by 0SourceScholar
2026

One-Shot Weighted Ensemble Estimation for Federated Quantile Regression: Optimal Statistical Guarantees under Heterogeneous Structured Data

ICML 2026poster

Federated Quantile Regression (FQR) has emerged as a powerful modelling paradigm for estimating conditional quantiles, offering a more comprehensive understanding of response distributions than standard conditional mean regression. However, achieving communication efficiency and optimal statistical …

Cited by 0SourceScholar
2026

RSA-CP: Efficient Conformal Prediction in Small-Sample Regimes via Random Score Alignment

ICML 2026poster

Conformal Prediction (CP) provides rigorous finite-sample coverage guarantees, yet its statistical efficiency hinges critically on the size of the calibration set. In data-scarce regimes, CP often suffers from volatile quantile estimation, leading to overly conservative and wide prediction intervals…

Cited by 0SourceScholar
2025

Advancing Fairness in Precision Medicine: A Universal Framework for Optimal Treatment Estimation in Censored Data

AISTATS 2025poster

In healthcare and precision medicine, estimating optimal treatment regimes for right-censored data while ensuring fairness across ethnic subgroups is crucial but remains underexplored. The problem presents two key challenges: measuring heterogeneous treatment effects (HTE) under fairness constraints…

Cited by 0SourceScholar
2025

Differentially Private Analysis for Binary Response Models: Optimality, Estimation, and Inference

ICML 2025poster

Randomized response (RR) mechanisms constitute a fundamental and effective technique for ensuring label differential privacy (LabelDP). However, existing RR methods primarily focus on the response labels while overlooking the influence of covariates and often do not fully address optimality. To addr…

Cited by 0SourcePDFScholar
2025

Intrinsic Benefits of Categorical Distributional Loss: Uncertainty-aware Regularized Exploration in Reinforcement Learning

NeurIPS 2025poster

The remarkable empirical performance of distributional reinforcement learning~(RL) has garnered increasing attention to understanding its theoretical advantages over classical RL. By decomposing the categorical distributional loss commonly employed in distributional RL, we find that the potential su…

Cited by 0SourceScholar
2025

Understanding Fairness and Prediction Error through Subspace Decomposition and Influence Analysis

NeurIPS 2025poster

Machine learning models have achieved widespread success but often inherit and amplify historical biases, resulting in unfair outcomes. Traditional fairness methods typically impose constraints at the prediction level, without addressing underlying biases in data representations. In this work, we pr…

Cited by 0SourceScholar
2024

Analysis of Differentially Private Synthetic Data: A Measurement Error Approach

AAAI 2024technical

Differentially private (DP) synthetic datasets have been receiving significant attention from academia, industry, and government. However, little is known about how to perform statistical inference using DP synthetic datasets. Naive approaches that do not take into account the induced uncertainty du…

Cited by 2SourcePDFScholar
2024

Debiasing with Sufficient Projection: A General Theoretical Framework for Vector Representations

NAACL 2024long

Pre-trained vector representations in natural language processing often inadvertently encode undesirable social biases. Identifying and removing unwanted biased information from vector representation is an evolving and significant challenge. Our study uniquely addresses this issue from the perspecti…

2024

Distributional Reinforcement Learning with Regularized Wasserstein Loss

NeurIPS 2024poster

The empirical success of distributional reinforcement learning (RL) highly relies on the choice of distribution divergence equipped with an appropriate distribution representation. In this paper, we propose \textit{Sinkhorn distributional RL (SinkhornDRL)}, which leverages Sinkhorn divergence—a regu…

2024

Probing Social Bias in Labor Market Text Generation by ChatGPT: A Masked Language Model Approach

NeurIPS 2024poster

As generative large language models (LLMs) such as ChatGPT gain widespread adoption in various domains, their potential to propagate and amplify social biases, particularly in high-stakes areas such as the labor market, has become a pressing concern. AI algorithms are not only widely used in the sel…

Cited by 1SourcePDFScholar
2024

Responsible Bandit Learning via Privacy-Protected Mean-Volatility Utility

AAAI 2024technical

For ensuring the safety of users by protecting the privacy, the traditional privacy-preserving bandit algorithm aiming to maximize the mean reward has been widely studied in scenarios such as online ride-hailing, advertising recommendations, and personalized healthcare. However, classical bandit le…

Cited by 1SourcePDFScholar
2024

Sample Average Approximation for Conditional Stochastic Optimization with Dependent Data

ICML 2024poster

Conditional Stochastic Optimization (CSO) is a powerful modelling paradigm for optimization under uncertainty. The existing literature on CSO is mainly based on the independence assumption of data, which shows that the solution of CSO is asymptotically consistent and enjoys a finite sample guarantee…

Cited by 0SourcePDFScholar
2023

Gaussian Differential Privacy on Riemannian Manifolds

NeurIPS 2023poster

We develop an advanced approach for extending Gaussian Differential Privacy (GDP) to general Riemannian manifolds. The concept of GDP stands out as a prominent privacy definition that strongly warrants extension to manifold settings, due to its central limit properties. By harnessing the power of th…

2023

The Sufficiency of Off-Policyness and Soft Clipping: PPO Is Still Insufficient according to an Off-Policy Measure

AAAI 2023technical

The popular Proximal Policy Optimization (PPO) algorithm approximates the solution in a clipped policy space. Does there exist better policies outside of this space? By using a novel surrogate objective that employs the sigmoid function (which provides an interesting way of exploration), we found th…

2022

Conformalized Fairness via Quantile Regression

NeurIPS 2022accept

Algorithmic fairness has received increased attention in socially sensitive domains. While rich literature on mean fairness has been established, research on quantile fairness remains sparse but vital. To fulfill great needs and advocate the significance of quantile fairness, we propose a novel fram…

2022

Identification, Amplification and Measurement: A bridge to Gaussian Differential Privacy

NeurIPS 2022accept

Gaussian differential privacy (GDP) is a single-parameter family of privacy notions that provides coherent guarantees to avoid the exposure of sensitive individual information. Despite the extra interpretability and tighter bounds under composition GDP provides, many widely used mechanisms (e.g., th…

Cited by 11SourcePDFScholar
2022

Sample Average Approximation for Stochastic Optimization with Dependent Data: Performance Guarantees and Tractability

AAAI 2022technical

Sample average approximation (SAA), a popular method for tractably solving stochastic optimization problems, enjoys strong asymptotic performance guarantees in settings with independent training samples. However, these guarantees are not known to hold generally with dependent samples, such as in onl…

Cited by 10SourcePDFScholar
2021

Damped Anderson Mixing for Deep Reinforcement Learning: Acceleration, Convergence, and Stabilization

NeurIPS 2021poster

Anderson mixing has been heuristically applied to reinforcement learning (RL) algorithms for accelerating convergence and improving the sampling efficiency of deep RL. Despite its heuristic improvement of convergence, a rigorous mathematical justification for the benefits of Anderson mixing in RL ha…

Cited by 19SourcePDFScholar