用标量场建模细胞分割,无需正则化即可实现高精度边缘检测。
Single Tensor Cell Segmentation using Scalar Field Representations
- 通过求解泊松方程构建连续标量场,直接以分水岭法分割细胞。
- 在公开数据集上达到竞争性精度,边界清晰且对异常值鲁棒。
- 仅需一个张量训练网络,适合边缘计算场景部署。
本文从标量场视角研究图像中的细胞分割问题。目标是学习图像域上的连续标量场,使其通过分水岭方法生成鲁棒的细胞实例分割结果。该标量场由训练网络参数化,实验采用泊松偏微分方程和热方程稳态解的扩散模型作为场表示,仅通过最小化场残差即可获得解,无需额外正则化。该方法对训练数据中的异常值具有强鲁棒性,能有效保留细胞锐利边界。整个过程仅需一个张量训练U-Net,极大简化实现,降低训练与推理时间及内存占用,显著减少能耗,适用于边缘计算场景。在多个公开数据集上取得具有竞争力的结果,验证了该几何直观、简洁高效的方法在细胞分割任务中的有效性。
原文摘要 · Abstract (English)
We investigate image segmentation of cells under the lens of scalar fields. Our goal is to learn a continuous scalar field on image domains such that its segmentation produces robust instances for cells present in images. This field is a function parameterized by the trained network, and its segmentation is realized by the watershed method. The fields we experiment with are solutions to the Poisson partial differential equation and a diffusion mimicking the steady-state solution of the heat equation. These solutions are obtained by minimizing just the field residuals, no regularization is needed, providing a robust regression capable of diminishing the adverse impacts of outliers in the training data and allowing for sharp cell boundaries. A single tensor is all that is needed to train a \unet\ thus simplifying implementation, lowering training and inference times, hence reducing energy consumption, and requiring a small memory footprint, all attractive features in edge computing. We present competitive results on public datasets from the literature and show that our novel, simple yet geometrically insightful approach can achieve excellent cell segmentation results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。