As a first step in treatment, accurate automatic cataract diagnosis is of vital importance. The classification and grading of slit -lamp images can realize the cataract type and severity diagnosis. Deep learning models are widely applied in existing methods. However, a key challenge to improve the performance is the noise existed in medical images. Inspired by the ability of the denoising diffusion probabilistic model to generate noise -robust features in image generation tasks, this work develops a new method for cataract classification to learn and share complementary representations among multiple tasks. To alleviate the existence of general noise, a dual -branch network is proposed to combine the image generation based on the denoising diffusion probabilistic model and the target classification task effectively. A cross fusion module is further designed by two cross attention to enhance the interaction of features generated from two branches. Compared to state-ofthe-art methods, the proposed model improves the performance by a significant margin on three classification datasets and has a more robust tolerance with noise interference. Most notably, for multi -granularity cataract classification, it achieves 73.86% in Recall, 81.18% in Precision, 76.94% in F1 -Score, and 81.79% in Accuracy, which surpasses the performance of the second -place model by 7.43%, 7.07%, 6.76% and 2.37% respectively.
基金:
Natural Science Foundation of China [U20A20196, U1908210, 62002326]
第一作者机构:[1]Zhejiang Univ Technol, Sch Comp Sci & Technol, Hangzhou, Peoples R China
通讯作者:
推荐引用方式(GB/T 7714):
Wang Qiong,Sun Hongdi,Feng Yu,et al.MGCNet: Multi-granularity cataract classification using denoising diffusion probabilistic model ☆[J].DISPLAYS.2024,83:doi:10.1016/j.displa.2024.102716.
APA:
Wang, Qiong,Sun, Hongdi,Feng, Yu,Dong, Zhe&Bai, Cong.(2024).MGCNet: Multi-granularity cataract classification using denoising diffusion probabilistic model ☆.DISPLAYS,83,
MLA:
Wang, Qiong,et al."MGCNet: Multi-granularity cataract classification using denoising diffusion probabilistic model ☆".DISPLAYS 83.(2024)