将数学先验融入深度学习,提升医学图像分割精度
M-Net: Integrating Spectral Features and Physical Field Operators into Deep Learning for Medical Image Segmentation

- 引入谱特征与物理场算子作为数学先验增强网络理解能力
- 在三个基准数据集上分割精度超越基线12.37%以上
- 适合关注医学图像建模与可解释性研究的学者
深度学习在医学图像分割中取得显著进展,但纯数据驱动方法常忽略图像固有的数学结构。本文提出M-Net(数学增强网络),在U-Net中集成三种互补数学先验:(1) 基于中心化局部像素矩阵条件数的连续谱特征,提供纹理病态性的可微度量;(2) 从图像梯度场计算的物理场算子(散度与离散类旋度边界不规则算子),捕捉焦点强度极值与边缘非光滑性;(3) 数学注意力门(MAG),在跳跃连接处自适应融合数学特征与CNN深层特征。在LiTS、KiTS和BraTS三个基准测试中,M-Net分别获得78.42%、76.15%、83.67%的Dice分数,优于基线U-Net 12.37%、3.52%、5.55%。消融实验表明,条件数特征比二值可逆特征提升2.14%,MAG比简单拼接提升1.45%。结论:数学归纳偏置能有效补充医学图像分割信息,连续条件数特征优于离散替代方案,MAG可保持先验信息贯穿网络。本工作为将线性代数与向量微积分融入深度架构提供了新路径。
原文摘要 · Abstract (English)
Purpose: Deep learning-based medical image segmentation has achieved remarkable success, yet purely data-driven approaches often fail to exploit the rich mathematical structure inherent in medical images. We investigate whether explicit mathematical inductive biases, specifically matrix spectral analysis and vector calculus operators, can enhance segmentation beyond data-driven learning alone. Methods: We propose M-Net (Math-Augmented Network), which integrates three complementary mathematical priors into U-Net: (1) continuous spectral features derived from the condition number of centered local pixel matrices, providing a differentiable measure of texture ill-conditioning; (2) physical field operators (divergence and a discrete curl-like boundary irregularity operator) computed from image gradient fields, capturing focal intensity extrema and edge non-smoothness; and (3) a Math-Attention Gate (MAG) that adaptively fuses mathematical features with CNN-extracted deep features at skip connections. Results: Experiments on three benchmarks (LiTS, KiTS, and BraTS) show that M-Net achieves Dice scores of 78.42%, 76.15%, and 83.67%, outperforming baseline U-Net by 12.37%, 3.52%, and 5.55% on liver, kidney, and brain tumor segmentation, respectively. Ablations reveal that the condition-number feature contributes a 2.14% gain over binary invertibility features, while MAG adds 1.45% over simple concatenation. Conclusion: M-Net establishes that mathematical inductive biases provide effective complementary information for medical image segmentation. The continuous condition-number feature offers superior gradient information over discrete alternatives, and MAG preserves these priors throughout the network. This work opens avenues for integrating linear algebra and vector calculus into deep architectures for medical imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。