A diffusion model for universal medical image enhancement.
Authors
Affiliations (5)
Affiliations (5)
- College of Computer Science and Artificial Intelligence, Fudan University, Shanghai, 200438, China.
- College of Computer Science and Artificial Intelligence, Fudan University, Shanghai, 200438, China. [email protected].
- Institute of Digestive Disease, School of Medicine, Tongji University, Shanghai, 200065, China. [email protected].
- National Engineering Center for Biochip, Shanghai, 201203, China.
- Endoscopy Center of Zhongshan Hospital, Fudan University, Shanghai, 200032, China.
Abstract
The development of medical imaging techniques has made a significant contribution to clinical decision-making. However, the existence of suboptimal imaging quality, as indicated by irregular illumination or imbalanced intensity, presents significant obstacles in automating disease screening, analysis, and diagnosis. Existing approaches for natural image enhancement are mostly trained with numerous paired images, presenting challenges in data collection and training costs, all while lacking the ability to generalize effectively. Here, we introduce a pioneering training-free Diffusion Model for Universal Medical Image Enhancement, named UniMIE. UniMIE demonstrates its unsupervised enhancement capabilities across various medical image modalities without the need for any fine-tuning. It accomplishes this by relying solely on a single pre-trained model from ImageNet. We conduct a comprehensive evaluation on 13 imaging modalities and over 15 medical types, demonstrating better qualities, robustness, and accuracy than other modality-specific and data-inefficient models. By delivering high-quality enhancement and corresponding accuracy downstream tasks across a wide range of tasks, UniMIE exhibits considerable potential to accelerate the advancement of diagnostic tools and customized treatment plans. UniMIE represents a transformative approach to medical image enhancement, offering a versatile and robust solution that adapts to diverse imaging conditions. By improving image quality and facilitating better downstream analyses, UniMIE has the potential to revolutionize clinical workflows and enhance diagnostic accuracy across a wide range of medical applications.