Abstract
Breast tumor segmentation is an important task in medical image analysis, especially in the diagnosis and treatment of breast cancer. Magnetic resonance imaging (MRI) is a widely used imaging modality for breast tumor segmentation due to its high spatial and temporal resolution. However, there are several challenges associated with breast tumor segmentation on MRI data. One of the main challenges is the heterogeneity of MRI data collected from different imaging centers, which is usually caused by inconsistencies in the imaging protocols, quality, and resolution. Additionally, there are privacy concerns associated with the use of patient data, which limit the sharing of data across institutions. To address these challenges, we propose a novel federated learning framework called Federated Channel Spatial attention adaptive weight Aggregation (Fed-CSA) for breast tumor segmentation on MRI data. The proposed framework uses a CSA module to handle the heterogeneity of MRI data collected from different centers, which can pay more attention to the target regions and ignore non-related regions. Furthermore, the adaptive weight aggregation (AWA) method is proposed to improve the efficiency of model weight aggregation in federated learning across multiple clients while preserving patients’ privacy. The proposed Fed-CSA framework is trained and tested on a large-scale dataset of 1460 patients from different centers, and the results demonstrate its effectiveness in terms of efficiency and feasibility, outperforming other baseline algorithms.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Sung, H., et al.: Global cancer statistics 2020: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA Cancer J. Clin. 71(3), 209–249 (2021). https://doi.org/10.3322/caac.21660
Dai, Q., Zheng, J., Zhang, M.: Current status and future of breast cancer imaging. Mod. Pract. Med. 30, 561–564 (2018)
Benjelloun, M., El Adoui, M., Larhmam, M.A., Mahmoudi, S.A.: Automated breast tumor segmentation in DCE-MRI using deep learning. In: 2018 4th International Conference on Cloud Computing Technologies and Applications (Cloudtech), pp. 1–6. IEEE (2018)
Baccouche, A., Garcia-Zapirain, B., Castillo Olea, C., Elmaghraby, A.S.: Connected-UNets: a deep learning architecture for breast mass segmentation. NPJ Breast Cancer 7, 151 (2021)
Hai, J., et al.: Fully convolutional DenseNet with multiscale context for automated breast tumor segmentation. J. Healthc. Eng. 2019, 1–11 (2019)
Jiang, Y., Edwards, A.V., Newstead, G.M.: Artificial intelligence applied to breast MRI for improved diagnosis. Radiology 298, 38–46 (2021)
McMahan, B., Moore, E., Ramage, D., Hampson, S., Arcas, B.A.Y.: Communication-efficient learning of deep networks from decentralized data. In: Aarti, S., Jerry, Z. (eds.) Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, vol. 54, pp. 1273–1282. PMLR, Proceedings of Machine Learning Research (2017)
Yang, D., et al.: Federated semi-supervised learning for COVID region segmentation in chest CT using multi-national data from China, Italy, Japan. Med. Image Anal. 70, 101992 (2021)
Xu, J., Glicksberg, B.S., Su, C., Walker, P., Bian, J., Wang, F.: Federated learning for healthcare informatics. J. Healthc. Inform. Res. 5, 1–19 (2020)
Adams, R., Bischof, L.: Seeded region growing. IEEE Trans. Pattern Anal. Mach. Intell. 16, 641–647 (1994)
Manjunath, B.S., Chellappa, R.: Unsupervised texture segmentation using Markov random field models. IEEE Trans. Pattern Anal. Mach. Intell. 13, 478–482 (1991)
Kass, M., Witkin, A., Terzopoulos, D.: Snakes: active contour models. Int. J. Comput. Vision 1, 321–331 (1988)
Maitra, I.K., Nag, S., Bandyopadhyay, S.K.: Automated digital mammogram segmentation for detection of abnormal masses using binary homogeneity enhancement algorithm. J. Comput. Sci. Eng. (IJCSE) 2, 416–427 (2011)
Dinsha, D., Manikandaprabu, N.: Breast tumor segmentation and classification using SVM and Bayesian from thermogram images. Unique J. Eng. Adv. Sci. 2, 147–151 (2014)
Long, G., Tan, Y., Jiang, J., Zhang, C.: Federated learning for open banking. In: Yang, Q., Fan, L., Yu, H. (eds.) Federated Learning. LNCS (LNAI), vol. 12500, pp. 240–254. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-63076-8_17
Chang, Q., et al.: Synthetic learning: learn from distributed asynchronized discriminator GAN without sharing medical image data. In: 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 13853–13863 (2020)
Sheller, M.J., Reina, G.A., Edwards, B., Martin, J., Bakas, S.: Multi-institutional deep learning modeling without sharing patient data: a feasibility study on brain tumor segmentation. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11383, pp. 92–104. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11723-8_9
Isensee, F., Jaeger, P.F., Kohl, S.A.A., Petersen, J., Maier-Hein, K.H.: nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation. Nat. Methods 18, 203–211 (2021)
Lin, T.-Y., Goyal, P., Girshick, R., He, K., Dollár, P.: Focal loss for dense object detection. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2980–2988 (2017)
Salehi, S.S.M., Erdogmus, D., Gholipour, A.: Tversky loss function for image segmentation using 3D fully convolutional deep networks. In: Wang, Q., Shi, Y., Suk, H.-I., Suzuki, K. (eds.) MLMI 2017. LNCS, vol. 10541, pp. 379–387. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-67389-9_44
Author information
Authors and Affiliations
Corresponding authors
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Dong, X. et al. (2023). Fed-CSA: Channel Spatial Attention and Adaptive Weights Aggregation-Based Federated Learning for Breast Tumor Segmentation on MRI. In: Huang, DS., Premaratne, P., Jin, B., Qu, B., Jo, KH., Hussain, A. (eds) Advanced Intelligent Computing Technology and Applications. ICIC 2023. Lecture Notes in Computer Science, vol 14088. Springer, Singapore. https://doi.org/10.1007/978-981-99-4749-2_27
Download citation
DOI: https://doi.org/10.1007/978-981-99-4749-2_27
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-4748-5
Online ISBN: 978-981-99-4749-2
eBook Packages: Computer ScienceComputer Science (R0)