連合学習(FL)を使用すると、参加サイトのデータを共有することなく、深層学習モデルの共同学習が可能になります。医療画像分析タスクにおけるFLは比較的新しく、拡張の余地があります。この研究では、一般化可能なモデルをトレーニングするための新しい連合学習アプローチであるFedDropoutAvgを提案します。提案された方法は、クライアントの選択とフェデレーション平均化プロセスの両方でランダム性を利用します。 FedDropoutAvgを、実際のマルチサイト組織病理学画像分類タスクのFLシナリオのいくつかのアルゴリズムと比較します。 FedDropoutAvgを使用すると、最終モデルが他のFLアプローチよりも優れたパフォーマンスを達成でき、集中トレーニングのためにすべてのデータを共有する必要がある従来の深層学習モデルに近いことを示します。 21の異なるセンターからの120万の画像タイルで構成される大規模なデータセットでトレーニング済みモデルをテストします。提案されたアプローチの一般化能力を評価するために、FLでデータが使用されたセンターからの差し出されたテストセットと、フェデレーショントレーニングでデータが使用されなかった他の独立したセンターからの見えないデータに使用します。提案されたアプローチは、他の最先端のフェデレーショントレーニングアプローチよりも一般化可能であることを示します。私たちの知る限りでは、私たちの研究は、医療画像分析タスクのフェデレーション設定でランダム化されたクライアントとローカルモデルパラメータの選択手順を使用した最初の研究です。
Federated learning (FL) enables collaborative learning of a deep learning model without sharing the data of participating sites. FL in medical image analysis tasks is relatively new and open for enhancements. In this study, we propose FedDropoutAvg, a new federated learning approach for training a generalizable model. The proposed method takes advantage of randomness, both in client selection and also in federated averaging process. We compare FedDropoutAvg to several algorithms in an FL scenario for real-world multi-site histopathology image classification task. We show that with FedDropoutAvg, the final model can achieve performance better than other FL approaches and closer to a classical deep learning model that requires all data to be shared for centralized training. We test the trained models on a large dataset consisting of 1.2 million image tiles from 21 different centers. To evaluate the generalization ability of the proposed approach, we use held-out test sets from centers whose data was used in the FL and for unseen data from other independent centers whose data was not used in the federated training. We show that the proposed approach is more generalizable than other state-of-the-art federated training approaches. To the best of our knowledge, ours is the first study to use a randomized client and local model parameter selection procedure in a federated setting for a medical image analysis task.