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