arXiv reaDer
継続的セグメント: CT スキャンにおける 143 の全身臓器の単一の統合されたアクセス可能な継続的セグメンテーション モデルに向けて
Continual Segment: Towards a Single, Unified and Accessible Continual Segmentation Model of 143 Whole-body Organs in CT Scans
ディープ ラーニングは、主流の医用画像セグメンテーション手法を強化します。それにもかかわらず、現在のディープ セグメンテーション アプローチでは、新しい増分セグメンテーション クラスを (新しいトレーニング データセットを伴うかどうかにかかわらず) 追加する必要がある場合に、トレーニング済みモデルを効率的かつ効果的に適応および更新することができません。実際の臨床環境では、患者のプライバシーとデータストレージの障害により、以前のトレーニングデータセットに (再) アクセスすることなく、セグメンテーションモデルを動的に拡張して新しい臓器/腫瘍をセグメント化できることが望ましい場合があります。このプロセスは、継続的なセマンティック セグメンテーション (CSS) の問題と見なすことができ、多臓器セグメンテーションについては十分に研究されていません。この作業では、合計 143 の全身器官をセグメント化するための単一のディープ セグメンテーション モデルを学習するための新しいアーキテクチャ CSS 学習フレームワークを提案します。エンコーダー/デコーダーのネットワーク構造を使用して、継続的にトレーニングされてから凍結されたエンコーダーを段階的に追加されたデコーダーと組み合わせて、新しいクラスを後で有効にセグメント化するのに十分な代表的な画像特徴を抽出および保存できることを示します。単一のネットワーク モデルの複雑さを維持するために、ニューラル アーキテクチャ検索と教師と生徒に基づく知識の蒸留を使用して、各デコーダーを段階的にトリミングします。異なるデータセットに現れる健康な器官と病理学的な器官の両方を組み込むために、異なるデコーダーに由来する重複した器官予測をマージする、新しい異常認識および信頼学習モジュールが提案されています。 4 つのデータセットからの 2500 人以上の患者の 3D CT スキャンでトレーニングおよび検証された単一のネットワークは、合計 143 の全身臓器を非常に高い精度でセグメント化でき、4 つの個別のセグメンテーション モデル (つまり、1 つのモデルにつきデータセット/タスク)。
Deep learning empowers the mainstream medical image segmentation methods. Nevertheless current deep segmentation approaches are not capable of efficiently and effectively adapting and updating the trained models when new incremental segmentation classes (along with new training datasets or not) are required to be added. In real clinical environment, it can be preferred that segmentation models could be dynamically extended to segment new organs/tumors without the (re-)access to previous training datasets due to obstacles of patient privacy and data storage. This process can be viewed as a continual semantic segmentation (CSS) problem, being understudied for multi-organ segmentation. In this work, we propose a new architectural CSS learning framework to learn a single deep segmentation model for segmenting a total of 143 whole-body organs. Using the encoder/decoder network structure, we demonstrate that a continually-trained then frozen encoder coupled with incrementally-added decoders can extract and preserve sufficiently representative image features for new classes to be subsequently and validly segmented. To maintain a single network model complexity, we trim each decoder progressively using neural architecture search and teacher-student based knowledge distillation. To incorporate with both healthy and pathological organs appearing in different datasets, a novel anomaly-aware and confidence learning module is proposed to merge the overlapped organ predictions, originated from different decoders. Trained and validated on 3D CT scans of 2500+ patients from four datasets, our single network can segment total 143 whole-body organs with very high accuracy, closely reaching the upper bound performance level by training four separate segmentation models (i.e., one model per dataset/task).
updated: Sun Sep 03 2023 20:25:27 GMT+0000 (UTC)
published: Wed Feb 01 2023 00:49:21 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アソシエイト