Hi there Welcome to my Homepage!

I am a CS Ph.D. student in the ASPIRE Lab at ShanghaiTech University, advised by Prof. Wenjie Wang. Prior to that, I received my bachelor’s degree from Northwest A&F University in 2024.

My research focuses on AI safety, alignment, and post-training for LLMs and VLMs. Recently, I have been exploring latent space reasoning and looped language models (LoopLMs), with an emphasis on efficient post-training and safety alignment.

News

Publications

Preprint

Recurrent Self-Improvement: Dynamic Cross-Loop On-Policy Distillation for Looped Language Models

Yi Wang, Rui Qian, Yu Li, Haoyang Yao, Wenjie Wang^†.

Preprint

[Paper] [Code] (Coming soon) [Models] (Coming soon)

Preprint

Safe at One Loop, Risky at Another: Aligning Safety Across Recurrent Depths in Looped Language Models

Yi Wang^*, Xiuyuan Qi^*, Dongqi Han, Dongsheng Li, Wenjie Wang^†.

Preprint

[Paper] [Code] (Coming soon) [Models] (Coming soon)

NeurIPS 2026

Safety-Aware Latent Space Reasoning in Large Language Models

Yi Wang^*, Wenjie Wang^*†, Hongye Qiu, Yu Pan.

NeurIPS 2026

[Paper] [Code] [Website]

TPAMI 2026

EVA: Editing for Versatile Alignment against Jailbreaks

Yi Wang, Hongye Qiu, Yue Xu, Sibei Yang, Zhan Qin, Minlie Huang, Wenjie Wang^†.

IEEE TPAMI 2026

[Paper] [IEEE] [Code] [Models] [Website] [RedNote]

NLPCC 2026

DR.GAP: Mitigating Bias in Large Language Models using Gender-Aware Prompting with Decoupled Reasoning

Hongye Qiu^*, Yue Xu^*, Yi Wang, Meikang Qiu, Wenjie Wang^†.

NLPCC 2026

[Paper] [Code]

ACML 2026

Differential Privacy Protected Preference Alignment on Survey Derived Data

Xiuyuan Qi, Yi Wang, Wenjie Wang^†.

ACML 2026

[Paper] (Coming soon)

Preprint

GPO-V: Jailbreak Diffusion Vision Language Model by Global Probability Optimization

Yu Pan, Andi Zhang, Yi Wang, Sibei Yang, Wenjie Wang^†.

Preprint

[Paper]

ACL 2025

DELMAN: Dynamic Defense Against Large Language Model Jailbreaking with Model Editing

Yi Wang, Fenghua Weng, Sibei Yang, Zhan Qin, Minlie Huang, Wenjie Wang^†.

ACL 2025

[Paper] [Code] [Website] [RedNote]

No preprints to display yet.

Education

2024.09 - Present

ShanghaiTech University

Ph.D. Student

2020.09 - 2024.06

Northwest A&F University

Undergraduate

Experience

2026.06 - 2026.09

Microsoft Research Asia (MSRA)

Research Collaboration

Services

• [Conference Reviewer]
  • ICLR 2027
  • ACL ARR 2025, 2026
• [Journal Reviewer]
  • Information Fusion（中科院一区 TOP，JCR Q1）
  • Pattern Recognition（中科院一区 TOP，JCR Q1）
• [Open Source Contributor]
  • ByteDance / Ouro-1.4B-Thinking
    Hugging Face model repository contributor

Awards

• [2024]Outstanding Student
• [2023]🏅Provincial First Prize, 15th National College Student Mathematics Competition
• [2022]Academic Scholarship
• [2021]Outstanding Student Leader