EDSNet: A Deep Supervision-Based Classification Framework for Non-Standard Medical Images of Lung Cancer

Ziming Lin,Zekai Huang,Chengchuang Lin,Zhaoliang Zheng,Youqun Wang,Hongwei Lin,Gansen Zhao,Haiyu Zhou
DOI: https://doi.org/10.1109/ijcnn60899.2024.10649976
2024-01-01
Abstract:Lung cancer is a most prevalent and deadly type of cancer worldwide. In clinical practice, it is crucial to accurately identify the risk level of lung cancer for diagnosing and treating lung lesions. Current AI methods for lung cancer image diagnosis mainly rely on CT images, histopathological images, and a few methods based on surgical lesion section images. However, there is a great need for methods for surgical lesion section images that can help surgeons quickly decide the surgical plan during the operation. Deep learning methods among them often supervise only the output layer of the network, while ignoring the constraints on the intermediate hidden layers, resulting in insufficient accuracy of high-risk and low-risk classification. This paper aims to design a method for the task of classifying surgical lesion section of lung tumours into high-risk and low-risk categories. The EDSNet model based on deep supervision is proposed, which enhances the performance of the hidden layer through deep supervision. Moreover, the MBConv module of baseline model is replaced with the SimAM-MBConv module proposed in this paper, which further enhances the representation of pathological features by deriving 3D attention weights for feature maps within one layer. Several experiments had been conducted on a clinical lung tumour surgical lesion section image dataset, achieving the classification accuracy of EDSNet is 95.90% with an AUC of 0.9935. Compared with most existing methods for similar tasks, our method is optimal in most of the evaluation metrics. The experimental results demonstrate the effectiveness and sophistication of our method for high-risk and low-risk classification of surgical section images of lung tumours, which can serve as a reference for doctors to determine the subsequent surgical plans and treatment strategies.
What problem does this paper attempt to address?