arXiv reaDer
医療画像セグメンテーションの残留モーメント損失
Residual Moment Loss for Medical Image Segmentation
位置情報は、ターゲットオブジェクトの多様な構造をキャプチャする際の深層学習モデルに役立つことが証明されており、それに応じて医療画像のセグメンテーションの精度が向上します。ただし、ほとんどの既存の方法は、ネットワークが学習できるように、各ピクセルから等高線境界までの相対距離を記述する距離変換マップなど、暗黙的な方法で位置情報をエンコードします。これらの暗黙のアプローチは、ターゲットの位置情報(つまり絶対位置)を十分に活用していません。本論文では、深層学習ネットワークのトレーニング中にセグメンテーションターゲットの位置情報を明示的に埋め込むための新しい損失関数、つまり残差モーメント(RM)損失を提案します。特に、画像モーメントに動機付けられて、セグメンテーション予測マップとグラウンドトゥルースマップは座標情報によって重み付けされます。次に、RM損失により、ネットワークが2つの重み付けされたマップ間の一貫性を維持するように促されます。これにより、セグメンテーションネットワークがターゲットを簡単に特定し、多様体構造に関連する特徴を抽出できるようになります。 2つの公的に利用可能なデータセット、すなわち2D眼杯とディスクのセグメンテーションと3D左心房セグメンテーションで広範な実験を行うことにより、提案されたRM損失を検証します。実験結果は、セグメンテーションネットワークの精度を大幅に向上させるRM損失の有効性を示しています。
Location information is proven to benefit the deep learning models on capturing the manifold structure of target objects, and accordingly boosts the accuracy of medical image segmentation. However, most existing methods encode the location information in an implicit way, e.g. the distance transform maps, which describe the relative distance from each pixel to the contour boundary, for the network to learn. These implicit approaches do not fully exploit the position information (i.e. absolute location) of targets. In this paper, we propose a novel loss function, namely residual moment (RM) loss, to explicitly embed the location information of segmentation targets during the training of deep learning networks. Particularly, motivated by image moments, the segmentation prediction map and ground-truth map are weighted by coordinate information. Then our RM loss encourages the networks to maintain the consistency between the two weighted maps, which promotes the segmentation networks to easily locate the targets and extract manifold-structure-related features. We validate the proposed RM loss by conducting extensive experiments on two publicly available datasets, i.e., 2D optic cup and disk segmentation and 3D left atrial segmentation. The experimental results demonstrate the effectiveness of our RM loss, which significantly boosts the accuracy of segmentation networks.
updated: Sun Jun 27 2021 09:31:49 GMT+0000 (UTC)
published: Sun Jun 27 2021 09:31:49 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アソシエイト