A Fully Decentralized Federated Learning Framework for Automated Image Segmentation in Cancer Radiotherapy

用于癌症放射治疗自动图像分割的完全去中心化联合学习框架

• 批准号: 10303437
• 负责人: Yading Yuan
• 金额: $ 21.06万
• 依托单位: ICAHN SCHOOL OF MEDICINE AT MOUNT SINAI
• 依托单位国家: 美国
• 财政年份: 2021
• 资助国家: 美国
• 起止时间: 2021-09-01 至 2023-03-31
• 项目状态: 已结题
• 来源: https://reporter.nih.gov/project-details/10303437
• 关键词: Address; Adoption; Anatomy; Appearance; Area; Artificial Intelligence; Client; Clinical; Collaborations; Communities; Data; Data Aggregation; Data Pooling; Data Set; Data Sources; Databases; Decentralization; Development; Environment; Ethics; Goals; Health system; Healthcare Systems; Hospitals; Image; Image Analysis; Institution; Learning; Legal; Liver; Location; Machine Learning; Medical; Medical Imaging; Medical center; Medicine; Methods; Modeling; National Institute of Biomedical Imaging and Bioengineering; Organ; Ownership; Participant; Patients; Performance; Privacy; Process; Radiation Oncology; Radiation therapy; Research Priority; Resources; Risk; Scheme; Shapes; Software Tools; Source; Techniques; Time; Training; Validation; X-Ray Computed Tomography; automated segmentation; base; cancer radiation therapy; clinical practice; data sharing; deep learning; deep learning algorithm; flexibility; head and neck cancer patient; imaging Segmentation; improved; interoperability; liver imaging; metropolitan; novel; open source; patient privacy; peer; precision medicine; soft tissue; software infrastructure; success; task analysis; treatment planning; tumor

PROJECT SUMMARY While the recent surge of artificial intelligence (AI) has made remarkable progress in various image analysis tasks, their performance in a broad range of clinical environment is largely restricted by the limited generalization capability when being applied to new data, primarily because most models have been generated using data from a single institution or public datasets with limited training data. Aggregating data from different institutions could improve model training, but such centralized data sharing is practically challenging due to various technical, legal, privacy and data ownership barriers. This proposal aims to address these barriers by developing a novel gossip federated learning (GFL) framework to build an effective AI model by learning from different data sources without the need of sharing patient data. As compared to the traditional client/server federated learning such as FedAvg, the proposed framework is fully decentralized in that the models trained in local datasets will directly communicate to each other in a peer-to-peer manner, making our method more robust and efficient. We will develop and evaluate the proposed scheme in the task of automated organ segmentation in CT images for liver and head and neck (H&N) cancer patients treated with radiation therapy (RT) because accurate, robust and efficient delineation of those organs at risk (OARs) is a clinically important but technically challenging problem. We hypothesize that the model trained with our framework can achieve segmentation performance not inferior to a model with data pooled from all the resources. The dynamics of our recently created healthcare system mimic a diverse multi-institutional environment, which places us in an ideal setting to systematically evaluate our framework. Our specific aims include: 1) Establish the GFL-based automated OAR segmentation framework, and develop the supporting software infrastructure; 2) Optimize the GFL-based auto- segmentation; 3) Evaluate GFL-based OAR segmentation framework with 400 liver and 400 H&N cancer patients collected from four hospitals within a metropolitan health system. This proposal addresses two key research priorities for NIBIB: machine learning based segmentation and approaches that facilitate interoperability among annotations used in image training databases. The success of this project will substantially increase the number and variety of data for model training without sacrificing the patient privacy, and thus improve the performance and generalization of the segmentation model on new data. We will open-source this framework, which may enable a larger scale of multi-institutional collaboration and could expedite the clinical adoption of AI-driven auto- segmentation in RT. More importantly, this framework provides a flexible and robust solution to the primary barrier of applying AI to the medical domain where learning on multi-institutional data sharing is impeded by patient privacy concerns, and is expected to have a catalytic impact on precision medicine by generalizing it to broader applications within medicine where a model needs to learn across multi-institutional data without sacrificing patient privacy.

虽然最近人工智能（AI）的激增在……方面取得了显著进展