基于带平均曲率项的相场模型与UNet的图像分割方法

Image Segmentation Method Based on a Phase-Field Model with Mean Curvature Term and UNet

  • 摘要: 针对深度学习方法在带噪图像分割任务中存在的鲁棒性不足及物理可解释性差等问题,本研究提出了一种新型图像分割网络Variational Model with Mean Curvature Based Tailored UNet(VMMC-TUNet)。该模型创新性地将具有物理演化特性的平均曲率相场模型与UNet架构相结合,通过整合带平均曲率项的Cahn-Hilliard方程与经典UNet结构,构建了一个端到端的混合模型。具体实现上,研究采用控制变量与神经网络逼近策略,并运用量身定制有限点方法进行数值求解。实验结果表明,在添加高斯噪声、椒盐噪声和泊松噪声等多种噪声类型的测试数据集上,相较于现有方法,VMMC-TUNet在分割精度和边缘保持能力方面均表现出显著优势,有效提升了模型在噪声干扰下的鲁棒性和稳定性。

     

    Abstract: To address the issues of insufficient robustness and poor physical interpretability in deep learning methods for noisy image segmentation tasks, this study proposes a novel image segmentation network called Variational Model with Mean Curvature Based Tailored UNet (VMMC-TUNet). The model innovatively integrates a physically meaningful mean curvature phase-field model with the UNet architecture, combining the Cahn-Hilliard equation with a mean curvature term and the classical UNet structure to construct an end-to-end hybrid model. In implementation, the study employs control variables and neural network approximation strategies, along with a tailored finite-point method for numerical solution. Experimental results on test datasets with various types and intensities of noise, including Gaussian noise, salt-and-pepper noise, and Poisson noise, demonstrate that VMMC-TUNet exhibits significant advantages in segmentation accuracy and edge preservation compared to existing methods, effectively enhancing the model's robustness and stability under noise interference.

     

/

返回文章
返回