arXiv reaDer
不確実性によるボトルネックへの注意 OAR セグメンテーションのための U-net
Uncertainty Driven Bottleneck Attention U-net for OAR Segmentation
コンピュータ断層撮影 (CT) 画像における危険臓器 (OAR) のセグメンテーションは、自動化されたセグメンテーション方法では困難な作業であり、下流の放射線治療計画にとって重要になる可能性があります。 U-net は、医用画像セグメンテーションのデファクト スタンダードとなり、医用画像セグメンテーション タスクの共通ベースラインとして頻繁に使用されます。この論文では、ノイズの多い補助デコーダを使用してノイズの多いセグメンテーションを生成する複数デコーダ U-net アーキテクチャを開発します。メイン ブランチからのセグメンテーションと補助ブランチからのノイズの多いセグメンテーションは、注意を推定するために一緒に使用されます。私たちの貢献は、2 つのデコーダー ブランチのソフトマックス確率から注意を引き出す新しい注意モジュールの開発です。 2 つのブランチからの 2 つのセグメンテーション マスクの結合と交差は、両方のデコーダが一致する情報と一致しない情報を伝達します。一致領域と不一致領域のソフトマックス確率は、不確実性の高低の指標です。したがって、これらの選択された領域の確率は、エンコーダのボトルネック層で注意として使用され、セグメンテーションのためにメインデコーダのみを通過します。正確な輪郭セグメンテーションのために、CT 強度統合正則化損失も開発しました。公開されている 2 つの OAR チャレンジ データセット、それぞれ Segthor と LCTSC でモデルをテストしました。注意モジュールと正則化損失の有効性を確認するために、提案された注意モデルと正則化損失の有無にかかわらず、各データセットで 12 個のモデルをトレーニングしました。実験では、両方のデータセットで明確な精度の向上 (2% から 5% のサイコロ) が示されています。実験のコードは、公開が承認された時点で利用可能になります。
Organ at risk (OAR) segmentation in computed tomography (CT) imagery is a difficult task for automated segmentation methods and can be crucial for downstream radiation treatment planning. U-net has become a de-facto standard for medical image segmentation and is frequently used as a common baseline in medical image segmentation tasks. In this paper, we develop a multiple decoder U-net architecture where a noisy auxiliary decoder is used to generate noisy segmentation. The segmentation from the main branch and the noisy segmentation from the auxiliary branch are used together to estimate the attention. Our contribution is the development of a new attention module which derives the attention from the softmax probabilities of two decoder branches. The union and intersection of two segmentation masks from two branches carry the information where both decoders agree and disagree. The softmax probabilities from regions of agreement and disagreement are the indicators of low and high uncertainty. Thus, the probabilities of those selected regions are used as attention in the bottleneck layer of the encoder and passes only through the main decoder for segmentation. For accurate contour segmentation, we also developed a CT intensity integrated regularization loss. We tested our model on two publicly available OAR challenge datasets, Segthor and LCTSC respectively. We trained 12 models on each dataset with and without the proposed attention model and regularization loss to check the effectiveness of the attention module and the regularization loss. The experiments demonstrate a clear accuracy improvement (2% to 5% Dice) on both datasets. Code for the experiments will be made available upon the acceptance for publication.
updated: Sun Mar 19 2023 23:45:32 GMT+0000 (UTC)
published: Sun Mar 19 2023 23:45:32 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アソシエイト