计算机科学
适应(眼睛)
人工智能
机器学习
方案(数学)
分割
人工神经网络
建筑
火车
联合学习
个性化
人口
数据挖掘
作者
Holger R. Roth,Dong Yang,Wenqi Li,Andriy Myronenko,Wentao Zhu,Ziyue Xu,Xiaosong Wang,Daguang Xu
标识
DOI:10.1007/978-3-030-87199-4_34
摘要
Building robust deep learning-based models requires diverse training data, ideally from several sources. However, these datasets cannot be combined easily because of patient privacy concerns or regulatory hurdles, especially if medical data is involved. Federated learning (FL) is a way to train machine learning models without the need for centralized datasets. Each FL client trains on their local data while only sharing model parameters with a global server that aggregates the parameters from all clients. At the same time, each client's data can exhibit differences and inconsistencies due to the local variation in the patient population, imaging equipment, and acquisition protocols. Hence, the federated learned models should be able to adapt to the local particularities of a client's data. In this work, we combine FL with an AutoML technique based on local neural architecture search by training a "supernet". Furthermore, we propose an adaptation scheme to allow for personalized model architectures at each FL client's site. The proposed method is evaluated on four different datasets from 3D prostate MRI and shown to improve the local models' performance after adaptation through selecting an optimal path through the AutoML supernet.
科研通智能强力驱动
Strongly Powered by AbleSci AI