arXiv:2607.03058cs.CV2026-07

轻量级后处理模块,精准修复肠镜息肉分割边界误差

RIGS-Refiner: Risk-Guided Recursive Refinement in Prediction Space for Colonoscopy Polyp Segmentation

论文配图:RIGS-Refiner: Risk-Guided Recursive Refinement in Prediction Space for Colonoscopy Polyp Segmentation
图 1 · 摘自论文原文
  • 基于预测空间递归修正,仅用519参数和0.631 GFLOPs
  • 在多个数据集上提升分割精度,尤其改善模糊边界处的轮廓清晰度
  • 适合部署在资源受限的临床实时系统中

后处理可提升肠镜图像分割效果,但许多方法依赖额外校正头或多阶段流水线,带来显著参数或计算开销。对于息肉分割,主干模型输出通常整体合理,剩余误差集中于模糊边界和复杂局部结构。这些残差误差在肠镜图像中至关重要,因有效掩码需准确覆盖病灶并清晰勾勒细微黏膜过渡区轮廓。因此,我们提出RIGS-Refiner,一种轻量级后处理插件,实现预测空间中的风险引导递归精修。从冻结的主干预测出发,提取轻量级图像先验与预测线索,通过风险引导更新,并以共享递归单元写回残差修正。该模块仅增加519个参数和0.631 GFLOPs,保持高效部署性。实验使用Kvasir-SEG训练,在Kvasir、ClinicDB、ColonDB和ETIS数据集上评估,采用PraNet和SegFormer-B0作为两个冻结主干模型。结果表明,RIGS-Refiner在两种主干上均取得一致性能提升,且在效率-精度权衡上优于代表性后处理方法。代码已开源。

原文摘要 · Abstract (English)

Post-refinement can improve colonoscopy segmentation after host inference, but many designs still rely on extra correction heads or multi-stage pipelines with non-negligible parameter or computational cost. For polyp segmentation, host predictions are often already reasonable globally, with remaining errors clustered around ambiguous boundaries and difficult local structures. These residual errors matter in colonoscopy images because useful masks need correct lesion coverage and clean contour delineation across subtle mucosal transitions. This setting favors selective local repair in prediction space over reprocessing the entire mask. We therefore propose RIGS-Refiner, a lightweight post-refinement plugin for risk-guided recursive refinement in prediction space. Starting from a frozen host anchor prediction, RIGS-Refiner extracts lightweight image priors and prediction cues, applies risk-guided update, and writes back residual corrections through a shared recursive cell. The module adds only +519 parameters and +0.631 GFLOPs, keeping the refinement path compact for deployment. Experiments use Kvasir-SEG for training and Kvasir, ClinicDB, ColonDB, and ETIS for evaluation under two frozen hosts, namely PraNet and SegFormer-B0. Results show consistent gains on both hosts and a favorable efficiency-accuracy trade-off against representative post-refinement methods. Code is available at https://github.com/tyui99/RIGS-Refiner.

医学图像分割后处理轻量级模型肠镜

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。