用几何方法自动修复3D细胞分割中过度分割的问题。
3D Cell Oversegmentation Correction via Geo-Wasserstein Divergence
- 基于2D几何与3D拓扑特征构建分类器识别过度分割
- 提出Geo-Wasserstein散度量化细胞形态变化趋势
- 支持跨物种迁移,适合生物图像分析研究者
3D细胞分割常因过度分割(single cell被错误切分为多个片段)而受损,此类错误难以修复,因其外观与相邻细胞间的自然间隙相似。本文首次将过度分割问题形式化,并提出几何框架进行识别与修正。方法利用从缺陷3D分割结果中提取的2D几何与3D拓扑特征,训练预训练分类器。同时引入新型度量Geo-Wasserstein散度,以几何感知方式量化2D细胞掩码形状的变化趋势。在植物数据集(含合成与真实过度分割样本)及动物数据集上进行充分验证,展示跨域迁移能力。消融实验表明该散度对性能贡献显著。提供清晰流程,供用户基于任意标注数据集构建预训练模型。
原文摘要 · Abstract (English)
3D cell segmentation methods are often hindered by \emph{oversegmentation}, where a single cell is incorrectly split into multiple fragments. This degrades the final segmentation quality and is notoriously difficult to resolve, as oversegmentation errors often resemble natural gaps between adjacent cells. Our work makes two key contributions. First, for 3D cell segmentation, we are the first work to formulate oversegmentation as a concrete problem and propose a geometric framework to identify and correct these errors. Our approach builds a pre-trained classifier using both 2D geometric and 3D topological features extracted from flawed 3D segmentation results. Second, we introduce a novel metric, Geo-Wasserstein divergence, to quantify changes in 2D geometries. This captures the evolving trends of cell mask shape in a geometry-aware manner. We validate our method through extensive experiments on in-domain plant datasets, including both synthesized and real oversegmented cases, as well as on out-of-domain animal datasets to demonstrate transfer learning performance. An ablation study further highlights the contribution of the Geo-Wasserstein divergence. A clear pipeline is provided for end-users to build pre-trained models to any labeled dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。