IDC-Net: Breast cancer classification network based on BI-RADS 4

Sanli Yi,Ziyan Chen,Furong She,Tianwei Wang,Xuelian Yang,Dong Chen,Xiaomao Luo
DOI: https://doi.org/10.1016/j.patcog.2024.110323
IF: 8
2024-02-15
Pattern Recognition
Abstract:In the diagnosis of breast cancer, the 3 sub-categories 4a-4c of BI-RADS 4 are of great significance to doctors. However, low resolution of ultrasound image and high similarity between different category images pose great challenges to this task, which requires the network to be more capable of extracting image features. Therefore, in response to the efficient classification of BI-RADS 4a-4c in breast ultrasound images, we developed a lightweight classification network IDCNet, a neural network model combining the advantages of convolutional neural network(CNN) and CapsNet. In this model: Firstly, we proposed ID-Net based on CNN architecture and mainly constructed by ID block and DD block, which ensure the ID-Net deep and wide enough to extract sufficient local semantic information of image, and at the same time being lightweight. Secondly, we use the CapsNet to learn the position and posture information between the global features of the image, which makes up for the defects of CNN. Finally, two parallel paths of IDCNet and CapsNet are fused to enhance IDCNet's capability of feature extraction. To verify our method, experiments have been conducted on the breast ultrasound dataset of Yunnan cancer hospital and two public datasets. The classification results of our method have been compared with those obtained by five existing approaches. The experimental results show that the proposed method IDCNet has the highest Accuracy (98.54 %), Precision (98.54 %) and F1 score (98.54 %).
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?