Multi-class semantic segmentation of breast tissues from MRI images using U-Net based on Haar wavelet pooling

Abstract

Abstract MRI images for breast cancer diagnosis are inappropriate for reconstructing the natural breast shape in a standing position because they are taken in a lying position. Some studies have proposed methods to present the breast shape in a standing position using ordinary differential equation of the finite element method. However, it is difficult to obtain meaningful results because breast tissues have different elastic moduli. This study proposed a multi-class semantic segmentation method for breast tissues to reconstruct breast shape using U-Net based on Haar wavelet pooling. First, a dataset was constructed by labeling the skin, fat, and fibro-glandular tissues and the background from MRI images taken in a lying position. Next, multi-class semantic segmentation was performed using U-Net based on Haar wavelet pooling to improve the segmentation accuracy for breast tissues. The U-Net based on Haar wavelet pooling effectively extracted breast tissue features while reducing information loss of the image in a subsampling stage using multiple sub-bands. In addition, the proposed network is robust to overfitting. The proposed network showed an mIOU of 87.48 for segmenting breast tissues. The proposed networks showed high-accuracy segmentation for breast tissue with different elastic moduli to reconstruct the natural breast shape.