arXiv reaDer
SegViz: 部分的な注釈を使用した異種データ セットでの多臓器セグメンテーションのためのフェデレーテッド ラーニング ベースのフレームワーク
SegViz: A federated-learning based framework for multi-organ segmentation on heterogeneous data sets with partial annotations
セグメンテーションは、複数のダウンストリーム臨床アプリケーションがあるため、医用画像の深層学習における最も主要なタスクの 1 つです。ただし、医療画像の注釈を手動で生成するには時間がかかり、高度なスキルが必要であり、特に 3D 画像の場合は費用がかかります。考えられる解決策の 1 つは、複数のグループの部分的に注釈が付けられたデータセットから知識を集約し、Federated Learning を使用してグローバル モデルを共同でトレーニングすることです。この目的のために、部分的な注釈を使用して分散された非 iid データセットからセグメンテーション モデルをトレーニングするための連合学習ベースのフレームワークである SegViz を提案します。 SegViz のパフォーマンスは、個々のモデルを各データセットで個別にトレーニングする場合と、すべてのデータセットを 1 か所に集約して単一のモデルをトレーニングする場合と比較されました。集計戦略として FedBN を使用する SegViz フレームワークは、外部 BTCV セットで優れたパフォーマンスを示し、肝臓、脾臓、膵臓、および腎臓のセグメンテーションでそれぞれ 0.93、0.83、0.55、および 0.75 のサイコロ スコアを示し、有意に (p<0.05) 優れていました。 (脾臓を除く) ベースライン モデルの 0.87、0.83、0.42、および 0.48 のサイコロ スコアよりも。対照的に、中央集約モデルは、0.65、0、0.55、および 0.68 のダイス スコアを持つテスト データセットで大幅に (p<0.05) パフォーマンスが低下しました。私たちの結果は、部分的なラベルを持つ分散データセットからマルチタスク モデルをトレーニングする SegViz フレームワークの可能性を示しています。すべての実装はオープンソースであり、https://anonymous.4open.science/r/SegViz-B746 で入手できます。
Segmentation is one of the most primary tasks in deep learning for medical imaging, owing to its multiple downstream clinical applications. However, generating manual annotations for medical images is time-consuming, requires high skill, and is an expensive effort, especially for 3D images. One potential solution is to aggregate knowledge from partially annotated datasets from multiple groups to collaboratively train global models using Federated Learning. To this end, we propose SegViz, a federated learning-based framework to train a segmentation model from distributed non-i.i.d datasets with partial annotations. The performance of SegViz was compared against training individual models separately on each dataset as well as centrally aggregating all the datasets in one place and training a single model. The SegViz framework using FedBN as the aggregation strategy demonstrated excellent performance on the external BTCV set with dice scores of 0.93, 0.83, 0.55, and 0.75 for segmentation of liver, spleen, pancreas, and kidneys, respectively, significantly (p<0.05) better (except spleen) than the dice scores of 0.87, 0.83, 0.42, and 0.48 for the baseline models. In contrast, the central aggregation model significantly (p<0.05) performed poorly on the test dataset with dice scores of 0.65, 0, 0.55, and 0.68. Our results demonstrate the potential of the SegViz framework to train multi-task models from distributed datasets with partial labels. All our implementations are open-source and available at https://anonymous.4open.science/r/SegViz-B746
updated: Sat Mar 11 2023 02:51:24 GMT+0000 (UTC)
published: Tue Jan 17 2023 18:36:57 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アソシエイト