arXiv:2503.03365cs.CV2025-03中稿 · BMVC 2025被引 2

首个专注拓扑准确性的图像分割数据集,助力评估分割方法真实性能。

TopoMortar: A dataset to evaluate image segmentation methods focused on topology accuracy

  • 构建含多种标签与挑战的数据集,分离方法性能与数据干扰
  • clDice在拓扑准确性上表现最佳,简单方法可超越复杂损失函数
  • 适合研究拓扑损失、数据增强与自蒸馏的学者使用

我们提出TopoMortar,一个专为评估拓扑感知图像分割方法(如拓扑损失函数)而设计的砖墙数据集。该数据集旨在通过两种方式检验方法在提升拓扑准确性方面的有效性:首先,消除影响拓扑损失函数效果的数据集挑战(如小训练集、噪声标签、分布外测试图像);其次,允许在同一数据集中引入不同挑战,从而将方法性能与数据干扰分离。TopoMortar包含三类标签(精确标签、伪标签、噪声标签)、两组固定训练集(大、小),以及分布内和分布外测试图像。我们在其上对比了八种损失函数,发现clDice在拓扑准确性上最优,其余损失函数的优势则依赖于具体设置。此外,我们表明数据增强与自蒸馏可使交叉熵Dice损失超越多数拓扑损失函数,并能进一步提升拓扑损失函数的表现。数据集与代码见https://jmlipman.github.io/TopoMortar。

原文摘要 · Abstract (English)

We present TopoMortar, a brick wall dataset that is the first dataset specifically designed to evaluate topology-focused image segmentation methods, such as topology loss functions. Motivated by the known sensitivity of methods to dataset challenges, such as small training sets, noisy labels, and out-of-distribution test-set images, TopoMortar is created to enable in two ways investigating methods' effectiveness at improving topology accuracy. First, by eliminating dataset challenges that, as we show, impact the effectiveness of topology loss functions. Second, by allowing to represent different dataset challenges in the same dataset, isolating methods' performance from dataset challenges. TopoMortar includes three types of labels (accurate, pseudo-labels, and noisy labels), two fixed training sets (large and small), and in-distribution and out-of-distribution test-set images. We compared eight loss functions on TopoMortar, and we found that clDice achieved the most topologically accurate segmentations, and that the relative advantageousness of the other loss functions depends on the experimental setting. Additionally, we show that data augmentation and self-distillation can elevate Cross entropy Dice loss to surpass most topology loss functions, and that those simple methods can enhance topology loss functions as well. TopoMortar and our code can be found at https://jmlipman.github.io/TopoMortar

图像分割拓扑优化数据集损失函数

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