arXiv reaDer
不確実性を意識した注意メカニズムによる肺結節のセグメンテーションと信頼度の低い領域の予測
Lung Nodule Segmentation and Low-Confidence Region Prediction with Uncertainty-Aware Attention Mechanism
放射線科医はさまざまなトレーニングと臨床経験を持っているため、肺結節のさまざまなセグメンテーション アノテーションが発生し、セグメンテーションの不確実性が生じる可能性があります。従来の方法では、通常、単一のアノテーションを学習対象として選択するか、さまざまなアノテーションの潜在空間を学習しようとしますが、これらのアプローチは、複数のアノテーションに根付いた合意または不一致の貴重な情報を無駄にします。この論文では、より良いセグメンテーションを促進するために、複数の注釈間のコンセンサスと不一致を利用する不確実性認識アテンションメカニズム (UAAM) を提案します。これを実現するために、低信頼性 (LC) マスクと高信頼性 (HC) マスクを組み合わせた多信頼性マスク (MCM) を導入します。 LC マスクは、セグメンテーションの信頼度が低い領域を示します。これにより、放射線科医の間でセグメンテーション オプションが異なる可能性があります。 UAAM に続いて、3 つのモジュールを含む Uncertainty-Guide Segmentation Network (UGS-Net) をさらに設計します。肺結節の一般的な特徴をキャプチャする特徴抽出モジュール、注釈の 3 つの特徴を生成する Uncertainty-Aware モジュールです。結合、交差、および注釈セット、および 3 つのフィーチャ間の距離を使用して最終セグメンテーション、LC マスク、および HC マスクの予測のバランスをとる交差-結合制約モジュール。私たちの方法のパフォーマンスを完全に実証するために、LIDC-IDRI で複雑な結節検証を提案します。これは、U-Net を使用してセグメント化することが困難な肺結節での UGS-Net のセグメンテーション パフォーマンスをテストします。実験結果は、私たちの方法がU-Netによるセグメンテーションが不十分な結節のセグメンテーションパフォーマンスを大幅に改善できることを示しています。
Radiologists have different training and clinical experiences, which may result in various segmentation annotations for lung nodules, causing segmentation uncertainty. Conventional methods usually select a single annotation as the learning target or try to learn a latent space of various annotations, but these approaches waste the valuable information of consensus or disagreements ingrained in the multiple annotations. In this paper, we propose an Uncertainty-Aware Attention Mechanism (UAAM) that utilizes consensus and disagreements among multiple annotations to facilitate better segmentation. To achieve this, we introduce the Multi-Confidence Mask (MCM), which is a combination of a Low-Confidence (LC) Mask and a High-Confidence (HC) Mask. The LC mask indicates regions with a low segmentation confidence, which may cause different segmentation options among radiologists. Following UAAM, we further design an Uncertainty-Guide Segmentation Network (UGS-Net), which contains three modules: a Feature Extracting Module that captures a general feature of a lung nodule, an Uncertainty-Aware Module that produces three features for the annotations' union, intersection, and annotation set, and an Intersection-Union Constraining Module that uses distances between the three features to balance the predictions of final segmentation, LC mask, and HC mask. To fully demonstrate the performance of our method, we propose a Complex Nodule Validation on LIDC-IDRI, which tests UGS-Net's segmentation performance on lung nodules that are difficult to segment using U-Net. Experimental results demonstrate that our method can significantly improve the segmentation performance on nodules with poor segmentation by U-Net.
updated: Wed Apr 12 2023 00:53:00 GMT+0000 (UTC)
published: Wed Mar 15 2023 07:31:55 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アソシエイト