Classification of breast tumor models with a prototype microwave imaging system
Raquel C Conceição,Hugo Medeiros,Daniela M Godinho,Martin O'Halloran,Diego Rodriguez-Herrera,Daniel Flores-Tapia,Stephen Pistorius
DOI: https://doi.org/10.1002/mp.14064
Abstract:Purpose: The assessment of the size and shape of breast tumors is of utter importance to the correct diagnosis and staging of breast cancer. In this paper, we classify breast tumor models of varying sizes and shapes using signals collected with a monostatic ultra-wideband radar microwave imaging prototype system with machine learning algorithms specifically tailored to the collected data. Methods: A database comprising 13 benign and 13 malignant tumor models with sizes between 13 and 40 mm was created using dielectrically representative tissue mimicking materials. These tumor models were placed inside two breast phantoms: a homogeneous breast phantom and a breast phantom with clusters of fibroglandular mimicking tissue, accounting for breast heterogeneity. The breast phantoms with tumors were imaged with a monostatic microwave imaging prototype system, over a 1-6 GHz frequency range. The classification of benign and malignant tumors embedded in the two breast phantoms was completed, and tumor classification was evaluated with Principal Component Analysis as a feature extraction method, and tuned Naïve Bayes (NB), decision trees (DT), and k-nearest neighbours (kNN) as classifiers. We further study which antenna positions are better placed to classify tumors, discuss the feature extraction method and optimize classification algorithms, by tuning their hyperparameters, to improve sensitivity, specificity and the receiver operating characteristic curve, while ensuring maximum generalization and avoiding overfitting and data contamination. We also added a realistic synthetic skin response to the collected signals and examined its global effect on classification of benign vs malignant tumors. Results: In terms of global classification performance, kNN outperformed DT and NB machine learning classifiers, achieving a classification accuracy of 96.2% when classifying between benign and malignant tumor phantoms in a homogeneous breast phantom (both when the skin artifact is and is not considered). Conclusions: We experimentally classified tumor models as benign or malignant with a microwave imaging system, and we showed a methodology that can potentially assess the shape of breast tumors, which will give further insight into the correct diagnosis and staging of breast cancer.