FANetv2通过迭代反馈机制提升肠镜息肉分割精度,助力早期癌症筛查。
Transformer-Enhanced Iterative Feedback Mechanism for Polyp Segmentation
- 引入迭代反馈注意力机制,利用前轮预测结果优化后续分割
- 在BKAI-IGH和CVC-ClinicDB上达0.9186和0.9481的DSC得分
- 结合文本提示区分息肉数量与大小,支持辅助分类
结直肠癌(CRC)是美国第三大常见癌症,也是男女中最主要的癌症死亡原因之一,尤其在50岁以下男性中发病率最高。结肠镜检查是早期诊断CRC的金标准,但内镜医师技能差异大,漏诊率高。自动化息肉分割可降低漏诊率,实现早期治疗。为此,本文提出 extit{ extbf{FANetv2}}——一种先进的编码器-解码器网络,用于从结肠镜图像中精准分割息肉。该模型基于Otsu阈值生成的初始掩码,通过新型反馈注意力机制,逐轮迭代优化二值分割结果。同时,采用文本引导策略,融合关于息肉数量(一个或多个)和大小(小、中、大)的关键信息,增强特征表示能力。该双任务方法显著提升了分割精度,并支持息肉属性的辅助分类。在公开数据集BKAI-IGH和CVC-ClinicDB上的全面评估显示,FANetv2分别取得0.9186和0.9481的高骰子相似系数(DSC),以及2.83和3.19的低豪斯多夫距离。代码已开源于https://github.com/xxxxx/FANetv2。
原文摘要 · Abstract (English)
Colorectal cancer (CRC) is the third most common cause of cancer diagnosed in the United States and the second leading cause of cancer-related death among both genders. Notably, CRC is the leading cause of cancer in younger men less than 50 years old. Colonoscopy is considered the gold standard for the early diagnosis of CRC. Skills vary significantly among endoscopists, and a high miss rate is reported. Automated polyp segmentation can reduce the missed rates, and timely treatment is possible in the early stage. To address this challenge, we introduce \textit{\textbf{\ac{FANetv2}}}, an advanced encoder-decoder network designed to accurately segment polyps from colonoscopy images. Leveraging an initial input mask generated by Otsu thresholding, FANetv2 iteratively refines its binary segmentation masks through a novel feedback attention mechanism informed by the mask predictions of previous epochs. Additionally, it employs a text-guided approach that integrates essential information about the number (one or many) and size (small, medium, large) of polyps to further enhance its feature representation capabilities. This dual-task approach facilitates accurate polyp segmentation and aids in the auxiliary classification of polyp attributes, significantly boosting the model's performance. Our comprehensive evaluations on the publicly available BKAI-IGH and CVC-ClinicDB datasets demonstrate the superior performance of FANetv2, evidenced by high dice similarity coefficients (DSC) of 0.9186 and 0.9481, along with low Hausdorff distances of 2.83 and 3.19, respectively. The source code for FANetv2 is available at https://github.com/xxxxx/FANetv2.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。