Bai, Junwen

12 publications

NeurIPS 2023 Conditional Adapters: Parameter-Efficient Transfer Learning with Fast Inference Tao Lei, Junwen Bai, Siddhartha Brahma, Joshua Ainslie, Kenton Lee, Yanqi Zhou, Nan Du, Vincent Zhao, Yuexin Wu, Bo Li, Yu Zhang, Ming-Wei Chang

AAAI 2022 A GNN-RNN Approach for Harnessing Geospatial and Temporal Information: Application to Crop Yield Prediction Joshua Fan, Junwen Bai, Zhiyun Li, Ariel Ortiz-Bobea, Carla P. Gomes

ICML 2022 Gaussian Mixture Variational Autoencoder with Contrastive Learning for Multi-Label Classification Junwen Bai, Shufeng Kong, Carla P Gomes

NeurIPSW 2022 Xtal2DoS: Attention-Based Crystal to Sequence Learning for Density of States Prediction Junwen Bai, Yuanqi Du, Yingheng Wang, Shufeng Kong, John Gregoire, Carla P Gomes

NeurIPS 2021 Contrastively Disentangled Sequential Variational Autoencoder Junwen Bai, Weiran Wang, Carla P. Gomes

NeurIPSW 2021 Gaussian Mixture Variational Autoencoder with Contrastive Learning for Multi-Label Classification Junwen Bai, Shufeng Kong, Carla P Gomes

AAAI 2021 HOT-VAE: Learning High-Order Label Correlation for Multi-Label Classification via Attention-Based Variational Autoencoders Wenting Zhao, Shufeng Kong, Junwen Bai, Daniel Fink, Carla P. Gomes

ICLR 2021 Representation Learning for Sequence Data with Deep Autoencoding Predictive Components Junwen Bai, Weiran Wang, Yingbo Zhou, Caiming Xiong

IJCAI 2020 Deep Hurdle Networks for Zero-Inflated Multi-Target Regression: Application to Multiple Species Abundance Estimation Shufeng Kong, Junwen Bai, Jae Hee Lee, Di Chen, Andrew Allyn, Michelle Stuart, Malin Pinsky, Katherine Mills, Carla P. Gomes

IJCAI 2020 Disentangled Variational Autoencoder Based Multi-Label Classification with Covariance-Aware Multivariate Probit Model Junwen Bai, Shufeng Kong, Carla P. Gomes

ICML 2019 SWALP : Stochastic Weight Averaging in Low Precision Training Guandao Yang, Tianyi Zhang, Polina Kirichenko, Junwen Bai, Andrew Gordon Wilson, Chris De Sa

AAAI 2017 Phase-Mapper: An AI Platform to Accelerate High Throughput Materials Discovery Yexiang Xue, Junwen Bai, Ronan Le Bras, Brendan Rappazzo, Richard Bernstein, Johan Bjorck, Liane Longpre, Santosh K. Suram, Robert Bruce van Dover, John M. Gregoire, Carla P. Gomes