用知识蒸馏让大模型教小模型,提升电子病历预测效率和准确率。
CKD-EHR:Clinical Knowledge Distillation for Electronic Health Records
- 大模型生成可解释软标签,小模型高效学习
- 诊断准确率提升9%,F1-score提高27%
- 适合临床部署的轻量化方案,加速22.2倍
基于电子健康记录(EHR)的疾病预测模型在推动精准医疗和早期干预方面具有重要临床价值。然而,现有大语言模型存在医学知识表征不足和临床部署效率低的问题。为此,本文提出CKD-EHR(临床知识蒸馏用于EHR)框架,通过知识蒸馏实现高效准确的疾病风险预测。具体地,首先在医学增强数据上微调Qwen2.5-7B作为教师模型,再通过多粒度注意力蒸馏机制生成可解释的软标签,最后将知识迁移到轻量级BERT学生模型。实验结果表明,在MIMIC-III数据集上,相比基线模型,诊断准确率提升9%,F1-score提高27%,推理速度加快22.2倍。该方案显著提升资源利用效率,同时增强诊断的准确性与时效性,为临床场景中的资源优化提供可行技术路径。代码与数据见https://github.com/209506702/CKD_EHR。
原文摘要 · Abstract (English)
Electronic Health Records (EHR)-based disease prediction models have demonstrated significant clinical value in promoting precision medicine and enabling early intervention. However, existing large language models face two major challenges: insufficient representation of medical knowledge and low efficiency in clinical deployment. To address these challenges, this study proposes the CKD-EHR (Clinical Knowledge Distillation for EHR) framework, which achieves efficient and accurate disease risk prediction through knowledge distillation techniques. Specifically, the large language model Qwen2.5-7B is first fine-tuned on medical knowledge-enhanced data to serve as the teacher model.It then generates interpretable soft labels through a multi-granularity attention distillation mechanism. Finally, the distilled knowledge is transferred to a lightweight BERT student model. Experimental results show that on the MIMIC-III dataset, CKD-EHR significantly outperforms the baseline model:diagnostic accuracy is increased by 9%, F1-score is improved by 27%, and a 22.2 times inference speedup is achieved. This innovative solution not only greatly improves resource utilization efficiency but also significantly enhances the accuracy and timeliness of diagnosis, providing a practical technical approach for resource optimization in clinical settings. The code and data for this research are available athttps://github.com/209506702/CKD_EHR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。