Aug, 2021

基于知识蒸馏的高效医学图像分割

TL;DR我们提出了一种高效的网络结构,通过从训练充分的医学图像分割网络中提炼知识来训练另一个轻量级网络,从而使得轻量级网络在保留其运行效率的同时,显著提高了其分割能力。通过使用我们为医学图像分割量身定制的新颖蒸馏模式,从教师网络向学生网络传递语义区域信息,避免了处理医学图像时遇到的模糊边界问题。在我们的实验中,轻量级网络的性能提高了 32.6%,同时在推理阶段保持了可移植性,并在 LiTS17 和 KiTS19 两个公认的公共 CT 数据集上进行了验证。