TY - JOUR
T1 - Mass segmentation for whole mammograms via attentive multi-task learning framework
AU - Hou, Xuan
AU - Bai, Yunpeng
AU - Xie, Yefan
AU - Li, Ying
N1 - Publisher Copyright:
© 2021 Institute of Physics and Engineering in Medicine.
PY - 2021/5/21
Y1 - 2021/5/21
N2 - Mass segmentation in the mammogram is a necessary and challenging task in the computer-aided diagnosis of breast cancer. Most of the existing methods tend to segment the mass by manually or automatically extracting mass-centered image patches. However, manual patch extraction is time-consuming, wheras automatic patch extraction can introduce errors that will affect the performance of subsequent segmentation. In order to improve the efficiency of mass segmentation and reduce segmentation errors, we proposed a novel mass segmentation method based on an attentive multi-task learning network (MTLNet), which is an end-to-end model to accurately segment mass in the whole mammogram directly, without the need for extraction in advance with the center of mass image patch. In MTLNet, we applied group convolution to the feature extraction network, which not only reduced the redundancy of the network but also improved the capacity of feature learning. Secondly, an attention mechanism is added to the backbone to highlight the feature channels that contain rich information. Eventually, the multi-task learning framework is employed in the model, which reduces the risk of model overfitting and enables the model not only to segment the mass but also to classify and locate the mass. We used five-fold cross validation to evaluate the performance of the proposed method under detection and segmentation tasks respectively on the two public mammographic datasets INbreast and CBIS-DDSM, and our method achieved a Dice index of 0.826 on INbreast and 0.863 on CBIS-DDSM.
AB - Mass segmentation in the mammogram is a necessary and challenging task in the computer-aided diagnosis of breast cancer. Most of the existing methods tend to segment the mass by manually or automatically extracting mass-centered image patches. However, manual patch extraction is time-consuming, wheras automatic patch extraction can introduce errors that will affect the performance of subsequent segmentation. In order to improve the efficiency of mass segmentation and reduce segmentation errors, we proposed a novel mass segmentation method based on an attentive multi-task learning network (MTLNet), which is an end-to-end model to accurately segment mass in the whole mammogram directly, without the need for extraction in advance with the center of mass image patch. In MTLNet, we applied group convolution to the feature extraction network, which not only reduced the redundancy of the network but also improved the capacity of feature learning. Secondly, an attention mechanism is added to the backbone to highlight the feature channels that contain rich information. Eventually, the multi-task learning framework is employed in the model, which reduces the risk of model overfitting and enables the model not only to segment the mass but also to classify and locate the mass. We used five-fold cross validation to evaluate the performance of the proposed method under detection and segmentation tasks respectively on the two public mammographic datasets INbreast and CBIS-DDSM, and our method achieved a Dice index of 0.826 on INbreast and 0.863 on CBIS-DDSM.
KW - attention mechanism
KW - group convolution
KW - mammogram segmentation
KW - multi-task learning
UR - http://www.scopus.com/inward/record.url?scp=85107049467&partnerID=8YFLogxK
U2 - 10.1088/1361-6560/abfa35
DO - 10.1088/1361-6560/abfa35
M3 - 文章
C2 - 33882475
AN - SCOPUS:85107049467
SN - 0031-9155
VL - 66
JO - Physics in Medicine and Biology
JF - Physics in Medicine and Biology
IS - 10
M1 - 105015
ER -