基于跨任务一致性的半监督肝脏CT图像分割 |
| |
作者姓名: | 李明漾 王庆凤 陈立伟 黄俊 周莹 |
| |
作者单位: | 1. 西南科技大学计算机科学与技术学院;2. 绵阳市中心医院放射科 |
| |
基金项目: | 四川省自然科学基金项目(2022NSFSC0940,2022NSFSC0894);;西南科技大学博士基金项目(19zx7143,20zx7137); |
| |
摘 要: | 目前基于深度学习的医学图像分割方法往往需要大量带标记数据训练网络模型,然而医学影像的标记数据获取通常非常昂贵,半监督学习能使模型利用大量未标记数据和少量标记数据学习。该文提出了一种基于跨任务一致性的半监督学习框架来降低神经网络模型训练时需要的标记数据成本。该方法利用V-Net网络作为主干框架并添加两个辅助解码器,同时在解码器中引入一个辅助回归任务,提高模型分割性能,并在主副解码器的分割任务和回归任务之间施加正则化约束的跨任务一致性损失,该框架能够学习到大量未标记数据的几何先验信息。在LiTS2017 Challenges数据集上验证了该方法的有效性。在使用20%标记数据的实验中,该方法的Dice系数和Jaccard指数分别达到了93.95%和88.87%,相比全监督V-Net网络模型训练下的Dice系数和Jaccard系数分别提高了3.60百分点和5.78百分点。实验结果表明,该方法在使用少量带标记数据情况下达到接近100%带标记数据训练分割肝脏的精度,与其他的半监督方法相比分割精度更优。
|
关 键 词: | 医学影像 半监督学习 神经网络 回归任务 一致性损失 |
|
|