Lei BAI

J03, Electrical Engineering Building, Darlington NSW 2008, Australia. lei.bai@sydney.edu.au

Hi, I am Lei, a Postdoctoral Research Fellow at School of Electrical and Information Engineering, the University of Sydney, Australia. Currently, I am working with Associate Prof. Wanli Ouyang and Dr. Chenyu Wang. Prior to that, I received my PhD degree in Computer Science from UNSW Sydney under the supervision of Scientia Associate Prof. Lina Yao and Prof. Salil Kanhere.

My research focuses on developing robust and versatile spatial-temporal Learning algorithms for pattern discovery and knowledge mining in the multi-modal sequential data, such as video and correlated time series. My work is motivated and can be applied to various real-world applications such as Smart Cities (e.g., Intelligent Transportation), Healthcare (e.g., elder caring), Fintech, and Neuroscience.

Always welcome discussions and cooperations, feel free to drop me an email anytime you want.


News

  • One paper was accepted to Transportation Research Part C
  • Two papers were accepted to ICCV 2021
  • One paper was accepted to IEEE Transactions on Circuits and Systems for Video Technology
  • I was invited as a reviewer for ICLR2022, NeurIPS 2021, ICCV 2021, TPAMI, and TIP
  • One paper was accepted to CVPR 2021
  • One paper was accepted to IEEE Transactions on Intelligent Transportation Systems
  • One paper was accepted to Future Generation Computer Systems
  • I received the Outstanding HDR Student award as part of the UNSW Engineering Excellence Awards
  • I was awarded the 2020 Google PhD Fellowship in Machine Learning
  • Our paper on traffic forecasting was accepted to NeurIPS 2020


Publications

Conference Papers

  1. 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*);
  2. 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*);
  3. 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*);
  4. 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*);
  5. 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);
  6. 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*);
  7. 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*);
  8. 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);
  9. 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);
  10. 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);
  11. 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);
  12. 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*);
  13. 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);
  14. 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);
  15. 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);
  16. 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. 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. (Accpeted)
  2. 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. (Accpeted)
  3. 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. (Accpeted)
  4. 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, 2021. (Accpeted);
  5. 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. (Accpeted)
  6. 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. (Accepted);

Under Review

  1. 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, 2021.
  2. 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.
  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, 2021.
  4. Zhe Liu, Xianzhi Wang, Lina Yao, Lei Bai, Jake An, Ee-Peng Lim, "Face to Purchase: Predicting Consumer Choices with Structured Facial and Behavioral Traits Embedding", 2020.

Professional Services

Conference PC Member/Reviewer

  • PC Member, International Joint Conference on Artificial Intelligence (IJCAI-ECAI 2022)
  • Reviewer, The Conference on Computer Vision and Pattern Recognition (CVPR), 2022
  • Reviewer, The Ninth International Conference on Learning Representations (ICLR), 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, IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2021
  • Reviewer, IEEE Transactions on Image Processing (TIP), 2021
  • Reviewer, ACM Transactions on Multimedia, Computing, Communications, and Applications (TOMM), 2021
  • Reviewer, Expert Systems With Applications, 2021
  • Reviewer, Knowledge-Based Systems (KBS), 2021
  • Reviewer, ACM Transactions on Sensor Networks (TOSN), 2020
  • Reviewer, Future Generation Computer Systems, 2020
  • Reviewer, IEEE Intelligent Systems, 2020
  • Reviewer, Journal of Spatial Science, 2020
  • Reviewer, Artificial Intelligence in Medicine, 2019
  • Reviewer, Knowledge and Information Systems (KAIS), 2019

Organizing Committee

  • Finance Chair, The International Conference on Digital Image Computing: Techniques and Applications (DICTA), 2022

Journal Editorships



Awards

  • 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)