DDPM-SKDNet: A Deep Learning Method for ICG Image Classification

Yuhao Wang,Yan Pei,Yong Li,Bo Liu,Bin Yang,Jianqiang Li
DOI: https://doi.org/10.1109/SMC53992.2023.10394412
2023-10-01
Abstract:Over the past several years, deep learning technologies have made tremendous progress in medical image tasks including classification, segmentation, and object detection. However, there are two main limitations of indocyanine green (ICG) images which are often used in breast cancer related lymphedema (BCRL): insufficient sample numbers and low image quality. Consequently, the conventional deep learning based classification methods such as ResNet have faced challenges in achieving satisfactory results. To tackle the concern, this paper puts forward a deep learning method named Denoising Diffusion Probabilistic Model Self-supervised Knowledge Distillation Net (DDPM-SKDNet) for the ICG images classification task, by incorporating a contrastive learning based approach as the network architecture and using DDPM as the image generator in the contrastive module to expand the dataset size. Furthermore, a knowledge distillation approach is utilized to increase the effectiveness of the network. The proposed method was validated on ICG datasets and achieved a significant improvement in classification accuracy, increasing it from 66.7% in the baseline method to 82.1% in the proposed method.
Medicine,Computer Science
What problem does this paper attempt to address?