U 字型ネットワークとそのバリアントは、医療画像のセグメンテーションにおいて優れた結果を示しています。この論文では、3D 医用画像セグメンテーション用の U 字型ネットワーク用の新しい二重自己蒸留 (DSD) フレームワークを提案します。 DSD は、グラウンド トゥルース セグメンテーション ラベルからデコーダ層まで、また単一の U 字型ネットワークのエンコーダ層とデコーダ層の間で知識を抽出します。 DSD は、セグメンテーションのパフォーマンスをさらに向上させるために、U 字型ネットワークのバックボーン アーキテクチャに接続できる汎用トレーニング戦略です。 2 つの最先端の U 字型バックボーンに DSD を取り付け、2 つの公開 3D 医用画像セグメンテーション データセット (心臓下部構造と脳腫瘍) に対する広範な実験により、これらのバックボーンに比べて大幅な改善が実証されました。平均して、DSD を U 字型バックボーンに取り付けた後、心臓下部構造と脳腫瘍セグメンテーションの Dice 類似性スコアでそれぞれ 4.25% と 3.15% の改善が観察されました。
U-shaped networks and its variants have demonstrated exceptional results for medical image segmentation. In this paper, we propose a novel dual self-distillation (DSD) framework for U-shaped networks for 3D medical image segmentation. DSD distills knowledge from the ground-truth segmentation labels to the decoder layers and also between the encoder and decoder layers of a single U-shaped network. DSD is a generalized training strategy that could be attached to the backbone architecture of any U-shaped network to further improve its segmentation performance. We attached DSD on two state-of-the-art U-shaped backbones, and extensive experiments on two public 3D medical image segmentation datasets (cardiac substructure and brain tumor) demonstrated significant improvement over those backbones. On average, after attaching DSD to the U-shaped backbones, we observed an improvement of 4.25% and 3.15% in Dice similarity score for cardiac substructure and brain tumor segmentation respectively.