Breast Tumor Detection and Classification in Mammogram Images Using Modified YOLOv5 Network

Abstract

Breast cancer incidence has been rising steadily during the past few decades. It is the second leading cause of death in women. If it is diagnosed early, there is a good possibility of recovery. Mammography is proven to be an excellent screening technique for breast tumor diagnosis, but its detection and classification in mammograms remain a significant challenge. Previous studies’ major limitation is an increase in false positive ratio (FPR) and false negative ratio (FNR), as well as a drop in Matthews correlation coefficient (MCC) value. A model that can lower FPR and FNR while increasing MCC value is required. To overcome prior research limitations, a modified network of YOLOv5 is used in this study to detect and classify breast tumors. Our research is conducted using publicly available datasets Curated Breast Imaging Subset of DDSM (CBIS-DDSM). The first step is to perform preprocessing, which includes image enhancing techniques and the removal of pectoral muscles and labels. The dataset is then annotated, augmented, and divided into 60% for training, 30% for validation, and 10% for testing. The experiment is then performed using a batch size of 8, a learning rate of 0.01, a momentum of 0.843, and an epoch value of 300. To evaluate the performance of our proposed model, our proposed model is compared with YOLOv3 and faster RCNN. The results show that our proposed model performs better than YOLOv3 and faster RCNN with 96% mAP, 93.50% MCC value, 96.50% accuracy, 0.04 FPR, and 0.03 FNR value. The results show that our suggested model successfully identifies and classifies breast tumors while also overcoming previous research limitations by lowering the FPR and FNR and boosting the MCC value.

Discovery of Knowledge in the Incidence of a Type of Lung Cancer for Patients through Data Mining Models

“…As a result of using the algorithms in the WEKA data mining software on the data set, first of all, detailed information was obtained about the algorithm that can make the most appropriate prediction estimation. Afterward, this algorithm was accessed from the developed software, the model was created, and a meaningful result was tried to be obtained [ 13 ]. During the development of the software, attention was paid to ensuring that it had the following criteria: User-friendliness of the software User (doctor or health personnel) can enter blood value test results into the system simply.…”

Breast Cancer Prediction Empowered with Fine-Tuning

“…Rasti [ 26 ] presented a mixed-set convolutional neural network for benign cancerous differentiation. Mohiyuddin et al [ 27 ] used the YOLOv5 network to predict breast tumor with the help of a publicly available dataset of curated imaging subset of DDSM [ 28 ], and they used augmented techniques and split data 60% and 30% of training and validation, respectively, and achieved 96.50% prediction accuracy. Mehmood et al [ 29 ] used a random forest feature selection technique to predict cervical cancer with the help of a sallow neural network and achieved 93.6% prediction accuracy and 0.07111 mean squared error.…”

Research on Two-Way Detection of YOLO V5s+Deep Sort Road Vehicles Based on Attention Mechanism

“…Where Distance_C is the minimum diagonal distance of the circumscribed rectangle of the two kinds of Box mentioned above, and Distance_2 is the Euclidean distance of the center point of the two kinds of Box [9]…”