用少量标注数据训练乳腺癌检测模型,显著提升性能与泛化能力
SelectiveKD: A semi-supervised framework for cancer detection in DBT through Knowledge Distillation and Pseudo-labeling
- 通过知识蒸馏和伪标签选择性利用未标注切片
- 在超1万例真实数据上实现AUC显著提升
- 适合标注成本高的医学影像建模场景
在构建数字乳腺断层扫描(DBT)计算机辅助检测(CAD)系统时,该模态的三维特性带来了大规模精确标注的严重挑战。由于获取DBT标注成本高昂,如何有效扩充训练数据仍是开放问题。本文提出SelectiveKD,一种半监督学习框架,仅需少量标注切片即可达到高性能。该方法利用知识蒸馏,使教师模型为整个DBT体积中的所有切片提供监督信号;同时引入选择性数据扩展策略,基于伪标签缓解劣质教师带来的噪声。我们在来自多个设备厂商和地点的超过10,000例真实世界DBT检查数据集上评估该方法。结果表明,SelectiveKD能有效利用未标注切片,显著提升癌症分类性能(AUC)和跨域泛化能力。
原文摘要 · Abstract (English)
When developing Computer Aided Detection (CAD) systems for Digital Breast Tomosynthesis (DBT), the complexity arising from the volumetric nature of the modality poses significant technical challenges for obtaining large-scale accurate annotations. Without access to large-scale annotations, the resulting model may not generalize to different domains. Given the costly nature of obtaining DBT annotations, how to effectively increase the amount of data used for training DBT CAD systems remains an open challenge. In this paper, we present SelectiveKD, a semi-supervised learning framework for building cancer detection models for DBT, which only requires a limited number of annotated slices to reach high performance. We achieve this by utilizing unlabeled slices available in a DBT stack through a knowledge distillation framework in which the teacher model provides a supervisory signal to the student model for all slices in the DBT volume. Our framework mitigates the potential noise in the supervisory signal from a sub-optimal teacher by implementing a selective dataset expansion strategy using pseudo labels. We evaluate our approach with a large-scale real-world dataset of over 10,000 DBT exams collected from multiple device manufacturers and locations. The resulting SelectiveKD process effectively utilizes unannotated slices from a DBT stack, leading to significantly improved cancer classification performance (AUC) and generalization performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。