arXiv reaDer
部分ラベルを使用した分散非 iid データセットでの医用画像分類のためのフェデレーテッド ラーニングの最適化
Optimizing Federated Learning for Medical Image Classification on Distributed Non-iid Datasets with Partial Labels
多数の大規模な胸部 X 線データセットは、ディープ ラーニングを使用した専門家レベルの異常検出の先頭に立っています。ただし、これらのデータセットは、存在する可能性のある疾患ラベルのサブセットを検出することに焦点を当てているため、それらは分散され、部分的なラベルで非 iid になります。最近の文献では、部分ラベルを持つ iid 以外のデータに関連するドメイン シフトが原因で、連合学習の収束に対するバッチ正規化レイヤーの影響が示されています。そのために、FedFBN を提案します。これは、モデル バックエンドとして事前トレーニング済みのネットワークを使用し、トレーニング プロセス全体でバッチ正規化レイヤーをフリーズすることにより、転移学習からインスピレーションを得たフェデレーテッド ラーニング フレームワークです。部分的および完全なラベルを持つシナリオ全体で、合成 iid おもちゃデータセットと大規模な非 iid データセットを使用して、現在の FL 戦略で FedFBN を評価します。私たちの結果は、FedFBN が、部分ラベルを持つ分散データと非 iid データを使用してグローバル モデルをトレーニングするための現在の集計戦略よりも優れていることを示しています。
Numerous large-scale chest x-ray datasets have spearheaded expert-level detection of abnormalities using deep learning. However, these datasets focus on detecting a subset of disease labels that could be present, thus making them distributed and non-iid with partial labels. Recent literature has indicated the impact of batch normalization layers on the convergence of federated learning due to domain shift associated with non-iid data with partial labels. To that end, we propose FedFBN, a federated learning framework that draws inspiration from transfer learning by using pretrained networks as the model backend and freezing the batch normalization layers throughout the training process. We evaluate FedFBN with current FL strategies using synthetic iid toy datasets and large-scale non-iid datasets across scenarios with partial and complete labels. Our results demonstrate that FedFBN outperforms current aggregation strategies for training global models using distributed and non-iid data with partial labels.
updated: Fri Mar 10 2023 19:23:33 GMT+0000 (UTC)
published: Fri Mar 10 2023 19:23:33 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アソシエイト