Nov, 2023
知识蒸馏中肾脏和肝脏肿瘤分割的中间层设计重新思考
Rethinking Intermediate Layers design in Knowledge Distillation for
Kidney and Liver Tumor Segmentation
TL;DR知识蒸馏在医学成像任务(如肾脏和肝脏肿瘤分割)中的应用面临挑战,为解决这些问题,我们提出了分层选择性反馈蒸馏(HLFD)方法,通过从中间层到较早层的蒸馏,以及将最终层的知识以特征和像素级别转移到中间层,实现模型从早期层学习更高质量的表示,从而生成一个稳健且紧凑的学生模型。大量的定量评估表明,HLFD方法显著优于现有方法,在肾脏分割任务中,与无知识蒸馏的学生模型相比,HLFD提升了10个百分点以上,显著改善了对肿瘤特定特征的关注。从质量上看,使用HLFD训练的学生模型能够抑制不相关信息,聚焦于肿瘤特定细节,为更高效准确的诊断工具开辟了新路径。