Deep fine-KNN classification of ovarian cancer subtypes using efficientNet-B0 extracted features: a comprehensive analysis

Abstract

This study presents a robust approach for the classification of ovarian cancer subtypes through the integration of deep learning and k-nearest neighbor (KNN) methods. The proposed model leverages the powerful feature extraction capabilities of EfficientNet-B0, utilizing its deep features for subsequent fine-grained classification using the fine-KNN approach. The UBC-OCEAN dataset, encompassing histopathological images of five distinct ovarian cancer subtypes, namely, high-grade serous carcinoma (HGSC), clear-cell ovarian carcinoma (CC), endometrioid carcinoma (EC), low-grade serous carcinoma (LGSC), and mucinous carcinoma (MC), served as the foundation for our investigation. With a dataset comprising 725 images, divided into 80% for training and 20% for testing, our model exhibits exceptional performance. Both the validation and testing phases achieved 100% accuracy, underscoring the efficacy of the proposed methodology. In addition, the area under the curve (AUC), a key metric for evaluating the model’s discriminative ability, demonstrated high performance across various subtypes, with AUC values of 0.94, 0.78, 0.69, 0.92, and 0.94 for MC. Furthermore, the positive likelihood ratios (LR+) were indicative of the model’s diagnostic utility, with notable values for each subtype: CC (27.294), EC (9.441), HGSC (12.588), LGSC (17.942), and MC (17.942). These findings demonstrate the effectiveness of the model in distinguishing between ovarian cancer subtypes, positioning it as a promising tool for diagnostic applications. The demonstrated accuracy, AUC values, and LR+ values underscore the potential of the model as a valuable diagnostic tool, contributing to the advancement of precision medicine in the field of ovarian cancer research.