用可微凸性先验提升图像分割形状精度,无需阈值处理。
D-Convexity: A Unified Differentiable Convex Shape Prior via Quasi-Concavity for Data-driven Image Segmentation

- 基于输出掩码的拟凹性构建统一可微凸性约束。
- 在多个数据集上显著改善分割形状规则性,优于专用模型。
- 统一了多种传统凸形模型,适合需要精准形状的分割任务。
凸性是自然与人造结构的基本几何先验,但在端到端可训练分割网络中难以有效施加。本文从函数视角重新审视凸性,提出一种基于网络输出掩码函数u的拟凹性、无需阈值的统一凸性先验。不直接约束单一二值分割,而是要求u的所有超水平集均为凸集,将全局形状约束转化为u及其导数的局部可微不等式。由此推导出零阶、一阶和二阶表征,分别对应局部中点凸化算法、与支撑超平面相关的梯度条件,以及表达为切平面上二次型的充分二阶不等式。一阶与二阶形式生成紧凑卷积损失,可密集应用于图像而不需阈值。所提出的凸梯度投影模块(CGPM)使这些拟凹性损失能无缝集成至现代分割网络,持续强化凸性并提升形状规整性,在多个数据集上超越专用于视网膜分割的模型及先前的形状感知方法。值得注意的是,该分析统一了从离散1-0-1线约束、图割凸性公式到基于曲率或符号距离拉普拉斯的水平集先验等多种以往凸形模型,形成单一连续可微框架。
原文摘要 · Abstract (English)
Convexity is a fundamental geometric prior that underlies many natural and man-made structures, yet remains challenging to impose effectively in end-to-end trainable segmentation networks. We revisit convexity from a functional perspective and propose a unified, threshold-free convexity prior based on the quasi-concavity of the network's output mask function u. Instead of constraining a single binary segmentation, we require all super-level sets of u to be convex, transforming global shape constraints into local, differentiable inequalities on u and its derivatives. From this principle, we derive zero, first, and second-order characterizations, yielding respectively a local midpoint convexification algorithm, a gradient-based condition linked to supporting hyperplanes, and a sufficient second-order inequality expressed as a quadratic form on the tangent plane. The first and second-order formulations produce a compact convolutional loss that can be densely applied across the image without thresholding. Our quasi-concavity losses integrate seamlessly with modern segmentation networks via the proposed convex gradient projection module (CGPM). They consistently enforce convexity and improve shape regularity across multiple datasets, outperforming networks tailored for retinal segmentation and surpassing previous shape-aware methods. Remarkably, our analysis unifies a wide spectrum of previous convex shape models, from discrete 1-0-1 line constraints and graph-cuts convexity formulations to curvature or signed distance Laplacian based level-set priors, within a single continuous and differentiable framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。