Weigao Sun
Weigao Sun

Young Scientist

Shanghai AI Laboratory

About Me

I am a Young Scientist at Shanghai AI Laboratory. Currently, I fortunately collaborate with Prof. Yu Cheng, work on Efficient AI for LLMs/VLMs, including algorithm & system co-design on Efficient Architectures, e.g., Linear Attention and Mixture-of-Experts, and Large-Scale Distributed Training.

See our research projects: Linear-MoE, LASP-2, Linearization (Liger), MoM, NHA and Comba for technical details. My previous work Lightning Attention and LASP series are the key techniques in MiniMax-01 456B LLM and VLM.

🔥 I am looking for talented interns to work with me on above projects and beyond. Please feel free to hit me up with your CV or any questions if you are interested.

From 2020 to 2022, I was an AI Researcher at Linx Lab, 2012 Lab, Huawei, supervised by Jiashu Lin and Dr. Heng Liao, worked on large scale distributed training algorithms. I earned my PhD degree from Huazhong University of Science and Technology (HUST) at 2020.04, co-supervised by Prof. Hai-Tao Zhang and Prof. Ye Yuan, and jointly trained at School of Artificial Intelligence and Automation (AIA) and HUST Innovation Institute (with the First-class Grant).

(Updated at 2025.05)

Interests
  • Efficient Sequence Modeling (e.g., Linear Attention, Sparse Attention)
  • Mixture-of-Experts
  • Large Language Model
  • Vision Language Model
Recent Papers
Xiaoye Qu, Yafu Li, Zhaochen Su, Weigao Sun, Jianhao Yan, Dongrui Liu, Ganqu Cui, Daizong Liu, Shuxian Liang, Junxian He, Peng Li, Wei Wei, Jing Shao, Chaochao Lu, Yue Zhang, Xian-Sheng Hua, Bowen Zhou, Yu Cheng (2025). A Survey of Efficient Reasoning for Large Reasoning Models: Language, Multimodality, and Beyond. In arXiv preprint arXiv:2503.21614.
PDF Cite Code
Weigao Sun, Disen Lan, Tong Zhu, Xiaoye Qu, Yu Cheng (2025). Linear-MoE: Linear Sequence Modeling Meets Mixture-of-Experts. In ICLR 2025-SCOPE Workshop (Oral).
PDF Cite Code
Disen Lan, Weigao Sun, Jiaxi Hu, Jusen Du, Yu Cheng (2025). Liger: Linearizing Large Language Models to Gated Recurrent Structures. In arXiv preprint arXiv:2503.01496.
PDF Cite Code
Jusen Du, Weigao Sun, Disen Lan, Jiaxi Hu, Yu Cheng (2025). MoM: Linear Sequence Modeling with Mixture-of-Memories. In arXiv preprint arXiv:2502.13685.
PDF Cite Code
Weigao Sun, Disen Lan, Yiran Zhong, Xiaoye Qu, Yu Cheng (2025). LASP-2: Rethinking Sequence Parallelism for Linear Attention and Its Hybrid. In arXiv preprint arXiv:2502.07563.
PDF Cite Code
Weigao Sun, Et Al (2025). Minimax-01: Scaling Foundation Models with Lightning Attention. In arXiv preprint arXiv:2501.08313.
PDF Cite Code
See all publications
Recent News