Junwen Bai

107 HOY RD, ITHACA, NY 14850     ·     junwen AT cs DOT cornell DOT edu

I am a PhD student in the Department of Computer Science at Cornell University, advised by Prof. Carla Gomes. I received my B.Sc in 2017 from Shanghai Jiao Tong University, where I spent four wonderful years in ACM honored Class. I am interested in the general areas of machine learning and signal processing, with research focuses on representation learning and probabilistic modeling, often under scenarios with low-supervision. My full CV can be found here.


Publications

Conferences:

Responsive image

Representation Learning for Sequence Data with Deep Autoencoding Predictive Components

Junwen Bai, Weiran Wang, Yingbo Zhou, Caiming Xiong

International Conference on Learning Representations (ICLR), 2021.

Responsive image

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 Gomes

AAAI Conference on Artificial Intelligence (AAAI), 2021.

Responsive image

Disentangled Variational Autoencoder based Multi-Label Classification with Covariance-Aware Multivariate Probit Model

Junwen Bai, Shufeng Kong, Carla Gomes

International Joint Conference on Artificial Intelligence - Pacific Rim International Conference on Artificial Intelligence (IJCAI-PRICAI), 2020. (Acceptance rate: 12.6%)

Responsive image

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, Michell Stuart, Malin Pinsky, Kathy Mills, Carla Gomes

International Joint Conference on Artificial Intelligence - Pacific Rim International Conference on Artificial Intelligence (IJCAI-PRICAI), 2020. (Acceptance rate: 12.6%)

Responsive image

SWALP: Stochastic Weight Averaging in Low-Precision Training

Guandao Yang, Tianyi Zhang, Polina Kirichenko, Junwen Bai, Andrew Wilson, Chris De Sa
International Conference on Machine Learning (ICML), 2019.
Responsive image

Imitation Refinement For X-Ray Diffraction Signal Processing

Junwen Bai, Zihang Lai, Runzhe Yang, Yexiang Xue, John Gregoire, Carla P. Gomes
International Conference on Acoustics, Speech, and Signal Processing (ICASSP), 2019.
Responsive image

An Efficient Relaxed Projection Method for Constrained Non-negative Matrix Factorization with Application to the Phase-Mapping Problem in Materials Science

Junwen Bai, Sebastian Ament, Guillaume Perez, John M. Gregoire, Carla P. Gomes
International Conference on the Integration of Constraint Programming, Artificial Intelligence, and Operations Research (CPAIOR), 2018.
Responsive image

Relaxation Methods for Constrained Matrix Factorization Problems: Solving the Phase Mapping Problem in Materials Discovery

Junwen Bai, Johan Bjorck, Yexiang Xue, Santosh K. Suram, John M. Gregoire, Carla P. Gomes
International Conference on the Integration of Constraint Programming, Artificial Intelligence, and Operations Research (CPAIOR), 2017.
Responsive image

Phase-Mapper: An AI Platform to Accelerate High Throughput Materials Discovery

Yexiang Xue, Junwen Bai, Ronan Le Bras, Brendan Rappazo, Richard Bernstein, Johan Bjorck, Liane Longpre, Santosh K. Suram, Robert B. van Dover, John Gregoire, Carla P. Gomes
AAAI Conference on Artificial Intelligence (AAAI), 2017. (Innovative Application Award)

Journals:

Responsive image

CRYSTAL: a multi-agent AI system for automated mapping of materials' crystal structures

Carla P Gomes, Junwen Bai, Yexiang Xue, Johan Björck, Brendan Rappazzo, Sebastian Ament, Richard Bernstein, Shufeng Kong, Santosh K Suram, R Bruce van Dover, John M Gregoire
In MRS Communications 9 (2) 600-608, 2019.

Responsive image

Phase Mapper: Accelerating Materials Discovery with AI

Junwen Bai, Yexiang Xue, Johan Bjorck, Ronan Le Bras, Brendan Rappazzo, Richard Bernstein, Santosh K. Suram, R. Bruce van Dover, John M. Gregoire, Carla P. Gomes
In AI Magazine 39 (1), 15-26, 2018. (Cover story)
Responsive image

Automated Phase Mapping with AgileFD and its Application to Light Absorber Discovery in the V-Mn-Nb Oxide System

Santosh K. Suram, Yexiang Xue, Junwen Bai, Ronan Le Bras, Brendan Rappazzo, Richard Bernstein, Johan Bjorck, Lan Zhou, Robert B.van Dover, Carla P. Gomes, John M. Gregoire
American Chemical Society Combinatorial Science 19(1), 37-46, 2016. (Editor's choice)

Experiences

Internships

Salesforce Research, Advisor: Weiran Wang, May 2020 - Aug 2020

Facebook Ads Ranking, Jun 2019 - Aug 2019

Professional Services

SPC: IJCAI '21

PC/reviewer: IJCAI '20, AAAI '21, IJCAI '21, ICML '21

Journal reviewer: Journal of Chemometrics and Intelligent Laboratory Systems


Education

Cornell University

PhD student, Computer Science
July 2017 - Present

Shanghai Jiao Tong University

Bachelor of Science, Computer Science
Sept 2013 - July 2017