画像セグメンテーション用の深層畳み込みニューラル ネットワークは、ラベル構造を明示的に学習せず、誤った構造のセグメンテーションを生成する可能性があります。たとえば、気道や血管などの木のような構造のセグメンテーションでは、切断された円筒構造があります。この論文では、ラベル構造に関する情報を暗黙的に組み込んで、初期セグメンテーションからそのようなエラーを修正するための新しいラベル改良方法を提案します。この方法は、2 つの新しい部分を備えています: 1) 合成構造エラーを生成するモデル、および 2) 外観が実際の初期セグメンテーションに似ている合成セグメンテーション (エラーあり) を生成するラベル外観シミュレーション ネットワーク。これらの合成セグメンテーションと元の画像を使用して、エラーを修正し、初期セグメンテーションを改善するようにラベル調整ネットワークをトレーニングします。提案された方法は、胸部コンピューター断層撮影 (CT) スキャンからの気道セグメンテーションと、脳の 3D CT 血管造影 (CTA) 画像からの脳血管セグメンテーションの 2 つのセグメンテーション タスクで検証されます。どちらのアプリケーションでも、私たちの方法は、標準の 3D U-Net やその他の以前の改良アプローチよりも大幅に優れていました。モデルのトレーニングに追加のラベルなしデータを使用すると、改善はさらに大きくなります。アブレーション研究では、提案された方法のさまざまなコンポーネントの値を示します。
Deep convolutional neural networks for image segmentation do not learn the label structure explicitly and may produce segmentations with an incorrect structure, e.g., with disconnected cylindrical structures in the segmentation of tree-like structures such as airways or blood vessels. In this paper, we propose a novel label refinement method to correct such errors from an initial segmentation, implicitly incorporating information about label structure. This method features two novel parts: 1) a model that generates synthetic structural errors, and 2) a label appearance simulation network that produces synthetic segmentations (with errors) that are similar in appearance to the real initial segmentations. Using these synthetic segmentations and the original images, the label refinement network is trained to correct errors and improve the initial segmentations. The proposed method is validated on two segmentation tasks: airway segmentation from chest computed tomography (CT) scans and brain vessel segmentation from 3D CT angiography (CTA) images of the brain. In both applications, our method significantly outperformed a standard 3D U-Net and other previous refinement approaches. Improvements are even larger when additional unlabeled data is used for model training. In an ablation study, we demonstrate the value of the different components of the proposed method.