Research
Statistics under modern information constraints
Modern learning rarely has access to clean, fully labeled, centrally pooled data. Confounders are hidden, labels are imperfect, datasets are fragmented across institutions, model outputs are modified by humans, and observations often obey complex structural constraints.
I develop statistical principles for recovering reliable information under these conditions. Across causal inference, distributed learning, and modern AI, I ask what remains identifiable, what information must be validated or communicated, and which procedures are statistically optimal.
My broader interests center on how learning systems acquire, retrieve, evaluate, and retain information. This includes active learning for data-efficient supervision, retrieval-augmented methods for grounding models in external evidence, statistically principled benchmarking and evaluation of large language models, continual learning under evolving tasks, and learning-system design that avoids shortcut solutions and improves out-of-distribution reliability.
Reliable Inference from Imperfect Evidence
Causal inference, real-world evidence, and surrogate-powered inference
Can reliable conclusions be recovered when the variables we need are hidden and the labels we use may be wrong?
My work turns auxiliary information into calibrated evidence. I use historical and negative controls to expose residual confounding, and flexible tree-based methods to learn heterogeneous treatment effects when standard identifying assumptions may fail. More recently, I have extended this perspective to surrogate-powered inference, combining abundant but noisy labels with limited high-quality validation data through regularization and adaptive labeling.
The unifying goal is to make hidden bias and imperfect labels empirically diagnosable, correctable, and useful for inference—rather than simply assuming them away.
Selected work
- Surrogate-Powered Inference: Regularization and AdaptivityPreprint, 2026+
Combines validated labels and imperfect surrogates, using regularization and adaptive multiwave labeling to improve efficiency without sacrificing validity.
- Negative-Control-Calibrated Difference-in-Difference Analysesnpj Digital Medicine, 2025
Uses pre- and post-intervention negative controls to detect and adjust for time-varying unmeasured confounding, while remaining robust to partially unreliable controls.
- Difference-in-Differences Meets Tree-Based MethodsICML, 2023
Learns heterogeneous treatment effects with a splitting rule that balances observed-data fit against violations of conditional parallel trends.
- Debiased Causal TreeNeurIPS, 2022
Uses historical controls and confounding entropy to recover heterogeneous causal effects in the presence of unmeasured confounding.
Collaborative Inference without Data Pooling
Distributed inference, heterogeneous populations, and reusable statistical summaries
When data cannot move, what information should?
I study how to retain the inferential value of pooled data when observations are distributed across institutions, heterogeneous across populations, or missing in incompatible blocks. My work first showed that more aggregation can hurt: the gain from larger samples must be balanced against bias from heterogeneity, and communication structure determines which sites should borrow from one another.
My recent work shifts the focus from aggregating model parameters to sharing information-preserving statistical objects. Transfer functions can safely augment an internal study with blockwise-missing external data; distributed calibration can reproduce pooled-data treatment comparisons; and MOSAiC compresses entire local risk functions so that sites communicate once while supporting accurate, flexible downstream analyses.
Selected work
- MOSAiC: Multi-site One-Shot Aggregation of Compressed Risk FunctionsJASA, 2026
Uses tensor-train compression to communicate local risk functions once, attaining pooled-like accuracy and enabling new submodel analyses without further site queries.
- Collaborative Indirect Treatment Comparisons with Multiple Distributed Single-Arm TrialsPreprint, 2026+
Provides doubly robust, pooled-equivalent treatment comparisons across isolated single-arm trials using only two communication rounds.
- Efficient Semiparametric Inference for Distributed Data with Blockwise MissingnessPreprint, 2026+
Develops one-round, do-no-harm augmentation that can attain the semiparametric efficiency bound while scaling to many external sites.
- The Aggregation–Heterogeneity Trade-off in Federated LearningCOLT, 2023
Shows why more data can be harmful under heterogeneity and characterizes when selective, neighbor-based aggregation is minimax optimal.
Statistical Foundations for Adaptive and Verifiable AI
Generalization, knowledge retention, and statistical content provenance
How can modern learning systems remain understandable and verifiable as models, tasks, and their outputs evolve?
I use non-asymptotic theory and optimal testing to study the full lifecycle of modern learning systems. On the learning side, I investigate why overparameterized neural networks generalize and how continual-learning algorithms balance forward transfer against retention of earlier knowledge. On the verification side, our team formulates LLM watermarking as a statistical testing problem and derives detection rules with explicit efficiency and optimality guarantees.
A recurring theme is that familiar methods can fail in modern regimes: classical bias–variance intuition can miss the benefit of overparameterization, standard regularizers can be suboptimal across tasks, and additive watermark detectors lose robustness when humans edit generated text. The theory reveals these failure mechanisms and points to better procedures.
Selected work
- Robust Detection of Watermarks for Large Language Models under Human EditsJRSS-B, 2026
Introduces an adaptive truncated goodness-of-fit detector that remains optimal under substantial edits, where common sum-based rules lose robustness.
- A Statistical Framework of Watermarks for Large Language ModelsAoS, 2025
Connects pivots, false-negative efficiency, and minimax optimization to derive powerful and principled watermark detection rules.
- Nonasymptotic Theory for Two-Layer Neural NetworksPreprint, 2025+
Uses a ridge–lasso duality to explain double descent and when overparameterized networks can outperform their underparameterized counterparts.
- A Statistical Theory of Regularization-Based Continual LearningICML, 2024
Characterizes the forward–backward transfer trade-off and derives regularization schemes that match the order of an all-data oracle.
Learning Structure and Dynamics from Complex Data
Constrained geometry, dynamic graphs, and scientific machine learning
Can structure recover information that standard data representations appear to lose?
I develop methods that use geometry, sparsity, and relational dynamics to reveal interactions hidden by nonstandard observations. CARE shows that although compositional measurements obscure absolute scale, sparsity and increasing dimension can restore identifiability: in sufficiently high dimensions, its precision-matrix estimator is minimax optimal and performs as if the latent basis were observed. Related work recovers nonlinear dependence graphs from asynchronous event streams and models out-of-distribution fluid dynamics through disentangled graph ODEs.
This direction treats structure not as a complication to work around, but as information that can make otherwise difficult learning and inference problems solvable.
Selected work
- CARE: Large Precision Matrix Estimation for Compositional DataJASA, 2025
Reveals a blessing of dimensionality: high-dimensional sparsity restores identifiability and enables optimal network recovery from compositional observations.
- Temporal Point Process Graphical ModelsPreprint, 2025+
Recovers nonlinear temporal dependence structures from high-dimensional, asynchronous streams of events.
- Prometheus: Out-of-Distribution Fluid Dynamics Modeling with Disentangled Graph ODEICML, 2024
Separates invariant physical dynamics from environment-specific factors to improve out-of-distribution modeling of complex fluid systems.