Abstract
Pancreatic cystic neoplasms (PCNs) are relatively rare neoplasms and difficult to be classified preoperatively. Ordinary deep learning methods have great potential to provide support for doctors in PCNs classification but require a quantity of labeled samples and exact segmentation of neoplasm. The proposed metric learning-based method using graph neural network (GNN) aims to overcome the limitations brought by small and imbalanced dataset and get fast and accurate PCNs classification result from computed tomography (CT)images. The proposed framework applies GNN. GNNs perform well in fusing information and modeling relational data and get better results on dataset with small size. Based on metric learning strategy, model learns distance from the data. The similarity-based algorithm enhances the classification performance, and more characteristic information is found. We use a convolutional neural network (CNN) to extract features from given images. Then GNN is used to find the similarity between each two feature vectors and complete the classification. Several subtasks consisting of randomly selected images are established to improve generalization of the model. The experiments are carried out on the dataset provided by Huashan Hospital. The dataset is labeled by postoperative pathological analysis and contains region of interest (ROI) information calibrated by experts. We set two tasks based on the dataset: benign or malignant diagnosis of PCNs and classification of specifictypes. Our model shows good performance on the two tasks with accuracies of 88.926% and 74.497%. The comparison of different methods' F1 scores in the benign or malignant diagnosis shows that the proposed GNN-based method effectively reduces the negative impact brought by imbalanced dataset, which is also verified by the macroaverage comparison in the four-class classificationtask. Compared with existing models, the proposed GNN-based model shows better performance in terms of imbalanced dataset with small size while reducing labeling cost. The result provides a possibility for its application into the computer-aided diagnosis ofPCNs.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.