Federated learning-based vertebral body segmentation. (November 2022)

• Record Type: Journal Article
• Authors: Liu, Junxiu; Liang, Xiuhao; Yang, Rixing; Luo, Yuling; Lu, Hao; Li, Liangjia; Zhang, Shunsheng; Yang, Su
• Is Part Of: Engineering applications of artificial intelligence. Volume 116(2022)
• Publication Date: 2022-11
• Subjects: Federated learning -- Vertebral body segmentation -- MRI images; Engineering -- Data processing -- Periodicals; Artificial intelligence -- Periodicals; Expert systems (Computer science) -- Periodicals; Ingénierie -- Informatique -- Périodiques; Intelligence artificielle -- Périodiques; Systèmes experts (Informatique) -- Périodiques; Artificial intelligence; Engineering -- Data processing; Expert systems (Computer science); Periodicals; 620.00285
• DOI: 10.1016/j.engappai.2022.105451
• Languages: English
• ISSNs: 0952-1976

Abstract

Abstract: To ensure the safety of spinal surgery, sufficiently labeled Magnetic Resonance Imaging (MRI) images are essential for training an accurate vertebral segmentation model, but the number of labeled MRI images owned by the independent medical institution such as the hospital is generally limited. Besides, in consideration of patients' privacy, annotated images are difficult to share directly as the medical data to train vertebral body segment models. To address these challenges, a Federated Learning-based Vertebral Body Segment Framework (FLVBSF) is proposed in this work, which includes a novel local Dual Attention Gates (DAGs)-based attention mechanism and a global federated learning framework. The model sensitivity to vertebral body pixels and segmentation accuracy can be improved by using the DAGs. The performance of vertebral body segmentation models is boosted by the global federated learning framework via collaboratively exploiting the labeled spine image data from different institutions. The centralized training-based experimental results show that 98.29% in pixel-level accuracy is achieved by the U-Net with DAGs, 88.04% in dice similarity coefficient, 88.25% in sensitivity, 99.16% in specificity, and 79.09% in Jaccard similarity coefficient and the mean segmentation time per case is 0.14 s. Meanwhile, the federated learning-based experimental results show that the proposed FLVBSF can enhance the performance of the vertebral segmentation model by a statistically significant margin.