arXiv reaDer
異種 3D MR 膝画像の処理: 二重知識蒸留によるフェデレーテッド フューショット学習法
Dealing With Heterogeneous 3D MR Knee Images: A Federated Few-Shot Learning Method With Dual Knowledge Distillation
Federated Learning は、データを集約することなくクライアント (病院など) 間の共同トレーニングを可能にするため、医療機関の間で人気を博しています。ただし、特に大規模な 3D 画像データセットの場合、注釈の作成に関連するコストが高いため、臨床機関はローカルでトレーニングするための十分な教師付きデータを持っていません。したがって、共同モデルのパフォーマンスは、限られた監督下では標準以下です。一方、大規模な機関には、高解像度の画像とラベルを使用してデータ リポジトリを編集するためのリソースがあります。したがって、個々のクライアントは、パブリック データ リポジトリで取得した知識を利用して、プライベートな注釈付き画像の不足を軽減できます。この論文では、二重知識蒸留による連合少数ショット学習法を提案します。この方法により、プライバシーを危険にさらすことなく、クライアント間で限られた注釈を使用して共同トレーニングを行うことができます。提案された方法の教師あり学習は、各クライアントの限られたラベル付きデータから特徴を抽出しますが、教師なしデータは、共同モデルの精度をさらに向上させ、コミュニケーションを減らすために、国のデータリポジトリから特徴と応答ベースの知識の両方を抽出するために使用されます。料金。プライベートな臨床データセットからの 3D 磁気共鳴膝画像に対して広範な評価が行われます。私たちの提案した方法は、他の半教師あり連合学習方法よりも優れたパフォーマンスと少ないトレーニング時間を示しています。コードと追加の視覚化結果は、https://github.com/hexiaoxiao-cs/fedml-knee で入手できます。
Federated Learning has gained popularity among medical institutions since it enables collaborative training between clients (e.g., hospitals) without aggregating data. However, due to the high cost associated with creating annotations, especially for large 3D image datasets, clinical institutions do not have enough supervised data for training locally. Thus, the performance of the collaborative model is subpar under limited supervision. On the other hand, large institutions have the resources to compile data repositories with high-resolution images and labels. Therefore, individual clients can utilize the knowledge acquired in the public data repositories to mitigate the shortage of private annotated images. In this paper, we propose a federated few-shot learning method with dual knowledge distillation. This method allows joint training with limited annotations across clients without jeopardizing privacy. The supervised learning of the proposed method extracts features from limited labeled data in each client, while the unsupervised data is used to distill both feature and response-based knowledge from a national data repository to further improve the accuracy of the collaborative model and reduce the communication cost. Extensive evaluations are conducted on 3D magnetic resonance knee images from a private clinical dataset. Our proposed method shows superior performance and less training time than other semi-supervised federated learning methods. Codes and additional visualization results are available at https://github.com/hexiaoxiao-cs/fedml-knee.
updated: Sat Mar 25 2023 04:46:25 GMT+0000 (UTC)
published: Sat Mar 25 2023 04:46:25 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト