Discrete Wavelet Transform-Based CNN for Breast Cancer Classification from Histopathology Images.

Yuan Qi,Xiaozhang Liu,Hua Li,Mingting Liu,Wei Li
DOI: https://doi.org/10.1007/978-3-031-20096-0_25
2022-01-01
Abstract:Breast cancer has the highest mortality rate among cancers in women, but its diagnosis is limited. In recent years, there have been many works applying deep learning to computer-aided diagnosis of breast cancer histopathology images. However, the existing methods focus only on the spatial domain approach and the extracted feature information is not comprehensive enough. Furthermore, to reduce the computational burden, they usually crop images into smaller-sized patches resulting in the loss of a large amount of feature information. To address the above issues, we propose a frequency domain learning method based on discrete wavelet transform and convolutional neural network (CNN) for the classification task of breast cancer histopathology images. The method first extracts the feature information in the time and frequency domains using the wavelet transform, while taking into account the different information contained in the three color components of RGB images. Then, the low-frequency information from the decomposition of different color components is then fused and inputs to the CNN for training, meanwhile, other non-critical information is filtered out to improve the computational efficiency. The accuracy of our method is 99.50%, 99.20%, 99.00% and 99.08% for the binary classification task and 95.79%, 92.91%, 90.83% and 93.37% for the multi-classification task, respectively, on the BreakHis dataset.
What problem does this paper attempt to address?