Zhang Yi

I am currently a fourth-year Ph.D. student at MMLab under the supervision of Prof. Hongsheng Li and Prof. Xiaogang Wang. Before joining CUHK, I received my bachelor's degree from Nanjing University in 2019.

** I'm currently looking for industry positions.

Email  /  Google Scholar  /  Github /  CV

profile photo

My current research interests include generative models and computational photography. I enjoy the research that can benefit real-world applications. Some of my papers have been used in some commercial products.

A Unified Conditional Framework for Diffusion-based Image Restoration
Yi Zhang, Xiaoyu Shi, Dasong Li, Xiaogang Wang, Jian Wang, Hongsheng Li
NeurIPS, 2023
arXiv / code / project

A unified conditional framework based on diffusion models for image restoration.

KBNet: Kernel Basis Network for Image Restoration
Yi Zhang, Dasong Li, Xiaoyu Shi, Dailan He, Kangning Song, Xiaogang Wang, Honwei Qin, Hongsheng Li

arXiv / code

A general-purpose backbone for image restoration tasks (e.g. denoising, deraining, and deblurring). A novel kernel basis attention (KBA) module has been proposed to effectively aggregate the spatial information via a series of learnable kernel bases.

No Attention is Needed: Grouped Spatial-temporal Shift for Simple and Efficient Video Restorers
Dasong Li, Xiaoyu Shi, Yi Zhang, Xiaogang Wang, Honwei Qin, Hongsheng Li
CVPR, 2023
arXiv / code / project

We propose a grouped spatial-temporal shift module for effective video restoration. It surpasses previous SOTA methods with only 43% computational cost on video deblurring and denoising.

Efficient Burst Raw Denoising with Variance Stabilization and Multi-frequency Denoising Network
Dasong Li*, Yi Zhang*, Ka Lung Law, Xiaogang Wang, Honwei Qin, Hongsheng Li
IJCV, 2022

A practical low-light imaging system for burst raw image denoising. It has been deployed to some commercial smartphones.

Learning Degradation Representations for Image Deblurring
Dasong Li, Yi Zhang, Ka Chun Cheung, Xiaogang Wang, Honwei Qin, Hongsheng Li
ECCV, 2022
arXiv / code

Learning degradation representations for blurry images. The learned degradation representation shows pleasing improvements on both image deblurring and reblurring.

Self-Supervised Image Denoising via Iterative Data Refinement
Yi Zhang, Dasong Li, Ka Lung Law, Xiaogang Wang, Honwei Qin, Hongsheng Li
CVPR, 2022
arXiv / code / SenseNoise dataset

A self-supervised image denoising method using noisy images and the noise model. We achieve SOTA results on both real-world noise and synthetic noises (both point-wise and spatial correlated noise types).

Rethinking Noise Synthesis and Modeling in Raw Denoising
Yi Zhang, Honwei Qin, Xiaogang Wang, Hongsheng Li
ICCV, 2021
arXiv / code

A method to synthesize "almost the most realistic" raw image noise by sampling directly from the real noise distribution. We also found existing comparisons of noise synthesis methods are based on inaccurate noise parameters.

On Reinforcement Learning for Full-length Game of StarCraft
Zhen-Jia Pang, Ruo-Ze Liu, Zhou-Yu Meng, Yi Zhang, Yang Yu, Tong Lu
AAAI, 2019   (Oral)
arXiv / demo

We achieve over 93% winning rate of Protoss against the most difficult non-cheating built-in AI (level-7) of Terran, training within two days using a single machine with only 48 CPU cores and 8 K40 GPUs.

Academic Services

  • Conference Reviewer: CVPR, ICCV, ECCV, ICML, NeurIPS, ICLR, AAAI
  • Journal Reviewer: TIP
  • Experience

  • Research intern at Snap Research, 2023.3 ~ 2023.6
  • Research intern at LAMDA Group, 2017 ~ 2018
  • Honors

  • Postgraduate Scholarship, the Chinese University of Hong Kong, 2019 ~ Present
  • Outstanding Graduate of Nanjing University, 2019
  • Merit student in Jiangsu Province, 2018
  • National Scholarship, 2018

  • The website template was borrowed from here.