arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.18539cs.CVcs.AI
收藏

基于骨架的方法增强边界分割的拓扑准确性

Enhancing Boundary Segmentation for Topological Accuracy with Skeleton-based Methods

University of Science and Technology Beijing(北京科技大学);Beijing Advanced Innovation Center for Materials Genome Engineering(北京材料基因组创新中心);School of Intelligence Science and Technology(智能科学与技术学院);Shunde Innovation School(顺德创新学校);Institute for Advanced Materials and Technology(先进材料与技术研究院);Key Laboratory of Intelligent Bionic Unmanned Systems(智能仿生无人系统重点实验室);Institute of Materials Intelligent Technology(材料智能技术研究院);Liaoning Academy of Materials(辽宁省材料科学院);School of Materials Science and Technology(材料科学与技术学院)

Chuni Liu, Boyuan Ma, Xiaojuan Ban, Yujie Xie, Hao Wang, Weihua Xue, Jingchao Ma, Ke Xu

AI总结:

提出Skea-Topo Aware损失函数,通过骨架感知加权和边界修正项提升网状图像边界分割的拓扑一致性,在三个数据集上相比13种方法VI指标提升最多7点。

摘要:

拓扑一致性在网状图像的边界分割任务中起着关键作用,例如神经元电子显微镜图像中的细胞膜分割、材料显微图像中的晶界分割以及航拍图像中的道路分割。在这些领域中,分割结果的拓扑变化对下游任务产生严重影响,甚至可能超过边界本身的错位。为了增强分割结果的拓扑准确性,我们提出了Skea-Topo Aware损失函数,这是一种新颖的损失函数,考虑了每个物体的形状和像素的拓扑重要性。它由两部分组成。首先,骨架感知加权损失通过更好地利用骨架建模物体几何来提高分割准确性。其次,边界修正项通过使用真实标签和预测中的前景和背景骨架,有效识别并强调预测误差中的拓扑关键像素。实验证明,在三个不同的边界分割数据集上,基于客观和主观评估,我们的方法在VI指标上相比13种最先进方法将拓扑一致性提高了最多7点。代码可在https://github.com/clovermini/Skea_topo获取。

英文摘要:

Topological consistency plays a crucial role in the task of boundary segmentation for reticular images, such as cell membrane segmentation in neuron electron microscopic images, grain boundary segmentation in material microscopic images and road segmentation in aerial images. In these fields, topological changes in segmentation results have a serious impact on the downstream tasks, which can even exceed the misalignment of the boundary itself. To enhance the topology accuracy in segmentation results, we propose the Skea-Topo Aware loss, which is a novel loss function that takes into account the shape of each object and topological significance of the pixels. It consists of two components. First, a skeleton-aware weighted loss improves the segmentation accuracy by better modeling the object geometry with skeletons. Second, a boundary rectified term effectively identifies and emphasizes topological critical pixels in the prediction errors using both foreground and background skeletons in the ground truth and predictions. Experiments prove that our method improves topological consistency by up to 7 points in VI compared to 13 state-of-art methods, based on objective and subjective assessments across three different boundary segmentation datasets. The code is available at https://github.com/clovermini/Skea_topo.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑