arXiv reaDer
クライアント間関係マッチングによる連合半教師あり医用画像分類
Federated Semi-supervised Medical Image Classification via Inter-client Relation Matching
連合学習(FL)は、深いネットワークをトレーニングするために分散型医療機関と協力する人気が高まるにつれて出現しました。ただし、既存のFLアルゴリズムでは教師ありトレーニングの設定しか許可されていませんが、現実的なほとんどの病院では、予算や専門知識がないため、通常、複雑なデータのラベル付けを行う余裕がありません。このホワイトペーパーでは、ラベル付きクライアントとラベルなしクライアント(病院など)の両方からのデータを共同で利用してフェデレーションモデルを学習することを目的とした、Federated Semi-supervised Learning(FSSL)という実用的でありながら挑戦的なFL問題について研究します。この問題に対する新しいアプローチを提示します。これは、新しいクライアント間関係マッチングスキームを使用して、従来の整合性正則化メカニズムを改善します。提案された学習スキームは、抽出された疾患の関係を調整することにより、ラベル付きクライアントとラベルなしクライアントの間で学習を明示的に結び付け、それによってラベルなしクライアントでのタスク知識の不足を軽減し、ラベルなしサンプルからの識別情報を促進します。 2つの大規模な医用画像分類データセットでメソッドを検証します。私たちの方法の有効性は、最先端技術に対する明確な改善と、両方のタスクの徹底的なアブレーション分析によって実証されています。コードはhttps://github.com/liuquande/FedIRMで利用可能になります。
Federated learning (FL) has emerged with increasing popularity to collaborate distributed medical institutions for training deep networks. However, despite existing FL algorithms only allow the supervised training setting, most hospitals in realistic usually cannot afford the intricate data labeling due to absence of budget or expertise. This paper studies a practical yet challenging FL problem, named Federated Semi-supervised Learning (FSSL), which aims to learn a federated model by jointly utilizing the data from both labeled and unlabeled clients (i.e., hospitals). We present a novel approach for this problem, which improves over traditional consistency regularization mechanism with a new inter-client relation matching scheme. The proposed learning scheme explicitly connects the learning across labeled and unlabeled clients by aligning their extracted disease relationships, thereby mitigating the deficiency of task knowledge at unlabeled clients and promoting discriminative information from unlabeled samples. We validate our method on two large-scale medical image classification datasets. The effectiveness of our method has been demonstrated with the clear improvements over state-of-the-arts as well as the thorough ablation analysis on both tasksCode will be made available at https://github.com/liuquande/FedIRM.
updated: Wed Jun 16 2021 07:58:00 GMT+0000 (UTC)
published: Wed Jun 16 2021 07:58:00 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アソシエイト