arXiv:2607.03062cs.CV2026-07

轻量级肠镜息肉分割新方法,通过递归修正预测空间提升精度。

Lightweight Polyp Segmentation via a Gain-Aware Prediction-Space Recursive Controller

论文配图:Lightweight Polyp Segmentation via a Gain-Aware Prediction-Space Recursive Controller
图 1 · 摘自论文原文
  • 在主干网络输出上直接递归修正预测结果,不依赖复杂结构。
  • 7种轻量模型在Kvasir-SEG上平均Dice提升2.1%,IoU提升1.8%。
  • 适合部署于移动端或嵌入式设备,静态开销极小。

尽管轻量级息肉分割对低成本部署至关重要,但现有性能提升常源于升级的主干网络、复杂解码器或重型精修分支,难以判断轻量修正机制本身是否有效。本文将精修建模为预测空间中的递归校正任务,提出一种直接作用于主干网络logits的递归控制器。在固定迭代次数下,该控制器聚合差异与不确定性证据,更新一个紧凑状态以追踪近期修正效用,并施加加性残差logit修正。该修正路径始终保持轻量,适配主机,且部署明确。采用统一的Kvasir训练协议,在七种轻量主干网络上于Kvasir-SEG及三个迁移数据集上评估,衡量分割精度(Dice/IoU)与部署效率(参数量、GMACs、峰值内存)。实验表明,该控制器在源域持续提升性能,在代表性设备上达到与训练侧基线及更重结构化精修器相当的水平,且在迁移任务中实现选择性增益,静态开销极低。代码已公开于https://github.com/tyui99/Gain-Aware-Prediction-Space-Recursive-Controller。

原文摘要 · Abstract (English)

While lightweight polyp segmentation is highly desirable for low-cost deployment, reported performance gains often stem from upgraded backbone encoders, complex decoders, or heavy refinement branches. Consequently, it remains difficult to isolate whether a lightweight correction mechanism is inherently effective on its own. We address this limitation by formulating refinement as a prediction-space recursive correction task, introducing a recursive controller that operates directly on backbone logits. Under a fixed recursion budget, this controller aggregates discrepancy and uncertainty evidence, updates a compact state tracking recent correction utility, and applies additive residual logit corrections. By design, this correction path remains small, host-portable, and deployment-explicit. Utilizing a unified Kvasir-trained protocol, we evaluate our approach across seven lightweight backbones on Kvasir-SEG and three transfer datasets, measuring segmentation accuracy (Dice/IoU) alongside deployment efficiency (parameters, GMACs, and peak memory). The controller yields consistent improvements in the source domain, achieves competitive performance against both training-side baselines and heavier structural refiners on representative hosts, and delivers selective transfer gains with minimal static overhead. Code is available at https://github.com/tyui99/Gain-Aware-Prediction-Space-Recursive-Controller.

医学图像轻量分割递归修正部署优化

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