Lei BAI

Shanghai AI Laboratory, baisanshi@gmail.com

Hi, I am Lei, a researcher at Shanghai AI Laboratory. Currently, I am working with Prof. Wanli Ouyang. Prior to that, I was a Postdoctoral Research Fellow at the University of Sydney, Australia. I received my PhD degree in Computer Science from UNSW Sydney under the supervision of Prof. Lina Yao and Prof. Salil Kanhere.

My research interests lay in Spatial-temporal Intelligence and Resource Efficient Machine Learning (e.g., self-supervised learning), focusing on developing versatile and robust deep learning algorithms to help computers understand our world both spatially and temporally with less resources (e.g., human annotations, computation cost). The related applications including Earth System Science and Intelligent Transportation Systems. I serve as the reviewers or program committee members for a set of prestigious conferences and journals such as NeurIPS, ICLR, ICML, AAAI, CVPR, ICCV, IJCAI, KDD, ECCV, TPAMI, TKDE, TIP, TMM etc, and serve as the Web Chair for DICTA 2022 (Top Australia Pattern Recognition Conference). I am also a recipient of the 2020 Google PhD Fellowship, 2020 UNSW Engineering Excellence Award, 2021 Dean’s Award for Outstanding PhD Theses, and 2022 WAIC Yunfan Award.

Welcome discussions and cooperations, feel free to drop me an email if you want.


News

  • Two papers were accepted to ICLR 2023
  • Two papers were accepted to DASFAA 2023
  • One paper was accepted to Nature Communications
  • One paper was accepted to AAAI 2023
  • One paper was accepted to IEEE Transactions on Dependable and Secure Computing
  • One paper was accepted to IEEE Transactions on Circuits and Systems for Video Technology
  • One paper was accepted to NeurIPS 2022
  • I was awarded the 2022 WAIC Yunfan Award.
  • One paper was accepted to ICDM 2022
  • One paper was accepted to CIKM 2022
  • Five papers were accepted to ECCV 2022
  • One paper was accepted to IEEE Transactions on Image Processing
  • One paper was accepted to IEEE Transactions on Pattern Analysis and Machine Intelligence
  • Two papers were accepted to CVPR 2022
  • One paper was accepted to AAAI 2022

Publications

Conference Papers

  1. Haiyang Yang, SHIXIANG TANG, Xiaotong Li, Feng Zhu, Yizhou Wang, Meilin Chen, LEI BAI, Rui Zhao, Wanli Ouyang, "Cycle-consistent Masked AutoEncoder for Unsupervised Domain Generalization", The Eleventh International Conference on Learning Representations (ICLR), 2023. (Accepted);
  2. Zhengyang Zhou, Qihe huang, Gengyu Lin, Kuo Yang, LEI BAI, Yang Wang, "GReTo: Remedying Dynamic Graph Topology-task Discordance via Target Homophily", The Eleventh International Conference on Learning Representations (ICLR), 2023. (Accepted);
  3. Pengkun Wang, Xu Wang, Binwu Wang, Yudong Zhang, Lei Bai*, Yang Wang*, "Long-tailed Time Series Classification via Feature Space Rebalancing", The 28th International Conference on Database Systems for Advanced Applications (DASFAA), 2023. (Accepted);
  4. Binwu Wang, Yudong Zhang, Pengkun Wang, Xu Wang, Lei Bai*, Yang Wang*, "A Knowledge-driven Memory System for Traffic Flow Prediction", The 28th International Conference on Database Systems for Advanced Applications (DASFAA), 2023. (Accepted);
  5. Lintao Wang, Kun Hu, Lei Bai, Yu Ding, Wanli Ouyang, Zhiyong Wang, "Multi-Scale Control Signal-Aware Transformer for Motion Synthesis without Phase", AAAI, 2023. (Accepted);
  6. Yizhou Wang, Meilin Chen, Shixiang Tang, Feng Zhu, Haiyang Yang, Lei Bai, Rui Zhao, Yunfeng Yan, Donglian Qi, Wanli Ouyang, "Unsupervised Object Detection Pretraining with Joint Object Priors Generation and Detector Learning", NeurIPS 2022. (Accepted, CORE Rank A*);
  7. Pengkun Wang, Xu Wang, Binwu Wang, Yudong Zhang, Lei Bai, and Yang Wang, "Countering Modal Redundancy and Heterogeneity: A Self-Correcting Multimodal Fusion", The IEEE International Conference on Data Mining (ICDM), 2022. (Accepted, CORE Rank A*);
  8. Zhenyu Mao, Ziyue Li, Dedong Li, Lei Bai and Rui Zhao, "Jointly Contrastive Representation Learning on Road Network and Trajectory", ACM International Conference on Information and Knowledge Management (CIKM), 2022. (Accepted, CORE Rank A);
  9. Peixia Li, Boyu Chen, Lei Bai*, Lei Qiao, Bo Li, Qiuhong Shen, Weihao Gan, Wei Wu, Wanli Ouyang, "Backbone is All Your Need: A Simplified Architecture for Visual Object Tracking", European Conference on Computer Vision (ECCV), 2022. (Accepted, CORE Rank A*);
  10. Shixiang Tang, Feng Zhu, Lei Bai*, Rui Zhao, Chenyu Wang, Wanli Ouyang, "Unifying Vision Unsupervised Contrastive Learning from a Graph Perspective", European Conference on Computer Vision (ECCV), 2022. (Accepted, CORE Rank A*);
  11. Shixiang Tang, Feng Zhu, Lei Bai*, Rui Zhao, Wanli Ouyang, "Relative Contrastive Loss for Self-supervised Representation Learning", European Conference on Computer Vision (ECCV), 2022. (Accepted, CORE Rank A*);
  12. Yuanzheng Ci, Chen Lin, Lei Bai*, Wanli Ouyang, "Fast-MoCo: Boost Momentum-based Contrastive Learning with Combinatorial Patches", European Conference on Computer Vision (ECCV), 2022. (Accepted, CORE Rank A*);
  13. Haiyang Yang, Meilin Chen, Yizhou Wang, Shixiang Tang, Feng Zhu, Lei Bai*, Rui Zhao, Wanli Ouyang, "Domain Invariant Autoencoders for Self-supervised Learning from Multi-domains", European Conference on Computer Vision (ECCV), 2022. (Accepted, CORE Rank A*);
  14. Tao Han*, Lei Bai*(equal contribution), Junyu Gao, Qi Wang, Wanli Ouyang. "DR.VIC: Decomposition and Reasoning for Video Individual Counting", IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022. (Accepted, CORE Rank A*);
  15. Yizhou Wang, Shixiang Tang, Feng Zhu, Lei Bai*, Rui Zhao, Donglian Qi, Wanli Ouyang. "Revisiting the Transferability of Supervised Pretraining: an MLP Perspective", IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022. (Accepted, CORE Rank A*);
  16. Dongzhan Zhou, Xinchi Zhou, Di Hu, Hang Zhou, Lei Bai, Ziwei Liu, Wanli Ouyang. "SepFusion: Finding Optimal Fusion Structures for Visual Sound Separation", Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI), 2022. (Accepted, CORE Rank A*);
  17. Boyu Chen, Peixia Li, Chuming Li, Baopu Li, Lei Bai, Chen Lin, Ming Sun, Wanli Ouyang. "GLiT: Neural Architecture Search for Global and Local Image Transformer", International Conference on Computer Vision (ICCV), 2021. (Accepted, CORE Rank A*);
  18. Size Wu, Sheng Jin, Wentao Liu, Lei Bai, Chen Qian, Dong Liu, Wanli Ouyang. "Graph-Based 3D Multi-Person Pose Estimation Using Multi-View Images", International Conference on Computer Vision (ICCV), 2021. (Accepted, CORE Rank A*);
  19. Shixiang Tang, Dapeng Chen, Lei Bai, Kaijian Liu, Yixiao Ge, Wanli Ouyang. "Mutual CRF-GNN Network for Few-shot Learning", 2021 Conference on Computer Vision and Pattern Recognition (CVPR), 2021. (Accepted, CORE Rank A*);
  20. Lei Bai, Lina Yao, Can Li, Xianzhi Wang and Can Wang. "Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting", Thirty-fourth Conference on Neural Information Processing Systems (NeurIPS), 2020. (Accepted, CORE Rank A*);
  21. Can Li, Lei Bai, Wei Liu, Lina Yao and Travis S Waller. "Knowledge Adaption for Demand Prediction based on Multi-task Memory Neural Network", The 29th International Conference on Information and Knowledge Management (CIKM), 2020. (Accepted, CORE Rank A);
  22. Zhe Liu, Lina Yao, Lei Bai, Xianzhi Wang and Can Wang. "Spectrum-Guided Adversarial Disparity Learning", The 26th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 2020. (Accepted, CORE Rank A*);
  23. Lei Bai, Lina Yao, Xianzhi Wang, Salil S. Kanhere, Bin Guo, Zhiwen Yu. "Adversarial Multi-view Networks for Activity Recognition", The Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies (IMWUT & Ubicomp), 2020. (Accepted, CORE Rank A*);
  24. Zhe Liu, Lina Yao, Xianzhi Wang, Lei Bai, Jake An. "Are You a Risk Taker? Adversarial Learning of Asymmetric Cross-Domain Alignment for Risk Tolerance Prediction", The 2020 International Joint Conference on Neural Networks (IJCNN), 2020. (Accepted, CORE Rank A);
  25. Lei Bai, Lina Yao, Xianzhi Wang, Salil S. Kanhere, Yang Xiao. "Prototype Similarity Learning for Activity Recognition", The 24th Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD), 2020. (Accepted, CORE Rank A);
  26. Xuesong Wang, Lina Yao, Wei Liu, Can Li, Lei Bai, Travis Waller. "Mobility Irregularity Detection with Smart Transit Card Data", The 24th Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD), 2020. (Accepted, CORE Rank A);
  27. Can Li, Lei Bai, Wei Liu, Lina Yao, Travis Waller, "Passenger Demographic Attributes Prediction for Human-centered Public Transport", 26th International Conference on Neural Information Processing of the Asia-Pacific Neural Network (ICONIP), 2019. (Accepted, CORE Rank A);
  28. Lei Bai, Lina Yao, Salil S. Kanhere, Xianzhi Wang, Michael Sheng. "STG2Seq: Spatial-temporal Graph to Sequence Model for Multi-step Passenger Demand Forecasting", The 28th International Joint Conference on Artificial Intelligence (IJCAI), 2019. (Accepted, CORE Rank A*);
  29. Lei Bai, Lina Yao, Salil S. Kanhere, Xianzhi Wang, Wei Liu. "Spatio-Temporal Graph Convolutional and Recurrent Networks for Citywide Passenger Demand Prediction", The 28th ACM International Conference on Information and Knowledge Management (CIKM), 2019. (Accepted, CORE Rank A);
  30. Lei Bai, Lina Yao, Salil S. Kanhere, Zheng Yang, Jing Chu & X. Wang. "Passenger Demand Forecasting with Multi-Task Convolutional Recurrent Neural Networks", The 23rd Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD), 2019. (Accepted, CORE Rank A);
  31. Lei Bai, Lina Yao, Salil S. Kanhere, Xianzhi Wang, Zheng Yang, "Automatic Device Classification from Network Traffic Streams of Internet of Things", in Proceedings of 43nd IEEE conference on Local Networks (LCN), 2018. (Accepted, CORE Rank A);
  32. Lei Bai, Lina Yao, Salil S. Kanhere, "AI Creates Better City Life: Human Mobility Prediction for Smart Transportation System", UNSW Engineering Postgraduate Research Symposium, 2018, ISBN: 978-0-9953910-2-4.

Journal Papers

  1. Fenghua Ling, Jingjia Luo, Yue Li, Tao Tang, Lei Bai, Wanli Ouyang, and Toshio Yamagata, "Multi-task Machine Learning Improves Multi-Seasonal Prediction of the Indian Ocean Dipole", Nature Communications, 2023. (Accepted);
  2. Peixia Li, Boyu Chen, Lei Bai*, Lei Qiao, Bo Li, Wanli Ouyang, "SiamSampler: Video-Guided Sampling for Siamese Visual Tracking", IEEE Transactions on Circuits and Systems for Video Technology (TCSVT), 2022.
  3. Yang Xiao, Zhuolin Xing, Alex Liu, Lei Bai, Qingqi Pei, Lina Yao, "Cure-GNN: A Robust Curvature-Enhanced Graph Neural Network against Adversarial Attacks", IEEE Transactions on Dependable and Secure Computing (TDSC), 2022.
  4. Peiqin Zhuang, Yu Guo, Zhipeng Yu, Luping Zhou, Lei Bai*, Ding Liang, Zhiyong Wang, Wanli Ouyang, "Action Recognition with Motion Diversification and Dynamic Selection", IEEE Transactions on Image Processing (TIP), 2022.
  5. Lingbo Liu, Yuying Zhu, Guanbin Li, Ziyi Wu, Lei Bai, Liang Lin, "Online Metro Origin-Destination Prediction via Heterogeneous Information Aggregation", IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2022.
  6. Zhe Liu, Xianzhi Wang, Lina Yao, Jake An, Lei Bai, Ee-Peng Lim, "Face to Purchase: Predicting Consumer Choices with Structured Facial and Behavioral Traits Embedding", Knowledge-Based Systems (KBS), 2022.
  7. Can Li, Lei Bai, Wei Liu, Lina Yao, Travis Waller, "A Multi-Task Memory Network with Knowledge Adaptation for Multimodal Demand Forecasting", Transportation Research Part C, 2021.
  8. Zhenxun Yuan, Xiao Song, Lei Bai*, Zhe Wang, Wanli Ouyang, "Temporal-Channel Transformer for 3D Lidar-Based Video Object Detection for Autonomous Driving", IEEE Transactions on Circuits and Systems for Video Technology (TCSVT), 2021.
  9. Lei Bai, Lina Yao, Xianzhi Wang, Can Li, Xiang Zhang, "Deep Spatial-Temporal Sequence Modeling for Multi-Step Passenger Demand Prediction", Future Generation Computer Systems (FGCS), 2021.
  10. Can Li, Lei Bai, Wei Liu, Lina Yao, Travis Waller, "Graph Neural Network for Robust Public Transit Demand Prediction", IEEE Transactions on Intelligent Transportation Systems (TITS), 2021.
  11. Can Li, Lei Bai, Wei Liu, Lina Yao, Travis Waller, "Urban Mobility Analytics: A Deep Temporal-Spatial Product Neural Network for Demographics Prediction", Transportation Research Part C, 2020.
  12. Yang Xiao, Qingqi Pei, Lina Yao, Shui Yu, Lei Bai, Xianzhi Wang. "An Enhanced Probabilistic Fairness-Aware Group Recommendation by Incorporating Social Activeness", Journal of Network and Computer Applications, 2020.

Under Review

  1. Peixia Li, Boyu Chen, Lei Bai*, Lei Qiao, Bo Li, Xinzhu Ma, Weihao Gan, Wei Wu, Wanli Ouyang, "DNVOT: Exploring Dual-branch Network for Visual Object Tracking", IEEE Transactions on Image Processing.
  2. Yinyan Liu, Lei Bai, Jin Ma, Wei Wang, Wanli Ouyang, "Unsupervised Feature Learning for Non-Intrusive Load Monitoring", IEEE Transactions on Industrial Informatics.
  3. Pu Zhang, Lei Bai, Yuning Wang, Jianwu Fang, Jianru Xue, Nanning Zheng, Wanli Ouyang, "Towards Trajectory Forecasting from Detection", IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI).
  4. Weitao Feng, Lei Bai*, Yongqiang Yao, Weihao Gan, Wei Wu, Wanli Ouyang, "Unified Relation Learning with Similarity- and Quality-Guided Attention for Joint Detection and Tracking", TMM.
  5. Weitao Feng, Lei Bai*, Yongqiang Yao, Fengwei Yu, Wanli Ouyang, "Towards Frame Rate Agnostic Multi-Object Tracking", International Journal of Computer Vision (IJCV).
  6. Yudong Zhang, Zhe Zhao, Binwu Wang, Pengkun Wang, Lei Bai*, Yang Wang*, "Adaptive and Interactive Multi-level Spatio-Temporal Network for Traffic Forecasting", TKDE.
  7. Xu Wang, Pengkun Wang, Pengfei Gu, Binwu Wang, Lei Bai*, Yang Wang*, "Graph-Free Learning in Graph-Structured Data: A More Efficient and Accurate Spatiotemporal Learning Perspective", Proceedings of the VLDB.
  8. Qing En, Lijuan Duan, Zhaoxiang Zhang, Lei Bai, Wanli Ouyang, "Mask-Guided Feature Purification for Object Segmentation", IEEE Transactions on Pattern Analysis and Machine Intelligence, 2022.
  9. Boyu Chen, Peixia Li, Baopu Li, Chuming Li, Lei Bai, Chen Lin, Ming Sun, Junjie Yan, Wanli Ouyang, "PSViT: Better Vision Transformer via Token Pooling and Attention Sharing", International Journal of Computer Vision, 2021.

Professional Services

Conference PC Member/Reviewer

  • PC Member, SIGKDD 2023, 2023
  • Reviewer, The Conference on Computer Vision and Pattern Recognition (CVPR), 2023
  • PC Member, The 37th AAAI Conference on Artifical Intelligence (AAAI-23)
  • Reviewer, The Eleventh International Conference on Learning Representations (ICLR), 2023
  • Reviewer, Thirty-Sixth Conference on Neural Information Processing Systems (NeurIPS), 2022
  • Reviewer, European Conference on Computer Vision, 2022
  • PC Member, SIGKDD 2022, 2022
  • Reviewer, International Conference on Machine Learning (ICML), 2022
  • PC Member, International Joint Conference on Artificial Intelligence (IJCAI-ECAI 2022)
  • Reviewer, The Conference on Computer Vision and Pattern Recognition (CVPR), 2022
  • Reviewer, The Tenth International Conference on Learning Representations (ICLR), 2022
  • PC Member, The ACM Web Conference (WWW), The First Workshop on Graph Learning, 2022
  • Reviewer, Thirty-fifth Conference on Neural Information Processing Systems (NeurIPS), 2021
  • Reviewer, International Conference on Computer Vision (ICCV), 2021
  • Reviewer, The Conference on Computer Vision and Pattern Recognition (CVPR), 2021
  • PC Member, International Joint Conference on Artificial Intelligence (IJCAI-21)
  • PC Member, The 35th AAAI Conference on Artifical Intelligence (AAAI-21)
  • Subreviewer, CIKM 2020, 2020
  • Subreviewer, KDD 2020, 2020
  • Subreviewer, International Conference on Mobility, Sensing and Networking (MSN), 2020
  • Reviewer, Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies (Ubicomp), 2019
  • Subreviewer, IEEE International Conference on Data Mining (ICDM), 2019
  • Subreviewer, the 28th ACM International Conference on Information and Knowledge Management (CIKM), 2019
  • Subreviewer, ACM MULTIMEDIA (MM), 2019
  • Subreviewer, International Joint Conference on Artificial Intelligence (IJCAI), 2019
  • Reviewer, the 20th IEEE International Symposium on A World of Wireless, Mobile and Multimedia Networks (WoWMoM), 2019
  • Subreviewer, the 23rd Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD), 2018
  • Subreviewer, the 24th International Conference on Database Systems for Advanced Applications (DASFAA), 2018
  • Reviewer, ACM/IEEE Conference on Internet of Things Design and Implementation, 2018

Journal Reviewer

  • Reviewer, Transactions on Machine Learning Research (TMLR)
  • Reviewer, IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)
  • Reviewer, IEEE Transactions on Image Processing (TIP)
  • Reviewer, IEEE Transactions on Multimedia (TMM)
  • Reviewer, IEEE Transactions on Neural Networks and Learning Systems (TNNLS)
  • Reviewer, IEEE Transactions on Cybernetics (TCYB)
  • Reviewer, Transportation Research Part C
  • Reviewer, ACM Transactions on Multimedia, Computing, Communications, and Applications (TOMM)
  • Reviewer, ACM Transactions on Sensor Networks (TOSN)
  • Reviewer, Neurocomputing
  • Reviewer, Expert Systems With Applications
  • Reviewer, Knowledge-Based Systems (KBS)
  • Reviewer, Future Generation Computer Systems
  • Reviewer, IEEE Intelligent Systems
  • Reviewer, Journal of Spatial Science
  • Reviewer, Artificial Intelligence in Medicine
  • Reviewer, Knowledge and Information Systems (KAIS)

Organizing Committee


Journal Editorships


Awards

  • World Artificial Intelligence Conference (WAIC) Yunfan Award, 2022
  • Dean’s Award for Outstanding PhD Theses, 2021
  • UNSW Engineering Excellence Award (Outstanding HDR Student category), 2020
  • Google PhD Fellowship, 2020
  • IJCAI 2019 Travel Grant, 2019
  • ICL-TUM Global Fellows Programme, 2019 (the only representive from the UNSW Engineering Faculty)
  • UNSW Postgraduate Research Student Support (PRSS) Scholarship, 2019
  • PAKDD/MCEA Travel Grant, 2019
  • Tuition Fee Scholarship, University of New South Wales, from 2017 to 2021
  • Excellent Postgraduate Student, Xidian University, 2017
  • The Second-Class University Scholarship, Xidian University, 2016
  • The First Award, China Information Security Technology Innovation and Entrepreneurship Competition, 2015
  • The First-Class University Scholarship, Xidian University, 2014

Supervision

  • Shan Wang, Master Project Student, Jun.2019 - Dec.2019. (co-supervised with Dr. Lina Yao)