arXiv:2505.05806cs.CV2025-05被引 3

融合变分法与UNet,提升分割边界精度与可解释性

Image Segmentation via Variational Model Based Tailored UNet: A Deep Variational Framework

  • 用高阶修正的Cahn-Hilliard方程结合UNet结构
  • 在多个基准数据集上实现更优的边界分割效果
  • 适合需要高精度边缘的医学图像分割场景

传统基于偏微分方程的变分模型具有强数学可解释性与精确边界建模能力,但对参数敏感且计算成本高。深度学习模型如UNet参数量小、自动特征提取能力强,但缺乏理论解释性且需大量标注数据。为此,本文提出变分模型驱动的定制化UNet(VM_TUNet),将四阶修正的Cahn-Hilliard方程与UNet深度网络结合,融合变分方法的可解释性与边缘保持特性,以及神经网络的自适应特征学习能力。具体地,引入数据驱动算子替代人工调参,并采用定制有限点法(TFPM)强化高精度边界保持。在多个基准数据集上的实验表明,VM_TUNet在细粒度边界分割方面显著优于现有方法。

原文摘要 · Abstract (English)

Traditional image segmentation methods, such as variational models based on partial differential equations (PDEs), offer strong mathematical interpretability and precise boundary modeling, but often suffer from sensitivity to parameter settings and high computational costs. In contrast, deep learning models such as UNet, which are relatively lightweight in parameters, excel in automatic feature extraction but lack theoretical interpretability and require extensive labeled data. To harness the complementary strengths of both paradigms, we propose Variational Model Based Tailored UNet (VM_TUNet), a novel hybrid framework that integrates the fourth-order modified Cahn-Hilliard equation with the deep learning backbone of UNet, which combines the interpretability and edge-preserving properties of variational methods with the adaptive feature learning of neural networks. Specifically, a data-driven operator is introduced to replace manual parameter tuning, and we incorporate the tailored finite point method (TFPM) to enforce high-precision boundary preservation. Experimental results on benchmark datasets demonstrate that VM_TUNet achieves superior segmentation performance compared to existing approaches, especially for fine boundary delineation.

图像分割变分模型UNet边界保持

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