Lei Bai

Shanghai AI Laboratory, China

66 papers at tracked venues · 63 at CORE A* · active 2024–2026

DBLP profile ↗ORCID 0000-0003-3378-7201 ↗Google Scholar ↗Homepage ↗

Venues

NeurIPS ×23 ACL ×12 AAAI ×6 CVPR ×5 ICCV ×5 ICML ×5 ICLR ×4 ACM MM ×1 AISTATS ×1 ECCV ×1 EMNLP ×1 ICDE ×1 WWW ×1

Frequent coauthors

Wanli Ouyang

×22

Wenlong Zhang

×12

Ben Fei

×11

Bo Zhang

×9

Peng Ye

×9

Zhenfei Yin

×9

Fenghua Ling

×8

Junchao Gong

×8

Kun Chen

×8

Shuyue Hu

×6

Wanghan Xu

×6

Tao Han

×6

Papers

A Scalable Multi-LLM Collaboration System with Retrieval-based Selection and Exploration-Exploitation-Driven Enhancement
ACL 2026 ·
Shengji Tang
,
Jianjian Cao
,
Weihao Lin
,
Jiale Hong
,
Bo Zhang
,
Shuyue Hu
, Lei Bai,
Tao Chen
,
Wanli Ouyang
,
Peng Ye
Easy Samples Are All You Need: Self-Evolving LLMs via Data-Efficient Reinforcement Learning
ACL 2026 ·
Zhiyin Yu
,
Bo Zhang
,
Qibin Hou
,
Zhonghai Wu
,
Xiao Luo
, Lei Bai
FlowSearch: Advancing Deep Research with Dynamic Structured Knowledge Flow
ACL 2026 ·
Yusong Hu
,
Runmin Ma
,
Yue Fan
,
Jinxin Shi
,
Zongsheng Cao
,
Yuhao Zhou
,
Jiakang Yuan
,
Shuaiyu Zhang
,
Shiyang Feng
,
Xiangchao Yan
,
Shufei Zhang
,
Wenlong Zhang
, Lei Bai,
Bo Zhang
LLMRouterBench: A Massive Benchmark and Unified Framework for LLM Routing
ACL 2026 ·
Hao Li
,
Yiqun Zhang
,
Zhaoyan Guo
,
Chenxu Wang
,
Shengji Tang
,
Qiaosheng Zhang
,
Yang Chen
,
Biqing Qi
,
Peng Ye
, Lei Bai,
Zhen Wang
,
Shuyue Hu
MSEarth: A Multimodal Benchmark for Earth Science Phenomenon Discovery with MLLMs
ACL 2026 ·
Xiangyu Zhao
,
Wanghan Xu
,
Bo Liu
,
Yuhao Zhou
,
Fenghua Ling
,
Ben Fei
,
Xiaoyu Yue
, Lei Bai,
Wenlong Zhang
,
Xiao-Ming Wu
MTRouter: Cost-Aware Multi-Turn LLM Routing with History-Model Joint Embeddings
ACL 2026 ·
Yiqun Zhang
,
Hao Li
,
Zihan Wang
,
Shi Feng
,
Xiaocui Yang
,
Daling Wang
,
Bo Zhang
, Lei Bai,
Shuyue Hu
MinerU2.5: A Decoupled Vision-Language Model for Efficient High-Resolution Document Parsing
ACL 2026 ·
Junbo Niu
,
Zheng Liu
,
Zhuangcheng Gu
,
Bin Wang
,
Linke Ouyang
,
Zhiyuan Zhao
,
Tao Chu
,
Tianyao He
,
Fan Wu
,
Qintong Zhang
,
Zhenjiang Jin
,
Guang Liang
,
Rui Zhang
,
Wenzheng Zhang
,
Yuan Qu
,
Zhifei Ren
,
Yuefeng Sun
,
Zirui Tang
,
Boyu Niu
,
Yuanhong Zheng
,
Dongsheng Ma
,
Ziyang Miao
,
Hejun Dong
,
Siyi Qian
,
Junyuan Zhang
,
Fangdong Wang
,
Jingzhou Chen
,
Xiaomeng Zhao
,
Liqun Wei
,
Wei Li
,
Shasha Wang
,
Ruiliang Xu
,
Yuanyuan Cao
,
Lu Chen
,
Qianqian Wu
,
Huaiyu Gu
,
Lindong Lu
,
Dechen Lin
,
Shenguanlin
,
Xuanhe Zhou
,
Linfeng Zhang
,
Yuhang Zang
,
Xiaoyi Dong
,
Jiaqi Wang
,
Bo Zhang
, Lei Bai,
Pei Chu
,
Weijia Li
,
Jiang Wu
,
Lijun Wu
,
Zhenxiang Li
,
Guangyu Wang
,
Zhongying Tu
,
Chao Xu
,
Kai Chen
,
Bowen Zhou
,
Dahua Lin
,
Wentao Zhang
,
Conghui He
Modeling Point-to-Point Dependency for High-Dimensional Long-Term Series Forecasting
WWW 2026 ·
Xinyu Li
,
Kexi Chen
,
Ying Zheng
,
Zhiyi Yao
,
Yi Xie
,
Jihan Dai
, Lei Bai,
Jin Zhao
,
Jiajie Shen
,
Yunqi Cai
,
Hong Lu
,
Xin Wang
Nature-Inspired Population-Based Evolution of Large Language Models
ACL 2026 ·
Yiqun Zhang
,
Peng Ye
,
Xiaocui Yang
,
Shi Feng
,
Shufei Zhang
, Lei Bai,
Wanli Ouyang
,
Shuyue Hu
R³: End-to-End Reasoning-based Planning for Multi-step Retrosynthesis via Reinforcement Learning
ACL 2026 ·
YiFei Wang
,
Qizhi Pei
,
Jiangtao Feng
,
Yuntian Shi
,
Yi Duan
,
Lihao Wang
, Lei Bai,
Lijun Wu
,
Wei-Ying Ma
,
Hao Zhou
Scaling Behaviors of LLM Reinforcement Learning Post-Training: An Empirical Study in Mathematical Reasoning
ACL 2026 ·
Zelin Tan
,
Hejia Geng
,
Xiaohang Yu
,
Mulei Zhang
,
Guancheng Wan
,
Yifan Zhou
,
Qiang He
,
Xiangyuan Xue
,
Heng Zhou
,
Yutao Fan
,
Zhong-Zhi Li
,
Zaibin Zhang
,
Guibin Zhang
,
Chen Zhang
,
Zhenfei Yin
,
Philip Torr
, Lei Bai
SynWeather: Weather Observation Data Synthesis Across Multiple Regions and Variables via a General Diffusion Transformer
AAAI 2026 ·
Kaiyi Xu
,
Junchao Gong
,
Zhiwang Zhou
,
Zhangrui Li
,
Yuandong Pu
,
Yihao Liu
,
Ben Fei
,
Fenghua Ling
,
Wenlong Zhang
, Lei Bai
The Avengers: A Routing Recipe for Collective Intelligence in Language Models
AAAI 2026 ·
Yiqun Zhang
,
Hao Li
,
Chenxu Wang
,
Linyao Chen
,
Qiaosheng Zhang
,
Peng Ye
,
Shi Feng
,
Xinrun Wang
,
Xu Jia
, Lei Bai,
Shuyue Hu
Actial: Activate Spatial Reasoning Ability of Multimodal Large Language Models
NeurIPS 2025 ·
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
Align-DA: Align Score-based Atmospheric Data Assimilation with Multiple Preferences
NeurIPS 2025 ·
Jing-An Sun
,
Hang Fan
,
Junchao Gong
,
Ben Fei
,
Kun Chen
,
Fenghua Ling
,
Wenlong Zhang
,
Wanghan Xu
,
Li Yan
,
Pierre Gentine
, Lei Bai
BMMR: A Large-Scale Bilingual Multimodal Multi-Discipline Reasoning Dataset
NeurIPS 2025 ·
Zhiheng Xi
,
Guanyu Li
,
Yutao Fan
,
Honglin Guo
,
Yufang Liu
,
Xiaoran Fan
,
Jiaqi Liu
,
Jingchao Ding
,
Wangmeng Zuo
,
Zhenfei Yin
, Lei Bai,
Tao Ji
,
Tao Gui
,
Qi Zhang
,
Xuanjing Huang
Breaking the Compression Ceiling: Data-Free Pipeline for Ultra-Efficient Delta Compression
NeurIPS 2025 ·
Xiaohui Wang
,
Peng Ye
,
Chenyu Huang
,
Shenghe Zheng
,
Bo Zhang
, Lei Bai,
Wanli Ouyang
,
Tao Chen
Chimera: Improving Generalist Model with Domain-Specific Experts
ICCV 2025 ·
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
ComfyBench: Benchmarking LLM-based Agents in ComfyUI for Autonomously Designing Collaborative AI Systems
CVPR 2025 ·
Xiangyuan Xue
,
Zeyu Lu
,
Di Huang
,
Zidong Wang
,
Wanli Ouyang
, Lei Bai
DAWP: A framework for global observation forecasting via Data Assimilation and Weather Prediction in satellite observation space
NeurIPS 2025 ·
Junchao Gong
,
Jingyi Xu
,
Ben Fei
,
Fenghua Ling
,
Wenlong Zhang
,
Kun Chen
,
Wanghan Xu
,
Weidong Yang
,
Xiaokang Yang
, Lei Bai
DIFFODE: Neural ODE with Differentiable Hidden State for Irregular Time Series Analysis
ICDE 2025 ·
Yudong Zhang
,
Xu Wang
,
Xuan Yu
,
Zhengyang Zhou
,
Xing Xu
, Lei Bai,
Yang Wang
Dolphin: Moving Towards Closed-loop Auto-research through Thinking, Practice, and Feedback
ACL 2025 ·
Jiakang Yuan
,
Xiangchao Yan
,
Bo Zhang
,
Tao Chen
,
Botian Shi
,
Wanli Ouyang
,
Yu Qiao
, Lei Bai,
Bowen Zhou
GauDP: Reinventing Multi-Agent Collaboration through Gaussian-Image Synergy in Diffusion Policies
NeurIPS 2025 ·
Ziye Wang
,
Li Kang
,
Yiran Qin
,
Jiahua Ma
,
Zhanglin Peng
, Lei Bai,
Ruimao Zhang
IceDiff: High Resolution and High-Quality Arctic Sea Ice Forecasting with Generative Diffusion Prior
CVPR 2025 ·
Jingyi Xu
,
Siwei Tu
,
Weidong Yang
,
Ben Fei
,
Shuhao Li
,
Keyi Liu
,
Yeqi Luo
,
Lipeng Ma
, Lei Bai
InfGen: A Resolution-Agnostic Paradigm for Scalable Image Synthesis
ICCV 2025 ·
Tao Han
,
Wanghan Xu
,
Junchao Gong
,
Xiaoyu Yue
,
Song Guo
,
Luping Zhou
, Lei Bai
LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents
NeurIPS 2025 ·
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
Latent Harmony: Synergistic Unified UHD Image Restoration via Latent Space Regularization and Controllable Refinement
NeurIPS 2025 ·
Yidi Liu
,
Xueyang Fu
,
Jie Huang
,
Jie Xiao
,
Dong Li
,
Wenlong Zhang
, Lei Bai,
Zheng-Jun Zha
Learning Urban Climate Dynamics via Physics-Guided Urban Surface-Atmosphere Interactions
NeurIPS 2025 ·
Jiyang Xia
,
Fenghua Ling
,
Zhenhui Jessie Li
,
Junjie Yu
,
Hongliang Zhang
,
David Topping
, Lei Bai,
Zhonghua Zheng
LoRA-EnVar: Parameter-Efficient Hybrid Ensemble Variational Assimilation for Weather Forecasting
NeurIPS 2025 ·
Yi Xiao
,
Hang Fan
,
Kun Chen
,
Ye Cao
,
Ben Fei
,
Wei Xue
, Lei Bai
Multi-agent Architecture Search via Agentic Supernet
ICML 2025 ·
Guibin Zhang
,
Luyang Niu
,
Junfeng Fang
,
Kun Wang
, Lei Bai,
Xiang Wang
NUTS: Eddy-Robust Reconstruction of Surface Ocean Nutrients via Two-Scale Modeling
NeurIPS 2025 ·
Hao Zheng
,
Shiyu Liang
,
Yuting Zheng
,
Chaofan Sun
, Lei Bai,
Enhui Liao
Native-Resolution Image Synthesis
NeurIPS 2025 ·
Zidong Wang
, Lei Bai,
Xiangyu Yue
,
Wanli Ouyang
,
Yiyuan Zhang
PostCast: Generalizable Postprocessing for Precipitation Nowcasting via Unsupervised Blurriness Modeling
ICLR 2025 ·
Junchao Gong
,
Siwei Tu
,
Weidong Yang
,
Ben Fei
,
Kun Chen
,
Wenlong Zhang
,
Xiaokang Yang
,
Wanli Ouyang
, Lei Bai
RadarQA: Multi-modal Quality Analysis of Weather Radar Forecasts
NeurIPS 2025 ·
Xuming He
,
Zhiyuan You
,
Junchao Gong
,
Couhua Liu
,
Xiaoyu Yue
,
Peiqin Zhuang
,
Wenlong Zhang
, Lei Bai
ReSo: A Reward-driven Self-organizing LLM-based Multi-Agent System for Reasoning Tasks
EMNLP 2025 ·
Heng Zhou
,
Hejia Geng
,
Xiangyuan Xue
,
Li Kang
,
Yiran Qin
,
Zhiyong Wang
,
Zhenfei Yin
, Lei Bai
Retro-R1: LLM-based Agentic Retrosynthesis
NeurIPS 2025 ·
Wei Liu
,
Jiangtao Feng
,
Hongli Yu
,
Yuxuan Song
,
Yuqiang Li
,
Shufei Zhang
, Lei Bai,
Wei-Ying Ma
,
Hao Zhou
RoboFactory: Exploring Embodied Agent Collaboration with Compositional Constraints
ICCV 2025 ·
Yiran Qin
,
Li Kang
,
Xiufeng Song
,
Zhenfei Yin
,
Xiaohong Liu
,
Xihui Liu
,
Ruimao Zhang
, Lei Bai
SIFusion: A Unified Fusion Framework for Multi-granularity Arctic Sea Ice Forecasting
NeurIPS 2025 ·
Jingyi Xu
,
Shengnan Wang
,
Weidong Yang
,
Keyi Liu
,
Yeqi Luo
,
Ben Fei
, Lei Bai
SURVEYFORGE : On the Outline Heuristics, Memory-Driven Generation, and Multi-dimensional Evaluation for Automated Survey Writing
ACL 2025 ·
Xiangchao Yan
,
Shiyang Feng
,
Jiakang Yuan
,
Renqiu Xia
,
Bin Wang
, Lei Bai,
Bo Zhang
Satellite Observations Guided Diffusion Model for Accurate Meteorological States at Arbitrary Resolution
CVPR 2025 ·
Siwei Tu
,
Ben Fei
,
Weidong Yang
,
Fenghua Ling
,
Hao Chen
,
Zili Liu
,
Kun Chen
,
Hang Fan
,
Wanli Ouyang
, Lei Bai
Scaling Physical Reasoning with the PHYSICS Dataset
NeurIPS 2025 ·
Shenghe Zheng
,
Qianjia Cheng
,
Junchi Yao
,
Mengsong Wu
,
Haonan He
,
Ning Ding
,
Yu Cheng
,
Shuyue Hu
, Lei Bai,
Dongzhan Zhou
,
Ganqu Cui
,
Peng Ye
Scientists' First Exam: Probing Cognitive Abilities of MLLM via Perception, Understanding, and Reasoning
NeurIPS 2025 ·
Yuhao Zhou
,
Yiheng Wang
,
Xuming He
,
Ruoyao Xiao
,
Zhiwei Li
,
Qiantai Feng
,
Zijie Guo
,
Yuejin Yang
,
Hao Wu
,
Wenxuan Huang
,
Jiaqi Wei
,
Dan Si
,
Xiuqi Yao
,
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
TimeKAN: KAN-based Frequency Decomposition Learning Architecture for Long-term Time Series Forecasting
ICLR 2025 ·
Songtao Huang
,
Zhen Zhao
,
Can Li
, Lei Bai
Understand Before You Generate: Self-Guided Training for Autoregressive Image Generation
NeurIPS 2025 ·
Xiaoyu Yue
,
Zidong Wang
,
Yuqing Wang
,
Wenlong Zhang
,
Xihui Liu
,
Wanli Ouyang
, Lei Bai,
Luping Zhou
UniSTD: Towards Unified Spatio-Temporal Learning across Diverse Disciplines
CVPR 2025 ·
Chen Tang
,
Xinzhu Ma
,
Encheng Su
,
Xiufeng Song
,
Xiaohong Liu
,
Wei-Hong Li
, Lei Bai,
Wanli Ouyang
,
Xiangyu Yue
VA-MoE: Variables-Adaptive Mixture of Experts for Incremental Weather Forecasting
ICCV 2025 ·
Hao Chen
,
Tao Han
,
Song Guo
,
Jie Zhang
,
Yonghan Dong
,
Yue Yu
, Lei Bai
VAE-Var: Variational Autoencoder-Enhanced Variational Methods for Data Assimilation in Meteorology
ICLR 2025 ·
Yi Xiao
,
Qilong Jia
,
Kun Chen
, Lei Bai,
Wei Xue
VIKI‑R: Coordinating Embodied Multi-Agent Cooperation via Reinforcement Learning
NeurIPS 2025 ·
Li Kang
,
Xiufeng Song
,
Heng Zhou
,
Yiran Qin
,
Jie Yang
,
Xiaohong Liu
,
Philip H. S. Torr
, Lei Bai,
Zhenfei Yin
VLIPP: Towards Physically Plausible Video Generation with Vision and Language Informed Physical Prior
ICCV 2025 ·
Xindi Yang
,
Baolu Li
,
Yiming Zhang
,
Zhenfei Yin
, Lei Bai,
Liqian Ma
,
Zhiyong Wang
,
Jianfei Cai
,
Tien-Tsin Wong
,
Huchuan Lu
,
Xu Jia
VQLTI: Long-Term Tropical Cyclone Intensity Forecasting with Physical Constraints
AAAI 2025 ·
Xinyu Wang
,
Lei Liu
,
Kang Chen
,
Tao Han
,
Bin Li
, Lei Bai
WeatherGFM: Learning a Weather Generalist Foundation Model via In-context Learning
ICLR 2025 ·
Xiangyu Zhao
,
Zhiwang Zhou
,
Wenlong Zhang
,
Yihao Liu
,
Xiangyu Chen
,
Junchao Gong
,
Hao Chen
,
Ben Fei
,
Shiqi Chen
,
Wanli Ouyang
,
Xiao-Ming Wu
, Lei Bai
WorldSimBench: Towards Video Generation Models as World Simulators
ICML 2025 ·
Yiran Qin
,
Zhelun Shi
,
Jiwen Yu
,
Xijun Wang
,
Enshen Zhou
,
Lijun Li
,
Zhenfei Yin
,
Xihui Liu
,
Lu Sheng
,
Jing Shao
, Lei Bai,
Ruimao Zhang
scMRDR: A scalable and flexible framework for unpaired single-cell multi-omics data integration
NeurIPS 2025 ·
Jianle Sun
,
Chaoqi Liang
,
Ran Wei
,
Peng Zheng
, Lei Bai,
Wanli Ouyang
,
Hongliang Yan
,
Peng Ye
Autoregressive Omni-Aware Outpainting for Open-Vocabulary 360-Degree Image Generation
AAAI 2024 ·
Zhuqiang Lu
,
Kun Hu
,
Chaoyue Wang
, Lei Bai,
Zhiyong Wang
CasCast: Skillful High-resolution Precipitation Nowcasting via Cascaded Modelling
ICML 2024 ·
Junchao Gong
, Lei Bai,
Peng Ye
,
Wanghan Xu
,
Na Liu
,
Jianhua Dai
,
Xiaokang Yang
,
Wanli Ouyang
FNP: Fourier Neural Processes for Arbitrary-Resolution Data Assimilation
NeurIPS 2024 ·
Kun Chen
,
Peng Ye
,
Hao Chen
,
Kang Chen
,
Tao Han
,
Wanli Ouyang
,
Tao Chen
, Lei Bai
FiT: Flexible Vision Transformer for Diffusion Model
ICML 2024 ·
Zeyu Lu
,
Zidong Wang
,
Di Huang
,
Chengyue Wu
,
Xihui Liu
,
Wanli Ouyang
, Lei Bai
G-Refine: A General Quality Refiner for Text-to-Image Generation
ACM MM 2024 ·
Chunyi Li
,
Haoning Wu
,
Hongkun Hao
,
Zicheng Zhang
,
Tengchuan Kou
,
Chaofeng Chen
, Lei Bai,
Xiaohong Liu
,
Weisi Lin
,
Guangtao Zhai
Generalizing Weather Forecast to Fine-grained Temporal Scales via Physics-AI Hybrid Modeling
NeurIPS 2024 ·
Wanghan Xu
,
Fenghua Ling
,
Wenlong Zhang
,
Tao Han
,
Hao Chen
,
Wanli Ouyang
, Lei Bai
Instruct-ReID: A Multi-Purpose Person Re-Identification Task with Instructions
CVPR 2024 ·
Weizhen He
,
Yiheng Deng
,
Shixiang Tang
,
Qihao Chen
,
Qingsong Xie
,
Yizhou Wang
, Lei Bai,
Feng Zhu
,
Rui Zhao
,
Wanli Ouyang
,
Donglian Qi
,
Yunfeng Yan
MotionGPT: Finetuned LLMs Are General-Purpose Motion Generators
AAAI 2024 ·
Yaqi Zhang
,
Di Huang
,
Bin Liu
,
Shixiang Tang
,
Yan Lu
,
Lu Chen
, Lei Bai,
Qi Chu
,
Nenghai Yu
,
Wanli Ouyang
Non-Neighbors Also Matter to Kriging: A New Contrastive-Prototypical Learning
AISTATS 2024 ·
Zhishuai Li
,
Yunhao Nie
,
Ziyue Li
, Lei Bai,
Yisheng Lv
,
Rui Zhao
On Learning Multi-Modal Forgery Representation for Diffusion Generated Video Detection
NeurIPS 2024 ·
Xiufeng Song
,
Xiao Guo
,
Jiache Zhang
,
Qirui Li
, Lei Bai,
Xiaoming Liu
,
Guangtao Zhai
,
Xiaohong Liu
PredBench: Benchmarking Spatio-Temporal Prediction Across Diverse Disciplines
ECCV 2024 ·
Zidong Wang
,
Zeyu Lu
,
Di Huang
,
Tong He
,
Xihui Liu
,
Wanli Ouyang
, Lei Bai
Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled Perspective
AAAI 2024 ·
Binwu Wang
,
Pengkun Wang
,
Yudong Zhang
,
Xu Wang
,
Zhengyang Zhou
, Lei Bai,
Yang Wang
Towards a Self-contained Data-driven Global Weather Forecasting Framework
ICML 2024 ·
Yi Xiao
, Lei Bai,
Wei Xue
,
Hao Chen
,
Kun Chen
,
Kang Chen
,
Tao Han
,
Wanli Ouyang