CV

Education

Shanghai Jiao Tong University2025.09–2030.06

Ph.D. Student in Computer Science and Technology, School of Artificial Intelligence
Advisor: Prof. Zhouhan Lin, LUMIA Group

Wuhan University2021.09–2025.06

B.E. in Information Security, School of Cyber Science and Engineering
GPA: 3.91/4.0 · Rank: 4/144

Intern-S2-Preview: Scientific Agentic Foundation ModelTechnical Report · 2026

Intern-S2-Preview Team, Shanghai AI Laboratory

Memory Decoder at Scale: A Pretrained, Parametric Long-Term MemoryPreprint · 2026

Rubin Wei, Jiaqi Cao, Jiarui Wang, Junming Zhang, Qipeng Guo, Bowen Zhou, Zhouhan Lin

MemSFT: Mitigating Alignment Tax with an External Parametric MemoryPreprint · 2026

Jiarui Wang*, Xiang Shi*, Jiaqi Cao, Rubin Wei, Xiquan Wang, Hao Sun, Jingzhi Wang, Zhiqi Yang, Qipeng Guo, Bowen Zhou, Zhouhan Lin

MLP Memory: A Retriever-Pretrained Memory for Large Language ModelsICLR 2026

Rubin Wei*, Jiaqi Cao*, Jiarui Wang, Jushi Kai, Qipeng Guo, Bowen Zhou, Zhouhan Lin

Memory Decoder: A Pretrained, Plug-and-Play Memory for Large Language ModelsNeurIPS 2025

Jiaqi Cao*, Jiarui Wang*, Rubin Wei, Qipeng Guo, Kai Chen, Bowen Zhou, Zhouhan Lin

Beowulf: Mitigating Model Extraction Attacks via Reshaping Decision RegionsACM CCS 2024

Xueluan Gong, Rubin Wei, Ziyao Wang, Yuchen Sun, Jiawen Peng, Yanjiao Chen, Qian Wang

Augmenting Model Extraction Attacks Against Disruption-Based DefensesIEEE TIFS 2024

Xueluan Gong, Shuaike Li, Yanjiao Chen, Mingzhe Li, Rubin Wei, Qian Wang, Kwok-Yan Lam

* Equal Contribution

Shanghai AI Lab · Research Intern2026.04–Present

Pre-training research for foundation models.

TeleAI · Research Intern2024.11–2025.02

Research on efficient vision-language models.

Shanghai AI Lab · Research Intern2024.04–2024.06

Research on multimodal vision-language models.

2025

Outstanding Undergraduate Thesis Award

2023

National Scholarship