LMS-Net:基于学习型Mumford-Shah模型的医学图像少样本分割方法

【字体: 时间:2025年06月17日 来源:Medical Image Analysis 10.7

编辑推荐:

  针对医学图像少样本分割(FSS)中结构先验利用不足和模型可解释性差的问题,研究人员提出学习型Mumford-Shah网络(LMS-Net)。该模型通过原型更新与掩码更新的交替优化框架,将像素-原型比对机制与深度先验技术结合,在三个公开医学数据集上展现出优越的准确性和鲁棒性,为医疗AI的可解释性研究提供了新范式。

  

在医学影像分析领域,精准分割器官和病变区域是辅助诊断的关键环节。尽管全监督深度学习方法在充足标注数据下表现优异,但医疗场景中特定类别的标注数据往往极其稀缺。这种"数据饥渴"问题催生了少样本语义分割(Few-shot Semantic Segmentation, FSS)技术的研究热潮。现有基于原型比对(prototypical matching)的FSS方法虽能利用少量标注样本,却存在两大瓶颈:一是未能充分融入语义区域的空间连续性等物理结构先验,二是模型决策过程如同"黑箱",难以满足医疗AI对可解释性的严苛要求。

针对这一挑战,研究人员创新性地将传统变分分割方法与深度学习相结合。受启发于Mumford-Shah模型在数学上的严谨性,团队提出学习型Mumford-Shah网络(LMS-Net)。该工作巧妙地将像素-原型比对机制对应为模型的数据保真项,同时采用深度先验技术捕捉复杂解剖结构,构建出兼具数学可解释性与深度学习性能的新型框架。相关成果发表在《Medical Image Analysis》期刊,为医疗影像的少样本分析提供了新思路。

研究团队采用深度展开(deep unfolding)技术实现算法创新。首先建立学习型Mumford-Shah(LMS)变分模型,通过动量更新变换器(Momentum Update Transformer, MUT)实现原型迭代优化,利用原始-对偶算法(primal-dual algorithm)解耦掩码更新任务,最终将优化步骤展开为可端到端训练的网络模块。实验采用CHAOS-T2、Synapse-CT和MS-CMRSeg三个多模态医学数据集验证性能,涵盖腹部器官MRI/CT和心脏MRI分割任务。

Variational image segmentation
研究揭示传统Mumford-Shah模型通过能量最小化框架平衡数据保真项与全变分正则项,但其手工设计先验难以适应医学图像复杂场景。LMS模型创新性地将原型比对机制数学化为保真项,深度先验替代传统正则项,形成统一理论框架。

Problem definition
采用episodic训练范式,将数据集划分为训练集Dtr
和测试集Dte
,确保类别无交集(Ctr
∩Cte
=?)。每个episode包含支持集(support set)和查询集(query set),模拟临床少样本场景。

Datasets
在包含3779张CT切片的Synapse-CT数据集中,模型对左右肾脏、肝脏等器官的分割Dice系数提升显著。MS-CMRSeg心脏MRI数据验证了方法对复杂解剖结构的适应性,CHAOS-T2测试显示优于对比方法3.2%的边界精度。

Conclusion
研究证实LMS-Net通过数学建模实现了FSS任务的可解释性突破。动量更新机制有效聚合历史原型信息,原始-对偶网络(PD-Net)将掩码优化分解为数据子问题(闭式解)与先验子问题(CNN去噪器),这种解耦式设计较传统端到端网络参数量减少18%。

讨论
该工作的核心价值在于建立了变分方法与深度学习的双向桥梁:既赋予深度学习模型数学可解释性,又为传统模型注入数据驱动能力。临床意义在于,当面对罕见病或新出现病变类型时,仅需少量标注即可实现可靠分割,极大降低医疗AI落地门槛。未来可扩展至3D医学影像和多器官联合分割场景。

相关新闻
生物通微信公众号
微信
新浪微博
  • 急聘职位
  • 高薪职位

知名企业招聘

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号