Overview

My work centers on causal inference and adaptive design.

• Causal inference with interference and complex dependence structures.
• Adaptive clinical trials and multi-arm bandits.
• Time-series and state-space modeling.

Publications and Manuscripts

1. Junzhe Shao, Waverley Wei, Jingshen Wang. Adaptive A/B Testing under Nonstationary Dynamics using State-Space Models, International Conference on Artificial Intelligence and Statistics (AISTATS), 2026.
2. Junzhe Shao, Aibo Gong, Juan Shen, Waverley Wei. Regularized Frameworks for Flexible and Valid Adaptive Enrichment Designs, Under review at Statistica Sinica, 2025.
3. Waverly Wei^†, Junzhe Shao^†, Rita Qiuran Lyu, Rebecca Hemono, Xinwei Ma, et al. Enhanced language models for predicting and understanding HIV care disengagement: a case study in Tanzania, npj Digital Medicine, 2026.
4. Junzhe Shao, Mingzhang Yin, Xiaoxuan Cai, Linda Valeri. Generalized Synthetic Control Method with State-space Model, short version accepted at NeurIPS 2022 Workshop for Causality for realworld impact. Oral presentation at American Causal Inference Conference (ACIC) 2023, manuscript in preparation.
5. Alton Barbehenn, Lei Shi, Junzhe Shao, et al. Rapid Biphasic Decay of Intact and Defective HIV DNA Reservoir During Acute Treated HIV Disease, Nature Communications, 2024.
6. Mona Alotaibi^†, Junzhe Shao^†, Michael W. Pauciulo, et al. Metabolomic Profiles Differentiate Scleroderma-PAH from Idiopathic PAH and Correspond with worsened Functional Capacity, Chest, 2022.

^† Co-first authors.