<p>Recent advances in medical image segmentation using deep learning have developed dental technology. However, many dental clinics lack the minimum hardware infrastructure required to run high-performance deep learning algorithms in real-time, hindering their commercialization. Although filter pruning and knowledge distillation hold promise as resource-efficient solutions, they remain underexplored in medical/dental image segmentation. This paper presents a novel framework for compressing dental image segmentation networks. First, we efficiently explored sub-networks by removing unnecessary filters through a learnable differentiable gate. Second, during sub-network exploration, we further utilized the baseline network’s information through masked knowledge distillation. Through this approach, the proposed compression framework efficiently explores a more appropriate sub-network with minimal loss of the baseline network’s information for each structure. As a result, the proposed method was tested on dental anatomical structure segmentation on 3D CBCT and teeth segmentation on panoramic radiographs, achieving reductions in MACs by 90% and 95%, respectively, while showing only around a 1% decrease in performance based on the Dice score. The ability to achieve up to 95% MAC reduction with minimal Dice degradation highlights its potential for real-time deployment in resource-limited dental clinics, paving the way for practical clinical adoption.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Structure-aware efficient compression for dental image segmentation using differentiable gates and masked knowledge distillation

  • Dongjun Lee,
  • Jae Hwan Han,
  • Tae-Hoon Yong,
  • Soon Hyoung Pyo,
  • Byung Cheol Song

摘要

Recent advances in medical image segmentation using deep learning have developed dental technology. However, many dental clinics lack the minimum hardware infrastructure required to run high-performance deep learning algorithms in real-time, hindering their commercialization. Although filter pruning and knowledge distillation hold promise as resource-efficient solutions, they remain underexplored in medical/dental image segmentation. This paper presents a novel framework for compressing dental image segmentation networks. First, we efficiently explored sub-networks by removing unnecessary filters through a learnable differentiable gate. Second, during sub-network exploration, we further utilized the baseline network’s information through masked knowledge distillation. Through this approach, the proposed compression framework efficiently explores a more appropriate sub-network with minimal loss of the baseline network’s information for each structure. As a result, the proposed method was tested on dental anatomical structure segmentation on 3D CBCT and teeth segmentation on panoramic radiographs, achieving reductions in MACs by 90% and 95%, respectively, while showing only around a 1% decrease in performance based on the Dice score. The ability to achieve up to 95% MAC reduction with minimal Dice degradation highlights its potential for real-time deployment in resource-limited dental clinics, paving the way for practical clinical adoption.