Using a Wavelet-Based and Fine-Tuned Convolutional Neural Network for Classification of Breast Density in Mammographic Images
Open Access
- 1 January 2020
- journal article
- research article
- Published by Scientific Research Publishing, Inc. in Open Journal of Medical Imaging
- Vol. 10 (01), 17-29
- https://doi.org/10.4236/ojmi.2020.101002
Abstract
Classification of breast density is significantly important during the process of breast diagnosis. The purpose of this study was to develop a useful computer-ized tool to help radiologists determine the patient’s breast density category on the mammogram. In this article, we presented a model for automatically classi-fying breast densities by employing a wavelet transform-based and fine-tuned convolutional neural network (CNN). We modified a pre-trained AlexNet model by removing the last two fully connected (FC) layers and appending two newly created layers to the remaining structure. Unlike the common CNN-based methods that use original or pre-processed images as inputs, we adopted the use of redundant wavelet coefficients at level 1 as inputs to the CNN model. Our study mainly focused on discriminating between scattered density and heterogeneously dense which are the two most difficult density cat-egories to differentiate for radiologists. The proposed system achieved 88.3% overall accuracy. In order to demonstrate the effectiveness and usefulness of the proposed method, the results obtained from a conventional fine-tuning CNN model was compared with that from the proposed method. The results demon-strate that the proposed technique is very promising to help radiologists and serve as a second eye for them to classify breast density categories in breast cancer screening.Keywords
This publication has 29 references indexed in Scilit:
- Discrimination of Breast Cancer with Microcalcifications on Mammography by Deep LearningScientific Reports, 2016
- Representation learning for mammography mass lesion classification with convolutional neural networksComputer Methods and Programs in Biomedicine, 2016
- Deep-learning convolution neural network for computer-aided detection of microcalcifications in digital breast tomosynthesisPublished by SPIE-Intl Soc Optical Eng ,2016
- Unsupervised Deep Learning Applied to Breast Density Segmentation and Mammographic Risk ScoringIEEE Transactions on Medical Imaging, 2016
- Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer LearningIEEE Transactions on Medical Imaging, 2016
- Breast Density: Clinical Implications and Assessment MethodsRadioGraphics, 2015
- A Modified Undecimated Discrete Wavelet Transform Based Approach to Mammographic Image DenoisingJournal of Digital Imaging, 2012
- Computer-Aided Detection (CAD) in Screening Mammography: Sensitivity of Commercial CAD Systems for Detecting Architectural DistortionAmerican Journal of Roentgenology, 2003
- Breast Imaging Reporting and Data SystemAmerican Journal of Roentgenology, 2000
- Face recognition: a convolutional neural-network approachIEEE Transactions on Neural Networks, 1997