Lei BAI

Shanghai AI Laboratory, baisanshi@gmail.com

Hi, I am Lei, a Research Scientist at Shanghai AI Laboratory. I am now leading the AI for Science Group, aiming at developing the next-generation AI techniques that are capable of achieving autonomous discovery.

Prior to that, I was a Postdoctoral Research Fellow at the University of Sydney, Australia, working with Prof. Wanli Ouyang. I received my PhD degree in Computer Science from the University of New South Wales, Australia, under the supervision of Prof. Lina Yao and Prof. Salil Kanhere.

My research interests lay in Science Foundatation Models, Multi-agent Systems, and their applications in autonomous scientific discovery. I serve as an Area Chair for NeurIPS, Associate Editor for Pattern Recognition and IEEE Open Journal of the Computer Society, and reviewers for a set of prestigious conferences and journals such as NeurIPS, ICLR, ICML, AAAI, CVPR, ICCV, IJCAI, KDD, ECCV, TPAMI, TKDE, TIP, TMM etc. 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.

If you are interested in a research scientist, research engineer, or a PhD position at the Shanghai AI Lab, feel free to drop me an email.


News

  • Four papers were accepted to ICCV 2025
  • Two papers were accepted to ACL 2025
  • Invited to serve as an Area Chair for NeurIPS 2025
  • Two papers were accepted to ICML 2025
  • One paper was accepted to Nature COMMUNICATIONS EARTH & ENVIRONMENT
  • We are organizing an ICML 2025 workshop about Multi-agent System, feel free to submit a paper.
  • Four papers were accepted to CVPR 2025
  • One paper was accepted to TPAMI
  • Three papers were accepted to ICLR 2025
  • One paper was accepted to AAAI 2025
  • Three papers were accepted to NeurIPS 2024
  • One paper was accepted to Science China Earth Sciences
  • One paper was accepted to npj Climate and Atmospheric Science
  • One paper was accepted to ECCV 2024
  • Three papers were accepted to ICML 2024
  • We received the 2024 best paper award of IEEE Transactions on Circuits and Systems for Video Technology
  • One paper was accepted to Nature Machine Intelligence
  • Two paper were accepted to Advances in Atmospheric Sciences
  • Two paper were accepted to IEEE Transactions on Intelligent Transportation Systems
  • One paper was accepted to AISTATS 2024
  • Three papers were accepted to AAAI 2024

Publications

Equal contribution+, Corresponding author*

Preprients

  1. Zhou et.al, "Scientists' First Exam: Probing Cognitive Abilities of MLLM via Perception, Understanding, and Reasoning", arxiv, 2025.
  2. Zhang et.al, "OriGene: A Self-Evolving Virtual Disease Biologist Automating Therapeutic Target Discovery", arxiv, 2025.
  3. Liu et.al, "Ml-agent: Reinforcing llm agents for autonomous machine learning engineering", arxiv, 2025.
  4. Li et.al, "LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents", arxiv, 2025.
  5. Cui et.al, "The entropy mechanism of reinforcement learning for reasoning language models", arxiv, 2025.
  6. Yuan et.al, "MME-Reasoning: A Comprehensive Benchmark for Logical Reasoning in MLLMs", arxiv, 2025.
  7. Zhang et.al, "The Avengers: A Simple Recipe for Uniting Smaller Language Models to Challenge Proprietary Giants", arxiv, 2025.
  8. Zhang et.al, "NovelSeek: When Agent Becomes the Scientist--Building Closed-Loop System from Hypothesis to Verification", arxiv, 2025.
  9. Kang et.al, "Viki-r: Coordinating embodied multi-agent cooperation via reinforcement learning", arxiv, 2025.
  10. Zhang et.al, "EvoFlow: Evolving Diverse Agentic Workflows On The Fly", arxiv, 2025.
  11. Peng et.al, "Chimera: Improving generalist model with domain-specific experts", arxiv, 2024.
  12. Z Guo, P Lyu, F Ling, JJ Luo, N Boers, W Ouyang, L Bai, "ORCA: A Global Ocean Emulator for Multi-year to Decadal Predictions", arxiv, 2024.
  13. Tao Han, Song Guo, Fenghua Ling, Kang Chen, Junchao Gong, Jingjia Luo, Junxia Gu, Kan Dai, Wanli Ouyang, Lei Bai*, "FengWu-GHR: Learning the Kilometer-scale Medium-range Global Weather Forecasting", arxiv, 2024.;

Conference Papers

  1. Kang et.al, "RoboFactory: Exploring Embodied Agent Collaboration with Compositional Constraints", ICCV, 2025.
  2. Han et.al, "InfGen: A Resolution-Agnostic Paradigm for Scalable Image Synthesis", ICCV, 2025.
  3. Yang et.al, "Towards Physically Plausible Video Generation via VLM Planning", ICCV, 2025.
  4. Han et.al, "VA-MoE: Variables-Adaptive Mixture of Experts for Incremental Weather Forecasting", ICCV, 2025.
  5. Zhang et.al, "Multi-agent Architecture Search via Agentic Supernet", ICML 2025 (Oral).
  6. Qin et.al, "WorldSimBench: Towards Video Generation Models as World Simulators", ICML, 2025.
  7. Yan et.al, "SURVEYFORGE : On the Outline Heuristics, Memory-Driven Generation, and Multi-dimensional Evaluation for Automated Survey Writing", ACL 2025 (main).
  8. Yuan et.al, "Dolphin: Moving Towards Closed-loop Auto-research through Thinking, Practice, and Feedback", ACL 2025 (main).
  9. Wanghan Xu, Fenghua Ling, Wenlong Zhang, Tao Han, Hao Chen, Wanli Ouyang, Lei Bai*, " Generalizing Weather Forecast to Fine-grained Temporal Scales via Physics-AI Hybrid Modeling", NeurIPS, 2024.;
  10. Kun Chen, Lei Bai*, Fenghua Ling, Peng Ye, Tao Chen*, Kang Chen, Tao Han, Wanli Ouyang, " Towards an End-to-End Artificial Intelligence Driven Global Weather Forecasting System", NeurIPS, 2024.;
  11. Xiufeng Song, Xiaohong Liu, Xiao Guo, Jiache Zhang, Qirui Li, Lei Bai, Xiaoming Liu, Guangtao Zhai, " On Learning Multi-Modal Forgery Representation for Diffusion Generated Video Detection", NeurIPS, 2024.;
  12. Zidong Wang, Zeyu Lu, Di Huang, Tong He, Xihui Liu, Wanli Ouyang, Lei Bai*, " PredBench: Benchmarking Spatio-Temporal Prediction across Diverse Disciplines", ECCV, 2024.;
  13. Zeyu Lu, Zidong Wang, Di Huang, Chengyue Wu, Xihui Liu, Wanli Ouyang, Lei Bai*, "FiT: Flexible Vision Transformer for Diffusion Model", ICML, 2024 (spotlight).;
  14. Junchao Gong, Lei Bai*, Peng Ye, Wanghan Xu, Na Liu, Jianhua Dai, Xiaokang Yang, Wanli Ouyang, "FiT: CasCast: Skillful High-resolution Precipitation Nowcasting via Cascaded Modelling", ICML, 2024.;
  15. Yi Xiao, Lei Bai*, Wei Xue*, Kang Chen, Tao Han, Xiangjun Tian, Wanli Ouyang, "FENGWU-4DVAR: COUPLING THE DATA-DRIVEN WEATHER FORECASTING MODEL WITH 4D VARIATIONAL ASSIMILATION", ICML, 2024.;
  16. Zhishuai Li, Yunhao Nie, Ziyue Li, Lei Bai, Yisheng Lv, Rui Zhao, "Non-Neighbors Also Matter to Kriging: A New Contrastive-Prototypical Learning", AISTATS, 2024.
  17. Binwu Wang, Pengkun Wang, Yudong Zhang, Xu Wang, Zhengyang Zhou, Lei Bai*, Yang Wang*, "Towards Streaming Spatial-Temporal Graph Learning: A Decoupled Perspective", AAAI, 2024.
  18. Zhuqiang Lu, Kun Hu, Chaoyue Wang, Lei Bai, Zhiyong Wang, "Autoregressive Omni-Aware Outpainting for Open-Vocabulary 360-Degree Image Generation", AAAI, 2024.
  19. Yaqi Zhang, Di Huang, Bin Liu, Shixiang, Yan Lu, Lu Chen, Lei Bai, Qi Chu, Nenghai Yu, Wanli Ouyang, "MotionGPT: Finetuned LLMs are General-Purpose Motion Generators", AAAI, 2024.
  20. Zeyu Lu, Chengyue Wu, Xinyuan Chen, Yaohui Wang, Lei Bai, Yu Qiao, Xihui Liu, "Hierarchical Diffusion Autoencoders with Disentangled Representations", WACV, 2024.
  21. Zeyu Lu+, Di Huang+, Lei Bai+, Xihui Liu, Jingjing Qu, Wanli Ouyang, "Seeing is not always believing: A Quantitative Study on Human Perception of AI-Generated Images", NeurIPS, 2023.
  22. Xu Liu, Yutong Xia, Yuxuan Liang, Junfeng Hu, Yiwei Wang, Lei Bai, Chao Huang, Zhenguang Liu, Bryan Hooi, Roger Zimmermann, "LargeST: A Benchmark Dataset for Large-Scale Traffic Forecasting", NeurIPS, 2023.
  23. Yaqi Zhang, Di Huang, Bin Liu, Shixiang Tang, Yan Lu, Lu Chen, Lei Bai, Qi Chu, Nenghai Yu, Wanli Ouyang, "MotionGPT: Finetuned LLMs are General-Purpose Motion Generators", NeurIPS, 2023.
  24. Binwu Wang, Yudong Zhang, Xu Wang, Pengkun Wang, Zhengyang Zhou, Lei Bai*, Yang Wang*, "Pattern Expansion and Consolidation on Evolving Graphs for Continual Traffic Prediction", KDD, 2023. (Accepted);
  25. Tian Lan, Ziyue Li, Zhishuai Li, Lei Bai, Man Li, Fugee Tsung, Wolfgang Ketter, Rui Zhao, Chen Zhang, "MM-DAG: Multi-task DAG Learning for Multi-modal Data--with Application for Traffic Congestion Analysis", KDD, 2023. (Accepted);
  26. Tao Han, Lei Bai, Lingbo Liu, Wanli Ouyang, "STEERER: Resolving Scale Variations for Counting and Localization via Selective Inheritance Learning", Proceedings of the IEEE/CVF International Conference on Computer Vision, 2023. (Accepted);
  27. Yuning Wang, Pu Zhang, LEI BAI, Jianru Xue, "FEND: A Future Enhanced Distribution-Aware Contrastive Learning Framework For Long-tail Trajectory Prediction", IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023. (Accepted);
  28. Yuanzheng Ci, Yizhou Wang, Meilin Chen, SHIXIANG TANG, LEI BAI*, Feng Zhu, Rui Zhao, Fengwei Yu, Donglian Qi, Wanli Ouyang, "UniHCP: A Unified Model for Human-Centric Perceptions", IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023. (Accepted);
  29. SHIXIANG TANG, Cheng Chen, Meilin Chen, Qingsong Xie, Yizhou Wang, Yuanzheng Ci, LEI BAI*, Feng Zhu, Haiyang Yang, Li Yi, Rui Zhao, Wanli Ouyang, "HumanBench: Towards General Human-centric Perception with Projector Assisted Pretraining", IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023. (Accepted);
  30. Yuhan Xiang, Kaijian Liu, Shixiang Tang, Lei Bai, Feng Zhu, Rui Zhao, Xianming Lin, "Trust Your Partner's Friends: Hierarchical Cross-modal Contrastive Pre-training for Video-Text Retrieval", IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2023. (Accepted);
  31. Yinyan Liu, Lei Bai, Jin Ma, "Low-Carbon Community Energy Management Incorporating Data-Driven User Segmentation", IEEE Power & Energy Society General Meeting (PESGM), 2023. (Accepted);
  32. 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);
  33. 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);
  34. 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);
  35. 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);
  36. 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);
  37. 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*);
  38. 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*);
  39. 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);
  40. 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*);
  41. 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*);
  42. 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*);
  43. 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*);
  44. 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*);
  45. 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*);
  46. 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*);
  47. 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*);
  48. 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*);
  49. 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*);
  50. 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*);
  51. 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*);
  52. 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);
  53. 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*);
  54. 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*);
  55. 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);
  56. 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);
  57. 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);
  58. 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);
  59. 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*);
  60. 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);
  61. 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);
  62. 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);

Journal Papers

  1. Kang Chen+, Tao Han, Junchao Gong, Lei Bai*, Fenghua Ling, Jing-Jia Luo, Xi Chen, Leiming Ma, Tianning Zhang, Rui Su, Yuanzheng Ci, Bin Li, Xiaokang Yang, Wanli Ouyang, "The operational medium-range deterministic weather forecasting can be extended beyond a 10-day lead time", COMMUNICATIONS EARTH & ENVIRONMENT, 2025.;
  2. Yizhou Wang, Yixuan Wu, Shixiang Tang, Weizhen He, Xun Guo, Feng Zhu, Lei Bai, Rui Zhao, Jian Wu, Tong He, Wanli Ouyang, " Hulk: A Universal Knowledge Translator for Human-Centric Tasks", arxiv, 2023.
  3. Fenghua Ling, Zeyu Lu, Jing-Jia Luo*, Lei Bai*, Swadhin K. Behera, Dachao Jin, Baoxiang Pan, Huidong Jiang, and Toshio Yamagata, "Diffusion Model-based Probabilistic Downscaling for Long-term East Asian Climate Reconstruction", npj climate and atmospheric science, 2024.;
  4. Lei Bai+, Dongang Wang+, Michael Barnett, Mariano Cabezas, Weidong Cai, Fernando Calamante, Kain Kyle, Dongnan Liu, Linda Ly, Aria Nguyen, Chun-Chien Shieh, Ryan Sullivan, Hengrui Wang, Geng Zhan, Wanli Ouyang, "Improving Multiple Sclerosis Lesion Segmentation Across Clinical Sites: A Federated Learning Approach with Noise-Resilient Training", Artificial Intelligence In Medicine, 2024.
  5. Yudong Zhang, Lei Bai*, Yang Wang*, "Adaptive and Interactive Multi-level Spatio-Temporal Network for Traffic Forecasting", IEEE Transactions on Intelligent Transportation Systems, 2024.
  6. Pumeng Lyu+, Tao Tang+, Fenghua Ling, Jing-Jia Luo*, Niklas Boers, Wanli Ouyang, Lei Bai*, "ResoNet: Robust and Explainable ENSO Forecasts with Hybrid Convolution and Transformer Networks", ADVANCES IN ATMOSPHERIC SCIENCES, 2024.
  7. Song Yang, Fenghua Ling, Jingjia Luo*, Lei Bai*"Improving Seasonal Forecast of Summer Precipitation in Southeastern China using CycleGAN Deep Learning Bias Correction", ADVANCES IN ATMOSPHERIC SCIENCES, 2024.
  8. Taoyong Cui, Chenyu Tang, Mao Su, Shufei Zhang, Yuqiang Li, Lei Bai, Yuhan Dong, Xingao Gong, Wanli Ouyang, "Gpip: Geometry-enhanced pre-training on interatomic potentials", Nature Machine Intelligence, 2024.
  9. Haoyuan Jiang, Ziyue Li, Lei Bai, Rui Zhao, "A General Scenario-Agnostic Reinforcement Learning for Traffic Signal Control", IEEE Transactions on Intelligent Transportation Systems, 2024.
  10. Weitao Feng, Lei Bai*, Yongqiang Yao, Fengwei Yu, Wanli Ouyang, "Towards Frame Rate Agnostic Multi-Object Tracking", IJCV, 2023.
  11. Xu Wang, Pengkun Wang, Binwu Wang, Yudong Zhang, Zhengyang Zhou, Lei Bai, Yang Wang, "Latent Gaussian Processes based Graph Learning for Urban Traffic Prediction", IEEE Transactions on Vehicular Technology, 2023.
  12. Dongnan Liu, Mariano Cabezas, Dongang Wang, Zihao Tang, Lei Bai, Geng Zhan, Yuling Luo, Kain Kyle, Linda Ly, James Yu, Chun-Chien Shieh, Aria Nguyen, Ettikan Kandasamy Karuppiah, Ryan Sullivan, Fernando Calamante, Michael Barnett, Wanli Ouyang, Weidong Cai, Chenyu Wang, "Multiple sclerosis lesion segmentation: revisiting weighting mechanisms for federated learning", Frontiers in Neuroscience, 2023.
  13. Yinyan Liu, Lei Bai, Jin Ma, Wei Wang, Wanli Ouyang, "Self-Supervised Feature Learning for Appliance Recognition in Non-Intrusive Load Monitoring", IEEE Transactions on Industrial Informatics, 2023.
  14. Yaning Cui, Kang Chen, Lingyao Zhang, Haotian Wang, Lei Bai, David Elliston, Wei Ren, "Atomic Positional Embedding-Based Transformer Model for Predicting the Density of States of Crystalline Materials", The Journal of Physical Chemistry Letters, 2023.
  15. 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, 2023. (Accepted)
  16. 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), 2023. (Accepted)
  17. Binwu Wang, Jiahao Shi, Yudong Zhang, Pengkun Wang, Lei Bai*, Yang Wang*, "Knowledge Expansion and Consolidation for Continual Traffic Prediction with Expanding Graphs", IEEE Transactions on Intelligent Transportation Systems, 2023. (Accepted);
  18. Lin Ouyang, Fenghua Ling, Yue Li, Lei Bai, Jingjia Luo, "Wave forecast in the Atlantic Ocean using a double-stage ConvLSTM network", Atmospheric and Oceanic Science Letters, 2023. (Accepted);
  19. Can Li, Lei Bai, Wei Liu, Lina Yao, S Travis Waller, "A Bibliometric Analysis and Review on Reinforcement Learning for Transportation Applications", Transportmetrica B: Transport Dynamics, 2023. (Accepted);
  20. 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, 2022. (Accepted);
  21. 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.
  22. 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.
  23. 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.
  24. 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.
  25. 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.
  26. 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.
  27. 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.
  28. 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.
  29. 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.
  30. 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.
  31. 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.

Professional Services

Conference PC Member/Reviewer

  • Reviewer, ICML 2024
  • Reviewer, ECCV 2024
  • Reviewer, CVPR 2024
  • Reviewer, IJCAI 2024
  • Reviewer, Thirty-seventh Conference on Neural Information Processing Systems (NeurIPS), 2023
  • 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, Climate Dynamics
  • 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

  • IEEE Circuits and Systems for Video Technology (CSVT) Transactions Best Paper Award, 2024
  • 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

Media

  • AI科学家眼里的青少年AI教育，中国信息技术教育，2024年第10期
  • 风乌大模型实现10公里级气象预报，新华网，2023年3月4日
  • 科学智算（AI4S）:交叉与赋能，中国计算机学会通讯，2024年2月第2期
  • AI会让科学家失业吗？中国新闻周刊，2023年5月15日总第1091期

Invited Talks

In English

  • Invited Talk by David John Gagne, MILES Group Meeting at National Center for Atmospheric Research, March 2024
  • Invited Talk by Niklas Bores, Workshop on AI for Weather and Climate Modelling, March 2024
  • Keynote Speaker, The ESCAP/WMO Typhoon Committee, the 18th Integrated Workshop conjoint with the 4th Training and Research Coordination Group, Dec 2023
  • Invited Speaker, Large-scale deep learning for the Earth system, Sept 2023

In Chinese

  • Hongkong-Shanghai AI Forum, 2024年4月
  • 2024全球开发者先锋大会, 2024年3月
  • 同济大学交通运输工程学院（邀请人：马万经教授）, 2024年3月
  • 2023年全国卫星数据同化研讨会暨首届国产卫星数据同化应用研讨会, 2023年10月
  • 复旦大学计算机学院（邀请人：陈阳教授）, 2023年6月
  • 中国气象局人工智能气象应用发展研讨会, 2023年5月
  • 中国气象服务协会年会2022-2023年会, 2023年4月