Papers
Research Areas
No papers match this area yet.
2026
Environment-Adaptive Covariate Selection: Learning When to Use Spurious Correlations for Out-of-Distribution Prediction
arXiv:2601.02322
Problems with Chinchilla Approach 2: Systematic Biases in IsoFLOP Parabola Fits
arXiv:2603.22339
Goal-Oriented Influence-Maximizing Data Acquisition for Learning and Optimization
arXiv:2602.19578
Identification and Estimation of the Conditional Average Treatment Effect with Nonignorable Missing Covariates, Treatment, and Outcome
arXiv:2602.19378
On the Identifiability of Regime-Switching Models with Multi-Lag Dependencies
arXiv:2601.03325
Adaptive Nonparametric Perturbations of Parametric Bayesian Models
Journal of Machine Learning Research, to appear.
Terminal-Bench: Benchmarking Agents on Hard, Realistic Tasks in Command Line Interfaces
International Conference on Learning Representations (ICLR), 2026.
Structured Flow Autoencoders: Learning Structured Probabilistic Representations with Flow Matching
International Conference on Learning Representations (ICLR), 2026.
Oral Presentation (Top 2%)
Meta-probabilistic Modeling
International Conference on Artificial Intelligence and Statistics (AISTATS), 2026.
Strategic Learning with Local Explanations as Feedback
International Conference on Artificial Intelligence and Statistics (AISTATS), 2026.
2025
Latency-Response Theory Model: Evaluating Large Language Models via Response Accuracy and Chain-of-Thought Length
arXiv:2512.07019
A Bayesian Reinforcement Learning Framework for Optimizing the BCI-utility of P300 Brain-Computer Interfaces
Annals of Applied Statistics, to appear.
Deep Generative Models: Complexity, Dimensionality, and Approximation
Journal of Machine Learning Research, to appear.
Permutative Preference Alignment from Listwise Ranking of Human Judgments
Conference on Empirical Methods in Natural Language Processing (EMNLP), 2025.
Identifying Neural Dynamics Using Interventional State Space Models
International Conference on Machine Learning (ICML), 2025.
Finding Information Quality: Counterfactual Voting Adjustment for Quality Assessment and Fairer Voting in Online Platforms with Helpfulness Evaluation
International Conference on Machine Learning (ICML), 2025.
Posterior Mean Matching: Generative Modeling through Online Bayesian Inference
International Conference on Artificial Intelligence and Statistics (AISTATS), 2025.
Accuracy on the Wrong Line: On the Pitfalls of Noisy Data for Out-of-distribution Generalisation
International Conference on Artificial Intelligence and Statistics (AISTATS), 2025.
Oral at ICML 2024 Workshop on Next Gen AI Safety
Breaking Feedback Loops in Recommender Systems with Causal Inference
ACM Transactions on Recommender Systems, to appear.
Addressing Discretization-Induced Bias in Demographic Prediction
PNAS Nexus, 4, pgaf027, 2025.
Valid Inference after Causal Discovery
Journal of the American Statistical Association, 120(550), 1127–1138, 2025.
Doubly Robust Identification of Treatment Effects from Multiple Environments
International Conference on Learning Representations (ICLR), 2025.
Let Me Grok for You: Accelerating Grokking via Embedding Transfer from a Weaker Model
International Conference on Learning Representations (ICLR), 2025.
Probabilistic Machine Learning: New Frontiers for Modeling Consumers and their Choices
International Journal of Research in Marketing, 42, 2025.
2024
From Unstructured Data to In-Context Learning: Exploring What Tasks Can Be Learned and When
Neural Information Processing Systems (NeurIPS), 2024.
Desiderata for Representation Learning: A Causal Perspective
Journal of Machine Learning Research, 25(275):1–65, 2024.
ACIC Tom Ten Have Award Honorable Mention
ICSA Junior Researcher Award
Spatiotemporal Clustering with Neyman-Scott Processes via Connections to Bayesian Nonparametric Mixture Models
Journal of the American Statistical Association, 119(547), 2382–2395, 2024.
Instrumental Variable Value Iteration for Causal Offline Reinforcement Learning
Journal of Machine Learning Research, 25(303), 1–56, 2024.
LLMs Are Prone to Fallacies in Causal Inference
Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
Causal Inference for Human-Language Model Collaboration
Conference of the North American Chapter of the Association for Computational Linguistics (NAACL), 2024.
Uncertainty Calibration for Tool-Using Language Agents
Findings of the Association for Computational Linguistics: EMNLP, 2024.
On the Identifiability of Switching Dynamical Systems
International Conference on Machine Learning (ICML), 2024.
Conformal Sensitivity Analysis for Individual Treatment Effects
Journal of the American Statistical Association, 119:545, 122–135, 2024.
Multi-Domain Causal Representation Learning via Weak Distributional Invariances
International Conference on Artificial Intelligence and Statistics (AISTATS), 2024.
Causal Fairness Assessment of Treatment Allocation with Electronic Health Records
Journal of Biomedical Informatics, 2024, 104656.
Optimization-based Causal Estimation from Heterogeneous Environments
Journal of Machine Learning Research, 25(168):1–44, 2024.
Offline Policy Evaluation and Optimization under Confounding
International Conference on Artificial Intelligence and Statistics (AISTATS), 2024.
A Causality-inspired Plus-minus Model for Player Evaluation in Team Sports
Conference on Causal Learning and Reasoning (CLeaR), 2024.
2023
Interventional Causal Representation Learning
International Conference on Machine Learning (ICML), 2023.
Oral Presentation (Top 2%)
Bidirectional Attention as a Mixture of Continuous Word Experts
Uncertainty in Artificial Intelligence (UAI), 2023.
Coherent Blending of Biophysics-Based Knowledge with Bayesian Neural Networks for Robust Protein Property Prediction
ACS Synthetic Biology, 12, 11, 3242–3251, 2023.
ACS Editors' Choice
Darwin Day Special Collection
Learning Equilibria in Matching Markets from Bandit Feedback
Journal of the ACM, 70, 3, 46, 2023.
NeurIPS 2021 Spotlight (Top 3%)
The Sample Complexity of Online Contract Design
ACM Conference on Economics and Computation (EC), 2023.
Delayed and Indirect Impacts of Link Recommendations
ACM Conference on Fairness, Accountability, and Transparency (FAccT), 2023.
On Learning Necessary and Sufficient Causal Graphs
Neural Information Processing Systems (NeurIPS), 2023.
Spotlight Presentation (Top 3%)
Recommendation Systems with Distribution-Free Reliability Guarantees
Symposium on Conformal and Probabilistic Prediction with Applications (COPA), 2023.
Alexey Chervonenkis Best Paper Award
Detecting Incidental Correlation in Multimodal Learning via Latent Variable Modeling
Transactions on Machine Learning Research (TMLR), 2023.
Adjusting Machine Learning Decisions for Equal Opportunity and Counterfactual Fairness
Transactions on Machine Learning Research (TMLR), 2023.
Team Resilience under Shock: An Empirical Analysis of GitHub Repositories during Early COVID-19 Pandemic
International AAAI Conference on Web and Social Media (ICWSM), 2023.
Clarifying Trust of Materials Property Predictions using Neural Networks with Distribution-Specific Uncertainty Quantification
Machine Learning: Science and Technology, 4, 2, 2023.
2022
Multi-Source Causal Inference Using Control Variates
Transactions on Machine Learning Research (TMLR), 2022.
Anticipating Performativity by Predicting from Predictions
Neural Information Processing Systems (NeurIPS), 2022.
Empirical Gateaux Derivatives for Causal Inference
Neural Information Processing Systems (NeurIPS), 2022.
Oral Presentation (Top 3%)
Identifiable Variational Autoencoders via Sparse Decoding
Transactions on Machine Learning Research (TMLR), 2022.
Adjusting for Indirectly Measured Confounding Using Large-scale Propensity Score
Journal of Biomedical Informatics, 104204, 2022.
Partial Identification with Noisy Covariates: A Robust Optimization Approach
Causal Learning and Reasoning (CLeaR), 2022.
Statistical and Computational Trade-offs in Variational Inference: A Case Study in Inferential Model Selection
arXiv:2207.11208
2021
Posterior Collapse and Latent Variable Non-identifiability
Neural Information Processing Systems (NeurIPS), 2021.
2020
Point Process Models for Sequence Detection in High-dimensional Neural Spike Trains
Neural Information Processing Systems (NeurIPS), 2020.
Oral Presentation (Top 1.1%)
Causal Inference for Recommender Systems
ACM Conference on Recommender Systems (RecSys), 2020.
Minimal Dispersion Approximately Balancing Weights: Asymptotic Properties and Practical Considerations
Biometrika, 107:1, 93–105, 2020.
2019
The Blessings of Multiple Causes
Journal of the American Statistical Association, 114:528, 1574–1596, 2019. (with discussion)
Frequentist Consistency of Variational Bayes
Journal of the American Statistical Association, 114:527, 1147–1161, 2019.
INFORMS Data Mining Best Paper Award
Variational Bayes under Model Misspecification
Neural Information Processing Systems (NeurIPS), 2019.
Using Embeddings to Correct for Unobserved Confounding in Networks
Neural Information Processing Systems (NeurIPS), 2019.
2018
2017
Evaluating Bayesian Models with Posterior Dispersion Indices
International Conference on Machine Learning (ICML), 2017.
2016
A Bayesian Hierarchical Model for Spatial Extremes with Multiple Durations
Computational Statistics & Data Analysis, 95, 39–56, 2016.