Under Construction…

I’m currently a master’s student at Beijing Jiaotong University, co-supervised by Prof.Yunchao Wei and Prof. Yao Zhao.

My research interest includes Unified Model, Personalized Image Generation, Image Matting.

📖 Educations

• 2023.09 - (now), Master’s Student. Beijing Jiaotong University, China.
• 2019.09 - 2023.06, Bachelor. Beijing Jiaotong University, China.

💻 Internships

• 2024.12 - now, MT Lab, Meitu Inc., China.

📝 Publications

Arxiv 2025

AlignGen: Boosting Personalized Image Generation with Cross-Modality Prior Alignment.

Yiheng Lin*, Shifang Zhao*, Ting Liu, Xiaochao Qu, Luoqi Liu, Yao Zhao, Yunchao Wei

Paper | GitHub

Keywords: Personalized Image Generation; DiT; In-Context Generation

AAAI 2025

Memory Efficient Matting with Adaptive Token Routing.

Yiheng Lin, Yihan Hu, Chenyi Zhang, Ting Liu, Xiaochao Qu, Luoqi Liu, Yao Zhao

Paper | GitHub | Dataset

Keywords: Image Matting; Vision Transformer; High-Resolution

ECCV 2024

Diffusion for Natural Image Matting.

Yihan Hu, Yiheng Lin, Wei Wang, Yao Zhao, Yunchao Wei, Humphrey Shi.

Paper | GitHub

Keywords: Image Matting; Diffusion

Under Review

MattePro: A Professional Matting Model for Anything.

Chenyi Zhang*, Yiheng Lin*, Ting Liu, Xiaochao Qu, Luoqi Liu, Fang Zhao, Yao Zhao, Yunchao.

Paper (Coming Soon) | GitHub

Keywords: Interactive Image Matting

• OmniAD: Detect and Understand Industrial Anomaly via Multimodal Reasoning.
  Shifang Zhao, Yiheng Lin, Lu Han, Yao Zhao, Yunchao Wei

• DCEdit: Dual-Level Controlled Image Editing via Precisely Localized Semantics.
  Yihan Hu, Jianing Peng, Yiheng Lin, Ting Liu, Xiaochao Qu, Luoqi Liu, Yao Zhao, Yunchao Wei