计算机工程

作者：郭敏, 张熙涵, 李阳

单位：1. 长春工业大学计算机科学与工程学院, 吉林 长春 1300122. 东北师范大学前沿交叉研究院, 吉林 长春 130024

关键词：医学图像分割,半监督学习,注意力机制,平均教师模型,一致性正则化

基金：国家自然科学基金(NSFC12226003); 国家自然科学基金(NSFC61806024); 吉林省科技发展基金(20210201081GX); 吉林省教育厅科研项目(JJKH20230760KJ); 吉林省发改委产业技术研究与开发项目(2022C043-8); 吉林省发改委产业技术研究与开发项目(2023C031-5); 长春市科技发展计划项目(21GD01)

医学图像分割在疾病辅助诊断中起着关键的作用。现有的深度分割模型需要依赖带有标注的数据完成大规模训练, 而医学影像标注需要具有专业背景的临床医生进行像素级标注, 导致标注数据获取困难。基于半监督的医学图像分割方法利用少量的标注数据和大量的未标注数据进行学习, 可以在一定程度上缓解标注数据获取困难的问题。针对半监督分割模型不能充分利用未标注数据中的可学习信息的问题, 提出一种半监督分割模型TCA-Net。该模型使用U-Net作为骨干网络, 通过在U-Net中引入卷积块注意力模块(CBAM)与多头自注意力模块(MHA)来解决其在下采样过程中的信息丢失问题; 为了充分利用未标注数据中的不确定性信息, 构建一个教师互一致性模型, 该模型由具有1个编码器和3个略有不同的解码器的学生模型与教师模型组成, 通过在学生模型的概率映射与教师模型的伪标签之间添加一致性约束, 以此在训练过程中最小化输出之间的差异, 从而提升模型的分割效果。在公开的WORD腹部多器官数据集与ACDC心脏数据集上进行实验, 结果表明, 在使用20%标注数据的WORD数据集上, TCA-Net的Dice系数、Jaccard指数、HD95和ASD分别达到90.81%、83.79%、21.38和6.08, 在ACDC数据集上分别达到89.69%、81.94%、1.66和0.45。消融实验与对比实验结果表明, TCA-Net能够有效提升未标注数据的利用率, 在不同数据集上均达到了较好的分割效果, 验证了模型的鲁棒性。

来源：2024年第9期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第9期