A Comprehensive Evaluation of Multi-task Learning and Multi-task Pre-training on EHR Time-series Data. - Matthew B. A. McDermott; Bret Nestor; Evan Kim; Wancong Zhang; Anna Goldenberg; Peter Szolovits; Marzyeh Ghassemi (CoRR, 2020)

A comprehensive EHR timeseries pre-training benchmark. - Matthew B. A. McDermott; Bret Nestor; Evan Kim; Wancong Zhang; Anna Goldenberg; Peter Szolovits; Marzyeh Ghassemi (CHIL, 2021)

MixUp Training Leads to Reduced Overfitting and Improved Calibration for the Transformer Architecture. - Wancong Zhang; Ieshan Vaidya (CoRR, 2021)

AttnOD: An Attention-Based OD Prediction Model with Adaptive Graph Convolution. - Wancong Zhang; Gang Wang 058; Xu Liu; Tongyu Zhu (ICONIP (12), 2023)

Light-weight Probing of Unsupervised Representations for Reinforcement Learning. - Wancong Zhang; Anthony GX-Chen; Vlad Sobal; Yann LeCun; Nicolas Carion (RLJ, 2024)

Bidirectional Mendelian randomization and colocalization analysis of gut microbiota on lipid profile. - Yu Cui; Yanzhu Chen; Mengting Hu; He Zhou; Jiarui Guo; Qijia Wang; Zaihua Xu; Liyun Chen; Wancong Zhang; Shijie Tang (Comput. Biol. Chem., 2025)

Learning from Reward-Free Offline Data: A Case for Planning with Latent Dynamics Models. - Vlad Sobal; Wancong Zhang; Kynghyun Cho; Randall Balestriero; Tim G. J. Rudner; Yann LeCun (CoRR, 2025)