Me
Yu Lei (雷宇)
CS Ph.D. Student @ ICT, CAS
leiyu2648 [at] gmail.com

About

Yu Lei (雷宇) is a first-year Ph.D. student at the Visual Information Processing and Learning (VIPL) Lab, Institute of Computing Technology, Chinese Academy of Sciences (ICT, CAS), where he explores AIGC, video generation, world models, and preference learning. His current research focuses on personalized image generation, preference optimization for visual generative models, and physically consistent world models.

Before joining ICT, Yu obtained his B.Eng. in Artificial Intelligence from Sichuan University (June 2025), where he began his journey in weakly supervised learning for medical imaging. As a research intern at TeleAI, China Telecom, he worked on personalized safety alignment for text-to-image diffusion models, preference optimization for visual generative models, and world-model research under the mentorship of Dr. Jinbin Bai and Dr. Kaidong Yu.

"I believe AI should understand what humans truly want, not just what they say they want."

Research Interests

  • Personalized & Safe AI Generation
  • Preference Optimization for Generative Models
  • World Models & Physical Consistency
  • Medical Image Segmentation (Previous)

Education

ICT Logo
Institute of Computing Technology (Sep. 2025 – Now)
Ph.D. in Computer Science and Technology
SCU Logo
Sichuan University (Sep. 2021 – Jun. 2025)
B.Eng in Artificial Intelligence

Experiences

Feb, 2025 - Apr, 2026
Research Intern @ TeleAI
Sep, 2022 - Aug, 2024
Research Assistant @ MachineILab

Publications ( / )

PSAlign
Personalized Safety Alignment for Text-to-Image Diffusion Models
Yu Lei, Jinbin Bai, Qingyu Shi, Aosong Feng, Kaidong Yu
Submitted to TMLR 2026
TGO
Threshold-Guided Optimization for Visual Generative Models
Jinbin Bai, Yu Lei, Qingyu Shi, Aosong Feng, Yi Xin, Zhuoran Zhao, Fei Shen, Kaidong Yu, Jason Li
ICML 2026
[Paper]
World Models
From Masks to Worlds: A Hitchhiker's Guide to World Models
Jinbin Bai, Yu Lei, Hecong Wu, Yuchen Zhu, Shufan Li, Yi Xin, Xiangtai Li, Molei Tao, Aditya Grover, Mingxuan Yang
arXiv 2025
PCLMix
PCLMix: Weakly Supervised Medical Image Segmentation via Pixel-Level Contrastive Learning and Dynamic Mix Augmentation
Yu Lei, Haolun Luo, Lituan Wang, Zhenwei Zhang, Lei Zhang
ICIC 2024
[Paper] [Code]

Projects

PSAlign: Personalized Safety Alignment for Text-to-Image Diffusion Models.
PCLMix: Weakly supervised medical image segmentation via pixel-level contrastive learning and dynamic mix augmentation (ICIC 2024).
Awesome World Models: A Hitchhiker's Guide.

Selected Honors & Awards

  • 1st Place, Phase 2, Image-to-Video Consistency Challenge · CVPR 2026 VGBE Workshop · 2026
  • Outstanding Graduate · Sichuan University · 2024–2025
  • National Scholarship · Sichuan University · 2023–2024
  • Baogang Scholarship (Top 6 undergraduates) · Sichuan University · 2023–2024
  • First Prize, Global Final · Huawei ICT Competition · 2023–2024

Academic Services

Conference Reviewer:
International Joint Conference on Neural Networks, IJCNN 2024
Top