arXiv:2510.15362stat.MLcs.CV2025-10NeurIPS

提出高效分割算法RankSEG-RMA,解决传统方法精度与速度的矛盾。

RankSEG-RMA: An Efficient Segmentation Algorithm via Reciprocal Moment Approximation

  • 用互反矩逼近法降低分割算法复杂度至线性时间
  • 在LiTS数据集上提速超千倍,仅需0.01秒完成预测
  • 支持非重叠分割场景,适用更广泛实际任务

语义分割需为图像中每个像素分配类别标签,通常以交并比(IoU)和Dice系数衡量预测与真实掩码的重叠程度。现有方法多通过估计像素级分类概率,再使用argmax或阈值化获得最终结果,但此类方法未直接优化分割指标,常导致不一致或次优性能。为此,研究者提出了新框架RankSEG,包含专为优化Dice和IoU设计的RankDice与RankIoU,虽能保证性能提升,却存在两大缺陷:计算开销大——RankDice复杂度为O(d log d),RankIoU达O(d^2),在LiTS数据集上预测耗时16.33秒,远高于argmax规则的0.01秒;且仅适用于允许多类重叠的场景,与主流非重叠分割基准不符。本文通过引入互反矩逼近(RMA),提出RankSEG-RMA,在保持相近性能的前提下将两类算法复杂度均降至O(d);同时基于RMA设计像素级得分函数,实现对非重叠分割场景的高效适配。

原文摘要 · Abstract (English)

Semantic segmentation labels each pixel in an image with its corresponding class, and is typically evaluated using the Intersection over Union (IoU) and Dice metrics to quantify the overlap between predicted and ground-truth segmentation masks. In the literature, most existing methods estimate pixel-wise class probabilities, then apply argmax or thresholding to obtain the final prediction. These methods have been shown to generally lead to inconsistent or suboptimal results, as they do not directly maximize segmentation metrics. To address this issue, a novel consistent segmentation framework, RankSEG, has been proposed, which includes RankDice and RankIoU specifically designed to optimize the Dice and IoU metrics, respectively. Although RankSEG almost guarantees improved performance, it suffers from two major drawbacks. First, it is its computational expense-RankDice has a complexity of O(d log d) with a substantial constant factor (where d represents the number of pixels), while RankIoU exhibits even higher complexity O(d^2), thus limiting its practical application. For instance, in LiTS, prediction with RankSEG takes 16.33 seconds compared to just 0.01 seconds with the argmax rule. Second, RankSEG is only applicable to overlapping segmentation settings, where multiple classes can occupy the same pixel, which contrasts with standard benchmarks that typically assume non-overlapping segmentation. In this paper, we overcome these two drawbacks via a reciprocal moment approximation (RMA) of RankSEG with the following contributions: (i) we improve RankSEG using RMA, namely RankSEG-RMA, reduces the complexity of both algorithms to O(d) while maintaining comparable performance; (ii) inspired by RMA, we develop a pixel-wise score function that allows efficient implementation for non-overlapping segmentation settings.

语义分割算法优化效率提升

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