Graph Convolutional Transformer: Learning the Graphical Structure of Electronic Health Records. - Edward Choi; Zhen Xu 006; Yujia Li; Michael W. Dusenberry; Gerardo Flores; Yuan Xue; Andrew M. Dai (CoRR, 2019)

Learning an Adaptive Learning Rate Schedule. - Zhen Xu 006; Andrew M. Dai; Jonas Kemp; Luke Metz (CoRR, 2019)

Deep Physiological State Space Model for Clinical Forecasting. - Yuan Xue; Denny Zhou; Nan Du 002; Andrew M. Dai; Zhen Xu 006; Kun Zhang 043; Claire Cui (CoRR, 2019)

Learning the Graphical Structure of Electronic Health Records with Graph Convolutional Transformer. - Edward Choi; Zhen Xu 006; Yujia Li; Michael Dusenberry; Gerardo Flores; Emily Xue; Andrew M. Dai (AAAI, 2020)

Flow Contrastive Estimation of Energy-Based Models. - Ruiqi Gao; Erik Nijkamp; Diederik P. Kingma; Zhen Xu 006; Andrew M. Dai; Ying Nian Wu (CVPR, 2020)

Deep State-Space Generative Model For Correlated Time-to-Event Predictions. - Yuan Xue; Denny Zhou; Nan Du 002; Andrew M. Dai; Zhen Xu 006; Kun Zhang 043; Claire Cui (KDD, 2020)

MUFASA: Multimodal Fusion Architecture Search for Electronic Health Records. - Zhen Xu 006; David R. So; Andrew M. Dai (AAAI, 2021)

BEDS-Bench: Behavior of EHR-models under Distributional Shift-A Benchmark. - Anand Avati; Martin Seneviratne; Emily Xue; Zhen Xu 006; Balaji Lakshminarayanan; Andrew M. Dai (CoRR, 2021)