1. Prior Guided Fundus Image Quality Enhancement Via Contrastive Learning
- Author
-
Yijin Huang, Pujin Cheng, Xiaoying Tang, Li Lin, and Junyan Lyu
- Subjects
Network architecture ,business.industry ,Image quality ,Computer science ,Pattern recognition ,010501 environmental sciences ,Fundus (eye) ,Translation (geometry) ,Semantics ,01 natural sciences ,030218 nuclear medicine & medical imaging ,Domain (software engineering) ,Constraint (information theory) ,03 medical and health sciences ,0302 clinical medicine ,Domain knowledge ,Artificial intelligence ,business ,0105 earth and related environmental sciences - Abstract
Fundus images of poor quality may seriously influence clinic judgments. Existing fundus image quality enhancement (FIQE) approaches mainly make use of general image features but no prior domain knowledge. In this paper, we proposed and validated an efficient FIQE method with a prior constraint, named Efficient Prior Contrastive unpaired Generative Adversarial Network (EPC-GAN). Inspired by the contrastive unpaired translation framework, we emphasized local features during the enhancing process via contrastive patchwise samples. Moreover, to utilize high-level features in the fundus domain (such as vessels, optic disc/cup, and even lesions), we designed a fundus prior loss to avoid information modification and over-enhancement. Besides, we presented an efficient network architecture to overcome the high consumption in terms of both time and GPU-memory. Through both qualitative and quantitative experiments on a public dataset EyeQ, we demonstrated the superior performance of our proposed method.
- Published
- 2021