arXiv reaDer
GAEI-UNet: 血管画像セグメンテーションのためのグローバルな注目と柔軟なインタラクション U-Net
GAEI-UNet: Global Attention and Elastic Interaction U-Net for Vessel Image Segmentation
血管画像のセグメンテーションは医療診断において極めて重要な役割を果たし、血管疾患の早期発見と治療に役立ちます。深層学習に基づくセグメンテーションは有望な結果を示していますが、小さな構造を効果的にセグメント化し、それらの間の接続を維持することは依然として課題です。これらの制限に対処するために、我々は、グローバルな注目と弾力的な相互作用ベースの技術を組み合わせた新しいモデルである GAEI-UNet を提案します。 GAEI-UNet は、グローバルな空間情報とチャネル コンテキスト情報を活用して、U-Net アーキテクチャ内の高レベルのセマンティック理解を強化し、小型船舶の正確なセグメンテーションを可能にします。さらに、これらの微細構造間の接続性を向上させるために、弾性相互作用に基づく損失関数を採用します。ターゲット形状と予測形状の間の位置ずれによって生成される力を捉えることにより、モデルは血管ネットワークの正しいトポロジーを維持することを効果的に学習します。網膜血管データセットの評価 -- DRIVE は、計算の複雑さを大幅に増加させることなく、SE と小さな構造の接続性の点で GAEI-UNet の優れたパフォーマンスを実証します。この研究は、血管画像セグメンテーションの分野を進歩させ、より正確で信頼性の高い診断ツールを医療界に提供することを目的としています。実装コードは Code で入手できます。
Vessel image segmentation plays a pivotal role in medical diagnostics, aiding in the early detection and treatment of vascular diseases. While segmentation based on deep learning has shown promising results, effectively segmenting small structures and maintaining connectivity between them remains challenging. To address these limitations, we propose GAEI-UNet, a novel model that combines global attention and elastic interaction-based techniques. GAEI-UNet leverages global spatial and channel context information to enhance high-level semantic understanding within the U-Net architecture, enabling precise segmentation of small vessels. Additionally, we adopt an elastic interaction-based loss function to improve connectivity among these fine structures. By capturing the forces generated by misalignment between target and predicted shapes, our model effectively learns to preserve the correct topology of vessel networks. Evaluation on retinal vessel dataset -- DRIVE demonstrates the superior performance of GAEI-UNet in terms of SE and connectivity of small structures, without significantly increasing computational complexity. This research aims to advance the field of vessel image segmentation, providing more accurate and reliable diagnostic tools for the medical community. The implementation code is available on Code.
updated: Sun Aug 20 2023 03:42:37 GMT+0000 (UTC)
published: Wed Aug 16 2023 13:10: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アソシエイト