1. CaMeL-Net: Centroid-aware metric learning for efficient multi-class cancer classification in pathology images.
- Author
-
Lee, Jaeung, Han, Chiwon, Kim, Kyungeun, Park, Gi-Ho, and Kwak, Jin Tae
- Subjects
- *
TUMOR classification , *CONVOLUTIONAL neural networks , *IMAGE recognition (Computer vision) , *ARTIFICIAL neural networks , *CARRAGEENANS , *STOMACH cancer - Abstract
• We design an efficient neural network, so called CaMeL-Net, to predict cancer grades in pathology images. • We propose to utilize centroids of different classes to calculate relative distances between input images and use them for the optimization of the network via metric learning. • We propose centroid-aware margin loss for efficient and effective metric learning that can exploit not only positive and negative samples but also centroids of different classes. • We propose a simple yet effective strategy, called exponential margin scheduling (EMS), that improves feature representations in the embedding space by gradually increasing the margin in metric learning. • We evaluate CaMeL-Net on multiple cancer datasets, achieving the state-of-the-art multi-class cancer classification performance and obtaining superior efficiency during training and inference. Cancer grading in pathology image analysis is a major task due to its importance in patient care, treatment, and management. The recent developments in artificial neural networks for computational pathology have demonstrated great potential to improve the accuracy and quality of cancer diagnosis. These improvements are generally ascribable to the advance in the architecture of the networks, often leading to increase in the computation and resources. In this work, we propose an efficient convolutional neural network that is designed to conduct multi-class cancer classification in an accurate and robust manner via metric learning. We propose a centroid-aware metric learning network for an improved cancer grading in pathology images. The proposed network utilizes centroids of different classes within the feature embedding space to optimize the relative distances between pathology images, which manifest the innate similarities/dissimilarities between them. For improved optimization, we introduce a new loss function and a training strategy that are tailored to the proposed network and metric learning. We evaluated the proposed approach on multiple datasets of colorectal and gastric cancers. For the colorectal cancer, two different datasets were employed that were collected from different acquisition settings. the proposed method achieved an accuracy, F1-score, quadratic weighted kappa of 88.7%, 0.849, and 0.946 for the first dataset and 83.3%, 0.764, and 0.907 for the second dataset, respectively. For the gastric cancer, the proposed method obtained an accuracy of 85.9%, F1-score of 0.793, and quadratic weighted kappa of 0.939. We also found that the proposed method outperforms other competing models and is computationally efficient. The experimental results demonstrate that the prediction results by the proposed network are both accurate and reliable. The proposed network not only outperformed other related methods in cancer classification but also achieved superior computational efficiency during training and inference. The future study will entail further development of the proposed method and the application of the method to other problems and domains. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF