Papers

Research Areas
2026
S. Zuo and Y. Wang
Environment-Adaptive Covariate Selection: Learning When to Use Spurious Correlations for Out-of-Distribution Prediction
arXiv:2601.02322
W. Zhang, Y. Wang, and Y. Gu
Discrete Causal Representation Learning
arXiv:2603.25017
E. Czech, Z. Xu, Y. Elmatad, Y. Wang, and W. Held
Problems with Chinchilla Approach 2: Systematic Biases in IsoFLOP Parabola Fits
arXiv:2603.22339
W. Yao, B. Dumitrascu, B.R. Goldsmith, and Y. Wang
Goal-Oriented Influence-Maximizing Data Acquisition for Learning and Optimization
arXiv:2602.19578
S. Zuo, Y. Wang, and F. Yang
Identification and Estimation of the Conditional Average Treatment Effect with Nonignorable Missing Covariates, Treatment, and Outcome
arXiv:2602.19378
J. Ren, Y. Wang, and B. Huang
Causal Representation Meets Stochastic Modeling under Generic Geometry
arXiv:2602.05033
C. Balsells-Rodas, T. Matsui, P.A.M. Mediano, Y. Wang, and Y. Li
On the Identifiability of Regime-Switching Models with Multi-Lag Dependencies
arXiv:2601.03325
B. Wu*, E.N. Weinstein*, S. Salehi, Y. Wang, and D.M. Blei
Adaptive Nonparametric Perturbations of Parametric Bayesian Models
Journal of Machine Learning Research, to appear.
M.A. Merrill, A.G. Shaw, et al.
Terminal-Bench: Benchmarking Agents on Hard, Realistic Tasks in Command Line Interfaces
International Conference on Learning Representations (ICLR), 2026.
Y. Xu, Y. Wang, and X.L. Nguyen
Structured Flow Autoencoders: Learning Structured Probabilistic Representations with Flow Matching
International Conference on Learning Representations (ICLR), 2026.
Oral Presentation (Top 2%)
K. Zhang and Y. Wang
Meta-probabilistic Modeling
International Conference on Artificial Intelligence and Statistics (AISTATS), 2026.
K.Q.H. Vo, S.L. Chau, M. Kato, Y. Wang, and K. Muandet
Strategic Learning with Local Explanations as Feedback
International Conference on Artificial Intelligence and Statistics (AISTATS), 2026.
2025
Z. Xu, J. Liu, Y. Wang, and Y. Gu
Latency-Response Theory Model: Evaluating Large Language Models via Response Accuracy and Chain-of-Thought Length
arXiv:2512.07019
L. Wu, M. Yin, Y. Wang, J.P. Cunningham, and D.M. Blei
Bayesian Invariance Modeling of Multi-Environment Data
arXiv:2506.22675
B. Zhao, Y. Wang, J.E. Huggins, and J. Kang
A Bayesian Reinforcement Learning Framework for Optimizing the BCI-utility of P300 Brain-Computer Interfaces
Annals of Applied Statistics, to appear.
K. Wang, H. Niu, Y. Wang, and D. Li
Deep Generative Models: Complexity, Dimensionality, and Approximation
Journal of Machine Learning Research, to appear.
Y. Zhao, Y. Wang, and M. Yin
Permutative Preference Alignment from Listwise Ranking of Human Judgments
Conference on Empirical Methods in Natural Language Processing (EMNLP), 2025.
A. Nejatbakhsh and Y. Wang
Identifying Neural Dynamics Using Interventional State Space Models
International Conference on Machine Learning (ICML), 2025.
C. Liu, Y. Wang, and M. Lee
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.
B. Zhang, Y. Wang, and P.S. Dhillon
Policy Learning with a Natural Language Action Space: A Causal Approach
arXiv:2502.17538
K.C. Wibisono and Y. Wang
Exponential Family Attention
arXiv:2501.16790
S. Salazar, M. Kucer, Y. Wang, E. Casleton, and D.M. Blei
Posterior Mean Matching: Generative Modeling through Online Bayesian Inference
International Conference on Artificial Intelligence and Statistics (AISTATS), 2025.
A. Sanyal, Y. Hu, Y. Yu, Y.A. Ma, Y. Wang, and B. Schoelkopf
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
K. Krauth, Y. Wang, and M.I. Jordan
Breaking Feedback Loops in Recommender Systems with Causal Inference
ACM Transactions on Recommender Systems, to appear.
E. Dong, A. Schein, Y. Wang, and N. Garg
Addressing Discretization-Induced Bias in Demographic Prediction
PNAS Nexus, 4, pgaf027, 2025.
P. Gradu*, T. Zrnic*, Y. Wang, and M.I. Jordan
Valid Inference after Causal Discovery
Journal of the American Statistical Association, 120(550), 1127–1138, 2025.
P. De Bartolomeis, J. Kostin, J. Abad, Y. Wang, and F. Yang
Doubly Robust Identification of Treatment Effects from Multiple Environments
International Conference on Learning Representations (ICLR), 2025.
Z. Xu, Z. Ni, Y. Wang*, and W. Hu*
Let Me Grok for You: Accelerating Grokking via Embedding Transfer from a Weaker Model
International Conference on Learning Representations (ICLR), 2025.
R. Dew, N. Padilla, L.E. Luo, S. Oblander, A. Ansari, K. Boughanmi, M. Braun, F.M. Feinberg, J. Liu, T. Otter, L. Tian, Y. Wang, and M. Yin
Probabilistic Machine Learning: New Frontiers for Modeling Consumers and their Choices
International Journal of Research in Marketing, 42, 2025.
G. Loaiza-Ganem, V. Villecroze, and Y. Wang
Deep Ensembles Secretly Perform Empirical Bayes
arXiv:2501.17917
2024
K.C. Wibisono and Y. Wang
From Unstructured Data to In-Context Learning: Exploring What Tasks Can Be Learned and When
Neural Information Processing Systems (NeurIPS), 2024.
Y. Wang and M.I. Jordan
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
Y. Wang*, A. Degleris*, A.H. Williams, and S.W. Linderman
Spatiotemporal Clustering with Neyman-Scott Processes via Connections to Bayesian Nonparametric Mixture Models
Journal of the American Statistical Association, 119(547), 2382–2395, 2024.
L. Liao*, Z. Fu*, Z. Yang, Y. Wang, D. Ma, M. Kolar, and Z. Wang
Instrumental Variable Value Iteration for Causal Offline Reinforcement Learning
Journal of Machine Learning Research, 25(303), 1–56, 2024.
S.J. Yang, Y. Wang, and K.Z. Lin
LCL: Contrastive Learning for Lineage Barcoded scRNA-seq Data
bioRxiv:2024.10.28.620670
N. Joshi, A. Saparov, Y. Wang, and H. He
LLMs Are Prone to Fallacies in Causal Inference
Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
B. Zhang, Y. Wang, and P.S. Dhillon
Causal Inference for Human-Language Model Collaboration
Conference of the North American Chapter of the Association for Computational Linguistics (NAACL), 2024.
H. Liu, Z.-Y. Dou, Y. Wang, N. Peng, and Y. Yue
Uncertainty Calibration for Tool-Using Language Agents
Findings of the Association for Computational Linguistics: EMNLP, 2024.
C. Balsells-Rodas, Y. Wang, and Y. Li
On the Identifiability of Switching Dynamical Systems
International Conference on Machine Learning (ICML), 2024.
M. Yin, C. Shi, Y. Wang, and D.M. Blei
Conformal Sensitivity Analysis for Individual Treatment Effects
Journal of the American Statistical Association, 119:545, 122–135, 2024.
K. Ahuja, A. Mansouri, and Y. Wang
Multi-Domain Causal Representation Learning via Weak Distributional Invariances
International Conference on Artificial Intelligence and Statistics (AISTATS), 2024.
L. Zhang, L.R. Richter, Y. Wang, A. Ostropolets, N. Elhadad, D.M. Blei, and G. Hripcsak
Causal Fairness Assessment of Treatment Allocation with Electronic Health Records
Journal of Biomedical Informatics, 2024, 104656.
M. Yin, Y. Wang, and D.M. Blei
Optimization-based Causal Estimation from Heterogeneous Environments
Journal of Machine Learning Research, 25(168):1–44, 2024.
C. Kausik*, K. Tan*, Y. Lu*, M. Makar, Y. Wang, and A. Tewari
Offline Policy Evaluation and Optimization under Confounding
International Conference on Artificial Intelligence and Statistics (AISTATS), 2024.
C. De Bacco, Y. Wang, and D.M. Blei
A Causality-inspired Plus-minus Model for Player Evaluation in Team Sports
Conference on Causal Learning and Reasoning (CLeaR), 2024.
2023
K. Ahuja, D. Mahajan, Y. Wang, and Y. Bengio
Interventional Causal Representation Learning
International Conference on Machine Learning (ICML), 2023.
Oral Presentation (Top 2%)
K.C. Wibisono and Y. Wang
Bidirectional Attention as a Mixture of Continuous Word Experts
Uncertainty in Artificial Intelligence (UAI), 2023.
H. Nisonoff, Y. Wang, and J. Listgarten
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
M. Jagadeesan*, A. Wei*, Y. Wang, M.I. Jordan, and J. Steinhardt
Learning Equilibria in Matching Markets from Bandit Feedback
Journal of the ACM, 70, 3, 46, 2023.
NeurIPS 2021 Spotlight (Top 3%)
B. Zhu, S. Bates, Z. Yang, Y. Wang, J. Jiao, and M.I. Jordan
The Sample Complexity of Online Contract Design
ACM Conference on Economics and Computation (EC), 2023.
H. Zhang, S. Lu, Y. Wang, and M. Curmei
Delayed and Indirect Impacts of Link Recommendations
ACM Conference on Fairness, Accountability, and Transparency (FAccT), 2023.
H. Cai, Y. Wang, M.I. Jordan, and R. Song
On Learning Necessary and Sufficient Causal Graphs
Neural Information Processing Systems (NeurIPS), 2023.
Spotlight Presentation (Top 3%)
A.N. Angelopoulos*, K. Krauth*, S. Bates, Y. Wang, and M.I. Jordan
Recommendation Systems with Distribution-Free Reliability Guarantees
Symposium on Conformal and Probabilistic Prediction with Applications (COPA), 2023.
Alexey Chervonenkis Best Paper Award
T. Makino, Y. Wang, K.J. Geras, and K. Cho
Detecting Incidental Correlation in Multimodal Learning via Latent Variable Modeling
Transactions on Machine Learning Research (TMLR), 2023.
Y. Wang, D. Sridhar, and D.M. Blei
Adjusting Machine Learning Decisions for Equal Opportunity and Counterfactual Fairness
Transactions on Machine Learning Research (TMLR), 2023.
X. Lu, W. Ai, Y. Wang, and Q. Mei
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.
Y. Wang and J.R. Zubizarreta
Large Sample Properties of Matching for Balance
Statistica Sinica, 33, 3, 2023.
C.J. Gruich, V. Madhavan, Y. Wang, and B.R. Goldsmith
Clarifying Trust of Materials Property Predictions using Neural Networks with Distribution-Specific Uncertainty Quantification
Machine Learning: Science and Technology, 4, 2, 2023.
2022
W. Guo*, S. Wang*, P. Ding, Y. Wang, and M.I. Jordan
Multi-Source Causal Inference Using Control Variates
Transactions on Machine Learning Research (TMLR), 2022.
C. Mendler-Dünner, F. Ding, and Y. Wang
Anticipating Performativity by Predicting from Predictions
Neural Information Processing Systems (NeurIPS), 2022.
M.I. Jordan*, Y. Wang*, and A. Zhou*
Empirical Gateaux Derivatives for Causal Inference
Neural Information Processing Systems (NeurIPS), 2022.
Oral Presentation (Top 3%)
G.E. Moran, D. Sridhar, Y. Wang, and D.M. Blei
Identifiable Variational Autoencoders via Sparse Decoding
Transactions on Machine Learning Research (TMLR), 2022.
L. Zhang, Y. Wang, M. Schuemie, D.M. Blei, and G. Hripcsak
Adjusting for Indirectly Measured Confounding Using Large-scale Propensity Score
Journal of Biomedical Informatics, 104204, 2022.
W. Guo, M. Yin, Y. Wang, and M.I. Jordan
Partial Identification with Noisy Covariates: A Robust Optimization Approach
Causal Learning and Reasoning (CLeaR), 2022.
K. Bhatia*, N.L. Kuang*, Y.-A. Ma*, and Y. Wang*
Statistical and Computational Trade-offs in Variational Inference: A Case Study in Inferential Model Selection
arXiv:2207.11208
2021
Y. Wang, D.M. Blei, and J.P. Cunningham
Posterior Collapse and Latent Variable Non-identifiability
Neural Information Processing Systems (NeurIPS), 2021.
Y. Wang and D.M. Blei
A Proxy Variable View of Shared Confounding
International Conference on Machine Learning (ICML), 2021.
2020
W. Tansey, Y. Wang, R. Rabadan, and D.M. Blei
Double Empirical Bayes Testing
International Statistical Review, 88:S91–S113, 2020.
A. Williams, A. Degleris, Y. Wang, and S.W. Linderman
Point Process Models for Sequence Detection in High-dimensional Neural Spike Trains
Neural Information Processing Systems (NeurIPS), 2020.
Oral Presentation (Top 1.1%)
Y. Wang, D. Liang, L. Charlin, and D.M. Blei
Causal Inference for Recommender Systems
ACM Conference on Recommender Systems (RecSys), 2020.
Y. Wang and J.R. Zubizarreta
Minimal Dispersion Approximately Balancing Weights: Asymptotic Properties and Practical Considerations
Biometrika, 107:1, 93–105, 2020.
Y. Wang and D.M. Blei
Towards Clarifying the Theory of the Deconfounder
arXiv:2003.04948
2019
Y. Wang and D.M. Blei
The Blessings of Multiple Causes
Journal of the American Statistical Association, 114:528, 1574–1596, 2019. (with discussion)
Y. Wang and D.M. Blei
Frequentist Consistency of Variational Bayes
Journal of the American Statistical Association, 114:527, 1147–1161, 2019.
INFORMS Data Mining Best Paper Award
Y. Wang and D.M. Blei
Variational Bayes under Model Misspecification
Neural Information Processing Systems (NeurIPS), 2019.
V. Veitch, Y. Wang, and D.M. Blei
Using Embeddings to Correct for Unobserved Confounding in Networks
Neural Information Processing Systems (NeurIPS), 2019.
Y. Wang, A.C. Miller, and D.M. Blei
Comment: Variational Autoencoders as Empirical Bayes
Statistical Science, 34:2, 229–233, 2019.
L. Zhang, Y. Wang, A. Ostropolets, J.J. Mulgrave, D.M. Blei, and G. Hripcsak
The Medical Deconfounder: Assessing Treatment Effect with Electronic Health Records
Machine Learning for Healthcare, 2019.
2018
W. Tansey, Y. Wang, D.M. Blei, and R. Rabadan
Black Box FDR
International Conference on Machine Learning (ICML), 2018.
2017
A. Kucukelbir, Y. Wang, and D.M. Blei
Evaluating Bayesian Models with Posterior Dispersion Indices
International Conference on Machine Learning (ICML), 2017.
Y. Wang, A. Kucukelbir, and D.M. Blei
Robust Probabilistic Modeling with Bayesian Data Reweighting
International Conference on Machine Learning (ICML), 2017.
ICSA Young Researcher Award
2016
Y. Wang and M.K.P. So
A Bayesian Hierarchical Model for Spatial Extremes with Multiple Durations
Computational Statistics & Data Analysis, 95, 39–56, 2016.