论文标题:FAMNet: Frequency-aware Matching Network forCross-domain Few-shot Medical Image Segmentation
作者:Yuntian Bo, Yazhou Zhu, Lunbo Li, Haofeng Zhang(南京理工大学计算机科学与工程学院)
发表会议:AAAI 2025(人工智能顶级会议,CCF A类)
出版日期:2024年12月27日(arXiv v4)
文献来源:arXiv:2412.09319v4 [cs.CV]
研究问题:跨域少样本医学图像分割(CD-FSMIS),解决传统少样本分割模型在跨成像域(如CT→MRI)时的域偏移问题。
技术目标:在目标域标注极少的情况下,实现对新域医学图像(如不同设备、机构采集的数据)的精确分割。
核心方案:提出频率感知匹配网络(FAMNet),通过频率域分解和跨域特征对齐,抑制域差异并增强泛化性。
关键技术术语:
频率感知匹配(FAM, Frequency-aware Matching):将特征分解为高、中、低频带,并进行域感知加权。
多谱融合(MSF, Multi-Spectral Fusion):通过跨注意力机制融合频域特征,提取域不变信息(DII)。
域无关频带(DAFBs, Domain-Agnostic FrequencyBands)与域特有频带(DSFBs, Domain-Specific Frequency Bands):区分结构信息(中频)和风格信息(高/低频)。
逻辑关系:FAM模块先解耦频域特征,MSF模块进一步强化DAFBs并抑制DSFBs,形成端到端的跨域适配流程。
现有局限性:传统空间域匹配方法(如原型网络)对域偏移敏感,易过拟合源域特征。
改进思路:通过二维快速傅里叶变换(2D-FFT)将特征映射到频域,解耦域变/域不变成分。
创新关联:将自然图像中的频域泛化理论(如FSDR[1])扩展到医学影像,结合元学习框架。
跨模态数据集:腹部CT(20例)与MRI(20例),包含肝、双肾、脾4类器官。
跨序列数据集:心脏LGE与b-SSFP MRI(各45例),分割左心室血池、心肌等。
跨机构数据集:前列腺MRI(UCLH 321例 + NCI 82例),评估设备差异影响。
预处理:使用3D超体素聚类生成伪掩码,遵循episode元学习范式。
FSMIS模型:PANet、SSL-ALPNet、ADNet等。
CD-FSS模型:PATNet、IFA,未微调直接比较。
公平性:所有模型在1-way 1-shot设置下评估。
Dice系数
训练细节:SGD优化器,初始学习率0.001,迭代39K次。
跨模态任务(CT→MRI):Dice均值65.79%,显著优于第二名(PATNet的57.29%)。
消融实验显示FAM模块贡献最大(提升29.56%),MSF模块进一步优化3.77%。
频带分频注意力(DAFBs加权 vs. DSFBs抑制)使模型聚焦结构信息,Dice提升7.11%(见表4)。
在心脏跨序列任务中,LV-BP分割Dice达86.64%,接近非跨域场景精度。
直接丢弃高/低频带会导致性能下降3.90%,验证MSF模块的必要性。
频带划分比例3:4:3(低:中:高)时最优,过窄的中频带会损失结构信息。
理论层面:首次将频域解耦思想系统化应用于医学图像跨域分割,提出DAFBs/DSFBs的定量划分准则。
工程层面:设计即插即用的2D-FFT模块,支持端到端集成至U-Net等主流架构。
计算复杂度:频域变换与跨注意力机制增加计算开销,实时性受限。
频带划分敏感性:最佳比例依赖人工调参,需自适应机制。
跨域任务:适用于CT/MRI模态转换、跨机构数据协同(如多医院联合诊断),解决设备差异导致的分布偏移。
少样本场景:在标注稀缺的罕见病分割(如特定肿瘤)中,通过频域对齐减少对大量标注的依赖。
边缘计算适配:FFT模块可优化为轻量级版本,部署至移动医疗设备。
局限性适配:对纹理稀疏图像(如X光)效果可能受限,需结合空间域增强。
引入动态频带划分(如借鉴ConvNeXt的频域自适应);
结合扩散模型增强少样本生成能力,合成跨域训练数据。