少样本图像分割

Parent: ai_keywords

少样本图像分割(Few-shot Image Segmentation)

核心定义

少样本图像分割旨在利用极少量的标注样本(通常为1-5个),对从未见过的新类别实现像素级精确分割。其核心思路是通过元学习或度量学习,从支持集(support set)中提取类别原型,并与查询图像(query image)特征进行相似性匹配,从而在无额外标注下泛化至新类别。该技术对医学图像分析意义重大,尤其适用于罕见病变、解剖变异等标注成本极高的场景。

关键技术点

  1. 原型网络
    将支持图像中的前景/背景特征映射为类别原型向量,查询图像的每个像素与该原型计算距离(如余弦相似度),生成分割概率图。该方法结构简洁,且在小样本下稳定性高。

  2. 跨层级注意力与特征融合
    引入可变形注意力或跨层特征交互,增强支持集与查询图像间的上下文对应关系。例如,基于Transformer的分割网络利用自注意力捕获全局依赖,缓解小样本导致的特征退化。

  3. 自监督预训练与迁移学习
    在大规模无标签医学图像上(如脑MRI)进行对比学习或掩码重建,获得强表征能力。再通过少量标注样本微调,显著降低对新类别分割的过拟合风险。

  4. 数据增强与伪标签策略
    通过旋转、弹性变形、色彩抖动及生成对抗网络(GAN)合成新样本,扩充支持集多样性。同时利用模型预测的伪标签迭代自训练,进一步提升分割鲁棒性。

医学/神经科学应用场景

以首都医科大学神经病学团队的研究为例:脑小血管病(CSVD)罕见影像标志物的少样本分割。临床上,CSVD的腔隙灶、扩大的血管周围间隙等亚型具有高度异质性,且标注样本极其稀缺(例如单个中心仅数十例可靠标注)。采用少样本分割框架,利用少量(如3~5个)已标注的脑白质高信号(WMH) 病灶作为支持集,模型即可快速迁移至新出现的微出血灶分割任务。结合高分辨率磁敏感加权成像(SWI),特征原型网络能在百秒内完成全脑分割,准确率达专业医师的90%以上。这一方法极大降低了人工标注成本,助力CSVD影像组学的大规模队列研究,为早期认知障碍风险评估提供自动化工具。