中华超声影像学杂志2024,Vol.33Issue(1) :21-26.DOI:10.3760/cma.j.cn131148-20230801-00045

知识蒸馏算法训练在胎儿心脏超声图像三血管气管切面精细化分割中的应用

Application of knowledge distillation technology for fine segmentation of three-vessel and trachea views in fetal echocardiographic images

狄敏 蔡祈文 马明明 田园诗 陈阳 赵博文 陈冉
中华超声影像学杂志2024,Vol.33Issue(1) :21-26.DOI:10.3760/cma.j.cn131148-20230801-00045

知识蒸馏算法训练在胎儿心脏超声图像三血管气管切面精细化分割中的应用

Application of knowledge distillation technology for fine segmentation of three-vessel and trachea views in fetal echocardiographic images

狄敏 1蔡祈文 2马明明 1田园诗 1陈阳 1赵博文 1陈冉1
扫码查看

作者信息

  • 1. 浙江大学医学院附属邵逸夫医院超声科 浙江省胎儿心脏超声诊断技术指导中心 浙江大学邵逸夫临床医学研究所,杭州 310016
  • 2. 浙江大学数学科学学院,杭州 310058
  • 折叠

摘要

目的 探索基于知识蒸馏算法训练构建的胎儿心脏超声图像分割网络模型在胎儿心脏超声图像三血管气管切面精细化分割中的应用价值。 方法 回顾性收集2016年1月至2021年12月在浙江大学医学院附属邵逸夫医院接受胎儿心脏超声检查的正常中晚孕期胎儿1 300例,分析胎儿心脏超声三血管气管切面二维灰阶超声图像,将其分为训练集、验证集和测试集。应用训练集与验证集构建辅助诊断网络模型,再用测试集对不同网络模型(U-Net、DeepLabv3+)进行测试,由一名有经验的医生收集并注释三血管气管切面作为参考标准。以交并比(IoU)、像素精度(PA)和骰子系数(Dice)为3个定量评估分割精度指标,评估该知识蒸馏算法训练模型的诊断效能。并对本模型及最常用的分割模型进行识别,对结果进行比较。随机选取101张图像,分别交由低年资医生、AI及低年资医生辅助AI判读,绘制Bland-Altman图像评价其分别与参考标准的一致性,并对三者结果进行比较。 结果 知识蒸馏算法训练模型在所有评价指标上均取得了比U-Net、DeepLabv3+模型更好的结果,平均IoU、PA、Dice分别为68.6%、81.4%、81.3%。与U-Net及DeepLabv3+模型相比,本模型获得了更精确的分割边界,并且在定量评价指标上均有提高。经过该模型辅助,低年资医生对于诊断的精确度有所提高。 结论 知识蒸馏算法训练模型分割方法可在胎儿心脏超声图像的三血管气管切面识别胎儿心脏的解剖结构,其识别结果明显优于相关方法,并可提高低年资医生对于其图像识别的准确度。 Objective To explore the application value of fetal heart ultrasound image segmentation network model based on knowledge distillation technology in the fine segmentation of fetal heart ultrasound image at three-vessel and trachea (3VT) views. Methods One thousand and three hundred fetals were retrospectively collected from Sir Run Run Shaw Hospital, Zhejiang University College of Medicine from January 2016 to December 2021, the two-dimensional grayscale ultrasound images of fetal heart at 3VT views were analyzed and then divided into training, validation, and test sets. The training and validation sets were used to construct the auxiliary diagnostic network models, and the test set was used to test the reliability of different network models (U-Net, DeepLabv3+ ). The 3VT views were collected and annotated by an experienced doctor as the reference standard. The intersection over union (IoU), pixel accuracy (PA) and Dice coefficient (Dice) were used as the 3 indexes to evaluate the segmentation accuracy, and the diagnostic efficiency of the training model was evaluated. The training model and the most commonly used segmentation models were identified, and the results were compared. A total of 101 images were randomly selected and assigned to junior doctors, AI and junior doctors assisted AI interpretation. Bland-Altman images were drawn to evaluate their consistency with the reference standard, and the results were compared. Results The training model of knowledge distillation algorithm achieved better results than U-Net, DeepLabv3+ models on all evaluation indexes, and the average IoU, PA and Dice were 68.6%, 81.4% and 81.3%, respectively. Compared with the U-Net model and DeepLabv3+ model, more accurate segmentation boundaries were obtained by the knowledge distillation algorithm training model, and the quantitative evaluation indexes were improved. With the aid of the model, the diagnostic accuracy of junior doctors was improved. Conclusions The knowledge distillation algorithm training model segmentation method can identify the anatomical structure of the fetal heart in the 3VT view of the fetal heart ultrasound image, and the recognition result is obviously better than other related methods, and can improve the accuracy of image recognition for doctors with low experience.

Abstract

Objective To explore the application value of fetal heart ultrasound image segmentation network model based on knowledge distillation technology in the fine segmentation of fetal heart ultrasound image at three-vessel and trachea (3VT) views. Methods One thousand and three hundred fetals were retrospectively collected from Sir Run Run Shaw Hospital, Zhejiang University College of Medicine from January 2016 to December 2021, the two-dimensional grayscale ultrasound images of fetal heart at 3VT views were analyzed and then divided into training, validation, and test sets. The training and validation sets were used to construct the auxiliary diagnostic network models, and the test set was used to test the reliability of different network models (U-Net, DeepLabv3+ ). The 3VT views were collected and annotated by an experienced doctor as the reference standard. The intersection over union (IoU), pixel accuracy (PA) and Dice coefficient (Dice) were used as the 3 indexes to evaluate the segmentation accuracy, and the diagnostic efficiency of the training model was evaluated. The training model and the most commonly used segmentation models were identified, and the results were compared. A total of 101 images were randomly selected and assigned to junior doctors, AI and junior doctors assisted AI interpretation. Bland-Altman images were drawn to evaluate their consistency with the reference standard, and the results were compared. Results The training model of knowledge distillation algorithm achieved better results than U-Net, DeepLabv3+ models on all evaluation indexes, and the average IoU, PA and Dice were 68.6%, 81.4% and 81.3%, respectively. Compared with the U-Net model and DeepLabv3+ model, more accurate segmentation boundaries were obtained by the knowledge distillation algorithm training model, and the quantitative evaluation indexes were improved. With the aid of the model, the diagnostic accuracy of junior doctors was improved. Conclusions The knowledge distillation algorithm training model segmentation method can identify the anatomical structure of the fetal heart in the 3VT view of the fetal heart ultrasound image, and the recognition result is obviously better than other related methods, and can improve the accuracy of image recognition for doctors with low experience.

关键词

人工智能/知识蒸馏算法训练/胎儿/心脏超声图像

Key words

Artificial intelligence/Knowledge distillation technology/Fetus/Cardiac ultrasound images

引用本文复制引用

基金项目

浙江省基础公益研究计划项目(LGF22H180008)

出版年

2024
中华超声影像学杂志
中华医学会

中华超声影像学杂志

CSTPCD北大核心
影响因子:0.986
ISSN:1004-4477
参考文献量22
段落导航相关论文