Bose, Anu, Nguyen, Tuan ORCID: 0000-0003-0055-8218, Du, Hongbo and AlZoubi, Alaa (2021) Faster RCNN hyperparameter selection for breast lesion detection in 2D ultrasound images. In: Advances in Computational Intelligence Systems Contributions Presented at the 20th UK Workshop on Computational Intelligence, September 8-10, 2021, Aberystwyth, Wales, UK. Advances in Intelligent Systems and Computing (AISC) (1409). Springer, Cham, pp. 179-190. ISBN 978-3030870942 ISSN 2194-5365 (doi:https://doi.org/10.1007/978-3-030-87094-2_16)

PDF (AAM) 36019_NGUYEN_Faster_RCNN_hyperparameter.pdf - Accepted Version Restricted to Repository staff only Download (288kB) | Request a copy

Abstract

Breast cancer is one of the leading forms of cancer. Breast lesion detection is a prerequisite for description of its characteristics and ultimately correct diagnosis of the lesion type. Compared to object detection in natural images, lesion detection in ultrasound images is a challenging task due to the inherent nature of the ultrasound images. This paper is concerned with adopting Faster Regions with Convolutional Neural Network (Faster RCNN) method for detecting breast lesions in 2D ultrasound images. Faster RCNN shows great promise in this application with very few misses of the actual lesion leading to high recall. However, the overall performance of this method suffers from high number of false positives. Therefore, we investigate different modelling hyperparameters of Faster RCNN to find an optimal configuration to improve the overall performance by essentially reducing false positives without significant increase in the number of misdetected lesions. Our empirical study using a total of 1183 images in three datasets shows that the optimal detection model outperformed original Faster RCNN due to significant reductions in false positives, resulting in 15% to 28% higher precision with only 3% to 11% drop in recall. Our optimal model also outperformed an existing breast lesion detection method by 3.4% in F1-score and provides better balance in precision and recall. Our study demonstrates that the optimal hyperparameter selection for Faster RCNN is a promising direction for breast lesion detection in ultrasound images.

Item Type:	Conference Proceedings
Title of Proceedings:	Advances in Computational Intelligence Systems Contributions Presented at the 20th UK Workshop on Computational Intelligence, September 8-10, 2021, Aberystwyth, Wales, UK
Uncontrolled Keywords:	faster RCNN; breast lesion detection; ultrasound image; deep learning; medical image analysis
Subjects:	R Medicine > RC Internal medicine > RC0254 Neoplasms. Tumors. Oncology (including Cancer) R Medicine > RG Gynecology and obstetrics T Technology > TJ Mechanical engineering and machinery
Faculty / School / Research Centre / Research Group:	Faculty of Engineering & Science Faculty of Engineering & Science > School of Computing & Mathematical Sciences (CMS)
Last Modified:	16 May 2022 11:55
URI:	http://gala.gre.ac.uk/id/eprint/36019

Actions (login required)

View Item

Downloads