Haochen Zhang

Haochen Zhang

Ph.D. student in ECE at UCSD. Master degree received from USTC in 2020.

Class-Conditioned Image Synthesis with Diffusion for Imbalanced Diabetic Retinopathy Grading

Published in International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI), 2025

Overview of our text-to-image diffusion model finetuning framework. The top section illustrates the process of stabilizing diffusion model finetuning on fundus dataset with limited data. The bottom right depicts our proposes semantic quality evaluation and filtering pipeline. The entire image demonstrates how semantic quality is enhanced through explicit class conditioning during diffusion model training.
Overview of our text-to-image diffusion model finetuning framework. The top section illustrates the process of stabilizing diffusion model finetuning on fundus dataset with limited data. The bottom right depicts our proposes semantic quality evaluation and filtering pipeline. The entire image demonstrates how semantic quality is enhanced through explicit class conditioning during diffusion model training.

In this paper, …

Paper | Poster | Code

Citation:

@inproceedings{zhang2025class,
  title={Class-Conditioned Image Synthesis with Diffusion for Imbalanced Diabetic Retinopathy Grading},
  author={Zhang, Haochen and Heinke, Anna and Nagel, Ines D and Bartsch, Dirk-Uwe G and Freeman, William R and Nguyen, Truong Q and An, Cheolhong},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={56--66},
  year={2025}
}