Statistical Inference & Learning

Population-Level Generative Modeling for Ranking Data

Zhaoyang Shi.

Preprint · 2026

Optimal Demixing of Nonparametric Densities

with Jianqing Fan and Zheng Tracy Ke.

Preprint · 2026

Multivariate Gaussian Approximation for Random Forest via Region-based Stabilization

with Chinmoy Bhattacharjee, Krishna Balasubramanian and Wolfgang Polonik.

Annals of Applied Probability · Under revision, 2026

Gaussian and Bootstrap Approximation for Matching-based Average Treatment Effect Estimators

with Chinmoy Bhattacharjee, Krishna Balasubramanian and Wolfgang Polonik.

Annals of Statistics · 2026

Minimax Optimal Nonsmooth Nonparametric Regression via Fractional Laplacian Eigenmaps

with Krishna Balasubramanian and Wolfgang Polonik.

UAI · 2025

Adaptive and Non-adaptive Minimax Rates for Weighted Laplacian-Eigenmap Based Nonparametric Regression

with Krishna Balasubramanian and Wolfgang Polonik.

AISTATS · 2024

A Flexible Approach for Normal Approximation of Geometric and Topological Statistics

with Krishna Balasubramanian and Wolfgang Polonik.

Bernoulli · 2024

Foundations of Generative AI

Intrinsic Associative Memory on Riemannian Manifolds: Curvature, Capacity, and Emergent Modes

with Krishna Balasubramanian.

Preprint · 2026, short version in New Frontiers in Associative Memories, ICLR, 2026.

Graph Memory: Spectral Associative Memory via Dirichlet Energy

Zhaoyang Shi.

Preprint · 2026

Dense Associative Memory with Epanechnikov Energy

with Benjamin Hoover, Krishna Balasubramanian, Dmitry Krotov and Parikshit Ram.

NeurIPS · Spotlight (Top 3%), 2025

Data Science & Decision Making

BRACE: Differential Privacy for Dense Associative Memory with LSR Energy

with Chang Qu.

Preprint · 2026

Intrinsic Associative Memory on Riemannian Manifolds: Curvature, Capacity, and Emergent Modes

with Krishna Balasubramanian.

Preprint · 2026, short version in New Frontiers in Associative Memories, ICLR, 2026.

Graph Memory: Spectral Associative Memory via Dirichlet Energy

Zhaoyang Shi.

Preprint · 2026

Population-Level Generative Modeling for Ranking Data

Zhaoyang Shi.

Preprint · 2026

LOD: Latent Objective Discovery in Heterogeneous Multi-Agent Reinforcement Learning

with Ke Sun.

Preprint · 2026

Dense Associative Memory with Epanechnikov Energy

with Benjamin Hoover, Krishna Balasubramanian, Dmitry Krotov and Parikshit Ram.

NeurIPS · Spotlight (Top 3%), 2025

The Analysis of Features Importance in Electrical Infrared Images Faults Diagnosis

with Qi Zhao, Lei Su, Ping Ling, Nannan Yan, Chunjie Gu and Zhixiong Shi.

ACM · 2018

Under Preparation