A Practical Computer Aided Diagnosis System for Breast Ultrasound Classifying Lesions into the ACR BI-RADS Assessment

Hsin-Ya Su,Chung-Yueh Lien,Pai-Jung Huang,Woei-Chyn Chu
DOI: https://doi.org/10.1007/s40846-024-00869-5
IF: 2
2024-06-02
Journal of Medical and Biological Engineering
Abstract:In this paper, we propose an open-source deep learning-based computer-aided diagnosis system for breast ultrasound images based on the Breast Imaging Reporting and Data System (BI-RADS).
engineering, biomedical
What problem does this paper attempt to address?
The main objective of this paper is to develop a deep learning-based computer-aided diagnosis (CADx) system for the classification of lesions in breast ultrasound images. Specifically, the system aims to categorize lesions in breast ultrasound into three categories of the American College of Radiology (ACR) Breast Imaging Reporting and Data System (BI-RADS) assessment: Category 2 (requiring routine follow-up), Category 3 (requiring short-term follow-up), and Category 4/5 (suggesting biopsy). To achieve this goal, the researchers employed the following methods: 1. **Dataset**: The study used a dataset containing 569 cases with a total of 8,026 regions of interest (ROI) images, annotated by professional radiologists and physicians trained in breast surgery. 2. **Preprocessing**: The images were subjected to histogram equalization and their pixel values were rescaled to between 0-1 for normalization. Additionally, the dataset was augmented through horizontal flipping and shifting. 3. **Model Selection and Training**: Three different deep learning architectures were used—VGG-16, ResNet-50, and DenseNet-121, all initialized with ImageNet pre-trained weights. During training, the stochastic gradient descent optimizer was employed, with hyperparameters such as learning rate, momentum, weight decay, and dropout rate set. 4. **Ensemble Methods**: To improve performance, the researchers tested two ensemble methods—average prediction and voting—to combine the predictions of different models. 5. **System Implementation**: A comprehensive CADx system was developed, capable of not only automatic diagnosis but also seamless integration with Picture Archiving and Communication Systems (PACS), supporting standardized report generation and other functionalities. Experimental results showed that the ensemble model performed best in classification, achieving an accuracy of 81.8%. Additionally, the study highlighted the challenges in classifying BI-RADS Category 3 (moderate risk) and improved the classification efficiency for this category through ensemble methods. Overall, this work provides a practical deep learning-based solution for breast ultrasound image analysis.