Bai, Lei

74 publications

TMLR 2026 The Landscape of Agentic Reinforcement Learning for LLMs: A Survey Guibin Zhang, Hejia Geng, Xiaohang Yu, Zhenfei Yin, Zaibin Zhang, Zelin Tan, Heng Zhou, Zhong-Zhi Li, Xiangyuan Xue, Yijiang Li, Yifan Zhou, Yang Chen, Chen Zhang, Yutao Fan, Zihu Wang, Songtao Huang, Francisco Piedrahita Velez, Yue Liao, Hongru Wang, Mengyue Yang, Heng Ji, Jun Wang, Shuicheng Yan, Philip Torr, Lei Bai

NeurIPS 2025 Actial: Activate Spatial Reasoning Ability of Multimodal Large Language Models Xiaoyu Zhan, Wenxuan Huang, Hao Sun, Xinyu Fu, Changfeng Ma, Shaosheng Cao, Bohan Jia, Shaohui Lin, Zhenfei Yin, Lei Bai, Wanli Ouyang, Yuanqi Li, Jie Guo, Yanwen Guo

NeurIPS 2025 Align-DA: Align Score-Based Atmospheric Data Assimilation with Multiple Preferences Jing-An Sun, Hang Fan, Junchao Gong, Ben Fei, Kun Chen, Fenghua Ling, Wenlong Zhang, Wanghan Xu, Li Yan, Pierre Gentine, Lei Bai

NeurIPS 2025 BMMR: A Large-Scale Bilingual Multimodal Multi-Discipline Reasoning Dataset Zhiheng Xi, Guanyu Li, YuTao Fan, Honglin Guo, Yufang Liu, Xiaoran Fan, Jiaqi Liu, Dingjinchao, Wangmeng Zuo, Zhenfei Yin, Lei Bai, Tao Ji, Tao Gui, Qi Zhang, Xuanjing Huang

NeurIPS 2025 Breaking the Compression Ceiling: Data-Free Pipeline for Ultra-Efficient Delta Compression Xiaohui Wang, Peng Ye, Chenyu Huang, Shenghe Zheng, Bo Zhang, Lei Bai, Wanli Ouyang, Tao Chen

ICCV 2025 Chimera: Improving Generalist Model with Domain-Specific Experts Tianshuo Peng, Mingsheng Li, Jiakang Yuan, Hongbin Zhou, Renqiu Xia, Renrui Zhang, Lei Bai, Song Mao, Bin Wang, Aojun Zhou, Botian Shi, Tao Chen, Bo Zhang, Xiangyu Yue

CVPR 2025 ComfyBench: Benchmarking LLM-Based Agents in ComfyUI for Autonomously Designing Collaborative AI Systems Xiangyuan Xue, Zeyu Lu, Di Huang, Zidong Wang, Wanli Ouyang, Lei Bai

NeurIPS 2025 DAWP: A Framework for Global Observation Forecasting via Data Assimilation and Weather Prediction in Satellite Observation Space Junchao Gong, Jingyi Xu, Ben Fei, Fenghua Ling, Wenlong Zhang, Kun Chen, Wanghan Xu, Weidong Yang, Xiaokang Yang, Lei Bai

NeurIPS 2025 GauDP: Reinventing Multi-Agent Collaboration Through Gaussian-Image Synergy in Diffusion Policies Ziye Wang, Li Kang, Yiran Qin, Jiahua Ma, Zhanglin Peng, Lei Bai, Ruimao Zhang

CVPR 2025 IceDiff: High Resolution and High-Quality Arctic Sea Ice Forecasting with Generative Diffusion Prior Jingyi Xu, Siwei Tu, Weidong Yang, Ben Fei, Shuhao Li, Keyi Liu, Yeqi Luo, Lipeng Ma, Lei Bai

ICCV 2025 InfGen: A Resolution-Agnostic Paradigm for Scalable Image Synthesis Tao Han, Wanghan Xu, Junchao Gong, Xiaoyu Yue, Song Guo, Luping Zhou, Lei Bai

NeurIPS 2025 LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents Rui Li, Zixuan Hu, Wenxi Qu, Jinouwen Zhang, Zhenfei Yin, Sha Zhang, Xuantuo Huang, Hanqing Wang, Tai Wang, Jiangmiao Pang, Wanli Ouyang, Lei Bai, Wangmeng Zuo, Lingyu Duan, Dongzhan Zhou, Shixiang Tang

NeurIPS 2025 Latent Harmony: Synergistic Unified UHD Image Restoration via Latent Space Regularization and Controllable Refinement Yidi Liu, Xueyang Fu, Jie Huang, Jie Xiao, Dong Li, Wenlong Zhang, Lei Bai, Zheng-Jun Zha

NeurIPS 2025 Learning Urban Climate Dynamics via Physics-Guided Urban Surface–Atmosphere Interactions Jiyang Xia, Fenghua Ling, Zhenhui Jessie Li, Junjie Yu, Hongliang Zhang, David Topping, Lei Bai, Zhonghua Zheng

NeurIPS 2025 LoRA-EnVar: Parameter-Efficient Hybrid Ensemble Variational Assimilation for Weather Forecasting Yi Xiao, Hang Fan, Kun Chen, Ye Cao, Ben Fei, Wei Xue, Lei Bai

ICML 2025 Multi-Agent Architecture Search via Agentic Supernet Guibin Zhang, Luyang Niu, Junfeng Fang, Kun Wang, Lei Bai, Xiang Wang

NeurIPS 2025 NUTS: Eddy-Robust Reconstruction of Surface Ocean Nutrients via Two-Scale Modeling Hao Zheng, Shiyu Liang, Yuting Zheng, Chaofan Sun, Lei Bai, Enhui Liao

NeurIPS 2025 Native-Resolution Image Synthesis ZiDong Wang, Lei Bai, Xiangyu Yue, Wanli Ouyang, Yiyuan Zhang

ICLR 2025 PostCast: Generalizable Postprocessing for Precipitation Nowcasting via Unsupervised Blurriness Modeling Junchao Gong, Siwei Tu, Weidong Yang, Ben Fei, Kun Chen, Zhangwenlong, Xiaokang Yang, Wanli Ouyang, Lei Bai

NeurIPS 2025 RadarQA: Multi-Modal Quality Analysis of Weather Radar Forecasts Xuming He, Zhiyuan You, Junchao Gong, Couhua Liu, Xiaoyu Yue, Peiqin Zhuang, Wenlong Zhang, Lei Bai

NeurIPS 2025 Retro-R1: LLM-Based Agentic Retrosynthesis Wei Liu, Jiangtao Feng, Hongli Yu, Yuxuan Song, Yuqiang Li, Shufei Zhang, Lei Bai, Wei-Ying Ma, Hao Zhou

ICCV 2025 RoboFactory: Exploring Embodied Agent Collaboration with Compositional Constraints Yiran Qin, Li Kang, Xiufeng Song, Zhenfei Yin, Xiaohong Liu, Xihui Liu, Ruimao Zhang, Lei Bai

NeurIPS 2025 SIFusion: A Unified Fusion Framework for Multi-Granularity Arctic Sea Ice Forecasting Jingyi Xu, Shengnan Wang, Weidong Yang, Keyi Liu, Yeqi Luo, Ben Fei, Lei Bai

CVPR 2025 Satellite Observations Guided Diffusion Model for Accurate Meteorological States at Arbitrary Resolution Siwei Tu, Ben Fei, Weidong Yang, Fenghua Ling, Hao Chen, Zili Liu, Kun Chen, Hang Fan, Wanli Ouyang, Lei Bai

NeurIPS 2025 Scaling Physical Reasoning with the PHYSICS Dataset Shenghe Zheng, Qianjia Cheng, Junchi Yao, Mengsong Wu, Haonan He, Ning Ding, Yu Cheng, Shuyue Hu, Lei Bai, Dongzhan Zhou, Ganqu Cui, Peng Ye

NeurIPS 2025 Scientists' First Exam: Probing Cognitive Abilities of MLLM via Perception, Understanding, and Reasoning Yuhao Zhou, Yiheng Wang, Xuming He, Ruoyao Xiao, Zhiwei Li, Qiantai Feng, Zijie Guo, Yuejin Yang, Hao Wu, Wenxuan Huang, Jiaqi Wei, Dan Si, Yao Xiuqi, Jia Bu, Haiwen Huang, Tianfan Fu, Shixiang Tang, Ben Fei, Dongzhan Zhou, Fenghua Ling, Yan Lu, Siqi Sun, Chenhui Li, Guanjie Zheng, Jiancheng Lv, Wenlong Zhang, Lei Bai

ICLR 2025 TimeKAN: KAN-Based Frequency Decomposition Learning Architecture for Long-Term Time Series Forecasting Songtao Huang, Zhen Zhao, Can Li, Lei Bai

NeurIPS 2025 Understand Before You Generate: Self-Guided Training for Autoregressive Image Generation Xiaoyu Yue, ZiDong Wang, Yuqing Wang, Wenlong Zhang, Xihui Liu, Wanli Ouyang, Lei Bai, Luping Zhou

CVPR 2025 UniSTD: Towards Unified Spatio-Temporal Learning Across Diverse Disciplines Chen Tang, Xinzhu Ma, Encheng Su, Xiufeng Song, Xiaohong Liu, Wei-Hong Li, Lei Bai, Wanli Ouyang, Xiangyu Yue

ICCV 2025 VA-MoE: Variables-Adaptive Mixture of Experts for Incremental Weather Forecasting Hao Chen, Han Tao, Guo Song, Jie Zhang, Yonghan Dong, Yunlong Yu, Lei Bai

ICLR 2025 VAE-Var: Variational Autoencoder-Enhanced Variational Methods for Data Assimilation in Meteorology Yi Xiao, Qilong Jia, Kun Chen, Lei Bai, Wei Xue

NeurIPS 2025 VIKI‑R: Coordinating Embodied Multi-Agent Cooperation via Reinforcement Learning Li Kang, Xiufeng Song, Heng Zhou, Yiran Qin, Jie Yang, Xiaohong Liu, Philip Torr, Lei Bai, Zhenfei Yin

ICCV 2025 VLIPP: Towards Physically Plausible Video Generation with Vision and Language Informed Physical Prior Xindi Yang, Baolu Li, Yiming Zhang, Zhenfei Yin, Lei Bai, Liqian Ma, Zhiyong Wang, Jianfei Cai, Tien-Tsin Wong, Huchuan Lu, Xu Jia

AAAI 2025 VQLTI: Long-Term Tropical Cyclone Intensity Forecasting with Physical Constraints Xinyu Wang, Lei Liu, Kang Chen, Tao Han, Bin Li, Lei Bai

ICLR 2025 WeatherGFM: Learning a Weather Generalist Foundation Model via In-Context Learning Xiangyu Zhao, Zhiwang Zhou, Zhangwenlong, Yihao Liu, Xiangyu Chen, Junchao Gong, Hao Chen, Ben Fei, Shiqi Chen, Wanli Ouyang, Xiao-Ming Wu, Lei Bai

ICML 2025 WorldSimBench: Towards Video Generation Models as World Simulators Yiran Qin, Zhelun Shi, Jiwen Yu, Xijun Wang, Enshen Zhou, Lijun Li, Zhenfei Yin, Xihui Liu, Lu Sheng, Jing Shao, Lei Bai, Ruimao Zhang

NeurIPS 2025 scMRDR: A Scalable and Flexible Framework for Unpaired Single-Cell Multi-Omics Data Integration Jianle Sun, Chaoqi Liang, Ran Wei, Peng Zheng, Lei Bai, Wanli Ouyang, Hongliang Yan, Peng Ye

AAAI 2024 Autoregressive Omni-Aware Outpainting for Open-Vocabulary 360-Degree Image Generation Zhuqiang Lu, Kun Hu, Chaoyue Wang, Lei Bai, Zhiyong Wang

ICML 2024 CasCast: Skillful High-Resolution Precipitation Nowcasting via Cascaded Modelling Junchao Gong, Lei Bai, Peng Ye, Wanghan Xu, Na Liu, Jianhua Dai, Xiaokang Yang, Wanli Ouyang

NeurIPS 2024 FNP: Fourier Neural Processes for Arbitrary-Resolution Data Assimilation Kun Chen, Peng Ye, Hao Chen, Kang Chen, Tao Han, Wanli Ouyang, Tao Chen, Lei Bai

ICML 2024 FiT: Flexible Vision Transformer for Diffusion Model Zeyu Lu, Zidong Wang, Di Huang, Chengyue Wu, Xihui Liu, Wanli Ouyang, Lei Bai

NeurIPS 2024 Generalizing Weather Forecast to Fine-Grained Temporal Scales via Physics-AI Hybrid Modeling Wanghan Xu, Fenghua Ling, Wenlong Zhang, Tao Han, Hao Chen, Wanli Ouyang, Lei Bai

WACV 2024 Hierarchical Diffusion Autoencoders and Disentangled Image Manipulation Zeyu Lu, Chengyue Wu, Xinyuan Chen, Yaohui Wang, Lei Bai, Yu Qiao, Xihui Liu

CVPR 2024 Instruct-ReID: A Multi-Purpose Person Re-Identification Task with Instructions Weizhen He, Yiheng Deng, Shixiang Tang, Qihao Chen, Qingsong Xie, Yizhou Wang, Lei Bai, Feng Zhu, Rui Zhao, Wanli Ouyang, Donglian Qi, Yunfeng Yan

AAAI 2024 MotionGPT: Finetuned LLMs Are General-Purpose Motion Generators Yaqi Zhang, Di Huang, Bin Liu, Shixiang Tang, Yan Lu, Lu Chen, Lei Bai, Qi Chu, Nenghai Yu, Wanli Ouyang

AISTATS 2024 Non-Neighbors Also Matter to Kriging: A New Contrastive-Prototypical Learning Zhishuai Li, Yunhao Nie, Ziyue Li, Lei Bai, Yisheng Lv, Rui Zhao

NeurIPS 2024 On Learning Multi-Modal Forgery Representation for Diffusion Generated Video Detection Xiufeng Song, Xiao Guo, Jiache Zhang, Qirui Li, Lei Bai, Xiaoming Liu, Guangtao Zhai, Xiaohong Liu

ECCV 2024 PredBench: Benchmarking Spatio-Temporal Prediction Across Diverse Disciplines ZiDong Wang, Zeyu Lu, Di Huang, Tong He, Xihui Liu, Wanli Ouyang, Lei Bai

AAAI 2024 Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled Perspective Binwu Wang, Pengkun Wang, Yudong Zhang, Xu Wang, Zhengyang Zhou, Lei Bai, Yang Wang

ICML 2024 Towards a Self-Contained Data-Driven Global Weather Forecasting Framework Yi Xiao, Lei Bai, Wei Xue, Hao Chen, Kun Chen, Kang Chen, Tao Han, Wanli Ouyang

ICLR 2023 Cycle-Consistent Masked AutoEncoder for Unsupervised Domain Generalization Haiyang Yang, Xiaotong Li, Shixiang Tang, Feng Zhu, Yizhou Wang, Meilin Chen, Lei Bai, Rui Zhao, Wanli Ouyang

CVPR 2023 FEND: A Future Enhanced Distribution-Aware Contrastive Learning Framework for Long-Tail Trajectory Prediction Yuning Wang, Pu Zhang, Lei Bai, Jianru Xue

ICLR 2023 GReTo: Remedying Dynamic Graph Topology-Task Discordance via Target Homophily Zhengyang Zhou, Qihe Huang, Gengyu Lin, Kuo Yang, Lei Bai, Yang Wang

CVPR 2023 HumanBench: Towards General Human-Centric Perception with Projector Assisted Pretraining Shixiang Tang, Cheng Chen, Qingsong Xie, Meilin Chen, Yizhou Wang, Yuanzheng Ci, Lei Bai, Feng Zhu, Haiyang Yang, Li Yi, Rui Zhao, Wanli Ouyang

NeurIPS 2023 LAMM: Language-Assisted Multi-Modal Instruction-Tuning Dataset, Framework, and Benchmark Zhenfei Yin, Jiong Wang, Jianjian Cao, Zhelun Shi, Dingning Liu, Mukai Li, Xiaoshui Huang, Zhiyong Wang, Lu Sheng, Lei Bai, Jing Shao, Wanli Ouyang

NeurIPS 2023 LargeST: A Benchmark Dataset for Large-Scale Traffic Forecasting Xu Liu, Yutong Xia, Yuxuan Liang, Junfeng Hu, Yiwei Wang, Lei Bai, Chao Huang, Zhenguang Liu, Bryan Hooi, Roger Zimmermann

AAAI 2023 Multi-Scale Control Signal-Aware Transformer for Motion Synthesis Without Phase Lintao Wang, Kun Hu, Lei Bai, Yu Ding, Wanli Ouyang, Zhiyong Wang

ICCV 2023 STEERER: Resolving Scale Variations for Counting and Localization via Selective Inheritance Learning Tao Han, Lei Bai, Lingbo Liu, Wanli Ouyang

NeurIPS 2023 Seeing Is Not Always Believing: Benchmarking Human and Model Perception of AI-Generated Images Zeyu Lu, Di Huang, Lei Bai, Jingjing Qu, Chengyue Wu, Xihui Liu, Wanli Ouyang

CVPR 2023 UniHCP: A Unified Model for Human-Centric Perceptions Yuanzheng Ci, Yizhou Wang, Meilin Chen, Shixiang Tang, Lei Bai, Feng Zhu, Rui Zhao, Fengwei Yu, Donglian Qi, Wanli Ouyang

ECCV 2022 Backbone Is All Your Need: A Simplified Architecture for Visual Object Tracking Boyu Chen, Peixia Li, Lei Bai, Lei Qiao, Qiuhong Shen, Bo Li, Weihao Gan, Wei Wu, Wanli Ouyang

CVPR 2022 DR.VIC: Decomposition and Reasoning for Video Individual Counting Tao Han, Lei Bai, Junyu Gao, Qi Wang, Wanli Ouyang

ECCV 2022 Domain Invariant Masked Autoencoders for Self-Supervised Learning from Multi-Domains Haiyang Yang, Shixiang Tang, Meilin Chen, Yizhou Wang, Feng Zhu, Lei Bai, Rui Zhao, Wanli Ouyang

ECCV 2022 Fast-MoCo: Boost Momentum-Based Contrastive Learning with Combinatorial Patches Yuanzheng Ci, Chen Lin, Lei Bai, Wanli Ouyang

ECCV 2022 Relative Contrastive Loss for Unsupervised Representation Learning Shixiang Tang, Feng Zhu, Lei Bai, Rui Zhao, Wanli Ouyang

CVPR 2022 Revisiting the Transferability of Supervised Pretraining: An MLP Perspective Yizhou Wang, Shixiang Tang, Feng Zhu, Lei Bai, Rui Zhao, Donglian Qi, Wanli Ouyang

AAAI 2022 SepFusion: Finding Optimal Fusion Structures for Visual Sound Separation Dongzhan Zhou, Xinchi Zhou, Di Hu, Hang Zhou, Lei Bai, Ziwei Liu, Wanli Ouyang

ECCV 2022 Unifying Visual Contrastive Learning for Object Recognition from a Graph Perspective Shixiang Tang, Feng Zhu, Lei Bai, Rui Zhao, Chenyu Wang, Wanli Ouyang

NeurIPS 2022 Unsupervised Object Detection Pretraining with Joint Object Priors Generation and Detector Learning Yizhou Wang, Meilin Chen, Shixiang Tang, Feng Zhu, Haiyang Yang, Lei Bai, Rui Zhao, Yunfeng Yan, Donglian Qi, Wanli Ouyang

ICCV 2021 GLiT: Neural Architecture Search for Global and Local Image Transformer Boyu Chen, Peixia Li, Chuming Li, Baopu Li, Lei Bai, Chen Lin, Ming Sun, Junjie Yan, Wanli Ouyang

ICCV 2021 Graph-Based 3D Multi-Person Pose Estimation Using Multi-View Images Size Wu, Sheng Jin, Wentao Liu, Lei Bai, Chen Qian, Dong Liu, Wanli Ouyang

CVPR 2021 Mutual CRF-GNN for Few-Shot Learning Shixiang Tang, Dapeng Chen, Lei Bai, Kaijian Liu, Yixiao Ge, Wanli Ouyang

NeurIPS 2020 Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting Lei Bai, Lina Yao, Can Li, Xianzhi Wang, Can Wang

IJCAI 2019 STG2Seq: Spatial-Temporal Graph to Sequence Model for Multi-Step Passenger Demand Forecasting Lei Bai, Lina Yao, Salil S. Kanhere, Xianzhi Wang, Quan Z. Sheng