arXiv:2409.00204eess.IVcs.CV2024-09中稿 · BIBM 2024 Oral被引 12

用生成对抗蒸馏提升颈椎间盘突出检测效率与抗噪能力

MedDet: Generative Adversarial Distillation for Efficient Cervical Disc Herniation Detection

  • 多教师单学生蒸馏结合对抗训练,压缩模型并增强性能
  • 在CDH-1848数据集上实现mAP提升5%,推理速度超5倍加快
  • 适合追求高效、抗噪的医疗影像实时检测系统开发者

颈椎间盘突出(CDH)是常见骨骼肌肉疾病,严重危害健康,需专家进行耗时分析。尽管医学影像自动检测技术不断进步,但两大挑战限制了其实际应用:一是计算复杂度高,难以实现实时处理;二是MRI噪声干扰特征提取,降低现有方法有效性。为此,本文提出MedDet,采用多教师单学生知识蒸馏实现模型压缩与效率提升,同时引入生成对抗训练增强性能。此外,定制二阶nmODE以增强模型对MRI噪声的鲁棒性。在CDH-1848数据集上,相比先前方法,mAP最高提升5%。推理速度提升超过5倍,参数量减少约67.8%,浮点运算量减少36.9%。这些改进显著提升了自动化CDH检测的性能与效率,展现出临床应用的广阔前景。

原文摘要 · Abstract (English)

Cervical disc herniation (CDH) is a prevalent musculoskeletal disorder that significantly impacts health and requires labor-intensive analysis from experts. Despite advancements in automated detection of medical imaging, two significant challenges hinder the real-world application of these methods. First, the computational complexity and resource demands present a significant gap for real-time application. Second, noise in MRI reduces the effectiveness of existing methods by distorting feature extraction. To address these challenges, we propose three key contributions: Firstly, we introduced MedDet, which leverages the multi-teacher single-student knowledge distillation for model compression and efficiency, meanwhile integrating generative adversarial training to enhance performance. Additionally, we customize the second-order nmODE to improve the model's resistance to noise in MRI. Lastly, we conducted comprehensive experiments on the CDH-1848 dataset, achieving up to a 5% improvement in mAP compared to previous methods. Our approach also delivers over 5 times faster inference speed, with approximately 67.8% reduction in parameters and 36.9% reduction in FLOPs compared to the teacher model. These advancements significantly enhance the performance and efficiency of automated CDH detection, demonstrating promising potential for future application in clinical practice. See project website https://steve-zeyu-zhang.github.io/MedDet

医学影像知识蒸馏生成对抗颈椎病检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。