摘要
本申请公开了一种基于SAM的半监督适配器微调与提示学习的医学图像分割方法及相关装置。该方法包括:导入并预处理医学图像数据集,加载预训练SAM模型及YOLO提示模型,在SAM编码器各视觉变换器块插入双适配器,解码器嵌入超提示适配器、MLP对齐适配器及交叉注意力适配器;基于标注数据训练提示模型,通过KL散度筛选合格未标注数据,生成点击提示并利用SAM模型进行伪标签分割;构建含对比流形正则化项的联合损失函数,在冻结SAM主干前提下,结合伪标签与标注数据执行适配器微调。所述装置包括数据导入单元、模型加载单元及提示训练单元。本发明有效减少标注依赖,增强模型泛化能力及分割精度,同时保留SAM主干的原有性能优势。
技术关键词
适配器
医学图像分割方法
医学图像数据集
注意力
核密度估计法
解码器
标签
编码器
计算机程序产品
加载单元
变换器
联合损失函数
计算机程序代码
视觉
存储单元