arXiv reaDer
多臓器セグメンテーションのための不確実性を意識したインクリメンタル学習
Uncertainty-aware Incremental Learning for Multi-organ Segmentation
複数の単一臓器データセットから統合された多臓器セグメンテーションモデルをトレーニングするためのほとんどの既存のアプローチでは、トレーニング中に複数のデータセットに同時にアクセスする必要があります。実際のシナリオでは、プライバシーと倫理上の懸念から、関心のある臓器のトレーニングデータは公開されていない可能性があります。この目的のために、データフリーの増分臓器セグメンテーションシナリオを調査し、それを解決するための新しい増分トレーニングフレームワークを提案します。プライバシー保護のために、独自のトレーニングデータの代わりに事前トレーニング済みモデルを使用します。具体的には、事前にトレーニングされたK臓器セグメンテーションモデルと新しい単一臓器データセットを前提として、前のトレーニング段階に属するデータにアクセスせずに、統合されたK +1臓器セグメンテーションモデルをトレーニングします。私たちのアプローチは、バックグラウンドラベルアライメント戦略と不確実性を意識したガイダンス戦略の2つの部分で構成されています。最初の部分は、保持されたモデルからトレーニングモデルへの知識の伝達に使用されます。 2番目の部分は、事前にトレーニングされたモデルから不確実性情報を抽出して、知識伝達プロセス全体をガイドするために使用されます。これら2つの戦略を組み合わせることにより、元のトレーニングデータなしで、事前トレーニング済みモデルからより信頼性の高い情報が抽出されます。複数の公開されている事前トレーニング済みモデルと多臓器データセットMOBAでの実験により、フレームワークの有効性が実証されました。
Most existing approaches to train a unified multi-organ segmentation model from several single-organ datasets require simultaneously access multiple datasets during training. In the real scenarios, due to privacy and ethics concerns, the training data of the organs of interest may not be publicly available. To this end, we investigate a data-free incremental organ segmentation scenario and propose a novel incremental training framework to solve it. We use the pretrained model instead of its own training data for privacy protection. Specifically, given a pretrained K organ segmentation model and a new single-organ dataset, we train a unified K+1 organ segmentation model without accessing any data belonging to the previous training stages. Our approach consists of two parts: the background label alignment strategy and the uncertainty-aware guidance strategy. The first part is used for knowledge transfer from the pretained model to the training model. The second part is used to extract the uncertainty information from the pretrained model to guide the whole knowledge transfer process. By combing these two strategies, more reliable information is extracted from the pretrained model without original training data. Experiments on multiple publicly available pretrained models and a multi-organ dataset MOBA have demonstrated the effectiveness of our framework.
updated: Tue Mar 09 2021 05:12:39 GMT+0000 (UTC)
published: Tue Mar 09 2021 05:12:39 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アソシエイト