Publications

Topic All Causal Inference Network Interference Meta-Learner Survey Sampling Bandits & RL Permutation High-Dimensional NLP / LLM Collaborative Design-based Inference
Venue All Preprint ML Conference Journal
Sort Year ↓ Newest Year ↑ Oldest
2026 Causal Inference Network Interference Design-based Inference Social Networks
Lu, S., Shi, L. and Ding, P. (2026) Estimating within-cluster and between-cluster spillover effects in randomized saturation designs. Social Networks, 87, 24–34. [Slides]
2026 Causal Inference Survey Sampling Preprint
Shi, L., Zhang, A., Lyu, R., Hu, Z., Yu, T., Arbour, D., Feller, A., Mitra, S. and Sinha, R. (2026) BACON: Budgeted Human Calibration for Modeling and Evaluation with Multiple AI Judges.
2026 Bandits & RL Collaborative KDD 2026
Hu, Z., Shi, L., Sinha, R., Grover, J. and Arbour, D. (2026) Experimentation Accelerator: Interpretable Insights and Creative Recommendations for A/B Testing with Content-Aware ranking. KDD 2026.
2026 NLP / LLM Preprint
Qi, S., Huang, C., Yu, T., Shi, L., Wu, J., McAuley, J. and Yao, L. (2026+) ColdSkill: Semantic-Graph Skill-Skill Expansion for Item-Side Cold-Start Routing in Agent Skills.
2026 NLP / LLM EMNLP 2026 (Industry)
Hooda, R., Machcha, S., MacDonald, J., Srinivasaraghavan, L., Nijasure, A., Wang, R., Shi, L., Wu, J. and Yu, T. (2026) Multi-View Structural Interpretability of Agentic Reasoning Traces. EMNLP 2026, Industry Track. Accepted.
2025 Causal Inference Design-based Inference Preprint
Lu, X., Shi, L., Liu, H. and Ding, P. (2025+) Conditional cross-fitting for unbiased machine-learning-assisted covariate adjustment in randomized experiments.
2025 Causal Inference Network Interference Preprint
Shi, L., Lyu, Q. and Lu, S. (2025+) GAUGER: Generalized Regression Adjustment via GER for Design-Based Inference Under Interference.
2025 Causal Inference Network Interference Design-based Inference Preprint
Lu, S., Shi, L., Fang, Y., Zhang, W. and Ding, P. (2025+) Design-based causal inference in bipartite experiments. [Slides]
2025 Causal Inference Meta-Learner Preprint
Shi, L., Lu, S., Lyu, Q., Ding, P. and Vlassis, N. (2025+) TERRA: A Transformer-Enabled Recursive R-learner for Longitudinal Heterogeneous Treatment Effect Estimation.
2025 Bandits & RL NeurIPS 2025
Shi, L., Arbour, D., Addanki, R., Sinha, R. and Feller, A. (2025+) Leveraging semantic similarity for experimentation with AI-generated treatments. NeurIPS 2025.
2025 NLP / LLM Preprint
Lyu, Q., Wang, M. and Shi, L. (2025+) QUARK: Robust Retrieval under Non-Faithful Queries via Query-Anchored Aggregation.
2024 Causal Inference Design-based Inference Ann. Stat.
Shi, L., Wang, J. and Ding, P. (2024) Forward selection and post-selection inference in factorial designs. Annals of Statistics. Accepted. [Slides]
2024 Causal Inference Design-based Inference J. Causal Inf.
Shi, L. and Li, X. (2024) Some theoretical foundations on the design and analysis of randomized experiments. Journal of Causal Inference. Accepted.
2024 Permutation Preprint
Shi, L. and Ding, P. (2024+) Asymptotic theory for the quadratic assignment procedure. [Slides]
2024 Bandits & RL NeurIPS 2024
Shi, L., Wei, W. and Wang, J. (2024+) Using Surrogates in Covariate-adjusted Response-adaptive Randomized Experiments with Delayed Outcomes. NeurIPS 2024. [Slides]
2024 High-Dimensional JMLR
Shi, L., Wang, G. and Zou, C. (2024) Low-Rank Matrix Estimation in the Presence of Change-Points. Journal of Machine Learning Research. [Distinguished Student Paper Award Winner for ENAR 2023]
2024 Collaborative Nature Comms
Barbehenn, A., Shi, L., Shao, J., Hoh, R., Hartig, H.M., Pae, V., Sarvadhavabhatla, S., Donaire, S., Sheikhzadeh, C., Milush, J., Laird, G.M., et al. (2024) Rapid Biphasic Decay of Intact and Defective HIV DNA Reservoir During Acute Treated HIV Disease. Nature Communications. Accepted.
2024 Causal Inference Collaborative J. Biopharm. Stat.
Shi, L., Pang, H., Chen, C. and Zhu, J. (2024+) rdborrow: An R package for Causal Inference Incorporating External Controls in Randomized Trials with Longitudinal Outcomes. Journal of Biopharmaceutical Statistics. Accepted. [Slides]
2026 Collaborative Preprint
Barbehenn, A., Shi, L., Shao, J., Hoh, R., Hartig, H.M., Pae, V., Sarvadhavabhatla, S., Donaire, S., Sheikhzadeh, C., Savur, S., Milush, J., Laird, G.M., et al. (2026) IL-10 and Coordinated Cytokine Responses Predict Rapid HIV Reservoir Decay in Acute Treated HIV Infection. medRxiv.
2023 Bandits & RL ICML 2023
Shi, L., Wang, J. and Wu, T. (2023) Statistical inference on multi-armed bandits with delayed feedback. ICML 2023.
2023 High-Dimensional JMLR
Cui, X., Shi, L., Zhong, W. and Zou, C. (2023) Robust high-dimensional low-rank matrix estimation: optimal rate and data-adaptive tuning. Journal of Machine Learning Research.
2023 High-Dimensional Preprint
Shi, L., Yang, H. and Xue, L. (2023+) Oracle Inequalities for Sparse Principal Component Analysis based on the Linear Manifold Approximation.
2022 Causal Inference Design-based Inference Ann. Stat.
Shi, L. and Ding, P. (2022+) Berry–Esseen bounds for design-based causal inference with possibly diverging treatment levels and varying group sizes. Annals of Statistics. Accepted. [Slides]
2020 High-Dimensional Stat
Shi, L. and Zou, C. (2020) Noisy Low Rank Matrix Completion Under General Bases. Stat.
2025 Bandits & RL Preprint
Contextual Bandits with LLM-Derived Priors and Adaptive Calibration.