arXiv:2501.00031cs.CL2025-01被引 6

用大模型教小模型提取病历信息,又快又省

Distilling Large Language Models for Efficient Clinical Information Extraction

  • 用Gemini等大模型做标签,训练出千分之一大小的精简BERT
  • 小模型在疾病/药物抽取上接近人工标注模型,症状略低
  • 推理速度提升12倍,成本降低101倍,适合临床部署

大型语言模型(LLM)在临床信息抽取中表现优异,但计算开销大,难以实际应用。知识蒸馏——将大模型知识迁移到小模型——提供了解决方案。本文评估了约比现代LLM小1000倍的蒸馏BERT模型在临床命名实体识别(NER)任务中的表现。采用Gemini和OpenAI模型及医学本体(RxNorm、SNOMED)作为教师标签生成器,用于药物、疾病和症状抽取。在覆盖五个公开数据集的3300余份临床笔记上进行测试,对比蒸馏BERT与教师模型及人工标注微调的BioBERT。外部验证使用MedAlign数据集。疾病抽取F1:教师模型0.82,人类标注微调模型0.89,蒸馏模型0.84;药物抽取F1:教师模型0.84,人类标注模型0.91,蒸馏模型0.87;症状抽取F1:教师模型0.73,蒸馏模型0.68。蒸馏模型推理速度分别比GPT-4o、o1-mini、Gemini Flash快12倍、4倍、8倍,成本更低(分别为85倍、101倍、2倍)。外部验证中,蒸馏模型取得药物F1 0.883,疾病F1 0.726,症状F1 0.699。蒸馏模型相比顶尖大模型,在保持相似性能的同时,实现最高101倍的成本节约和12倍的速度提升,为临床信息抽取提供了高效可扩展的替代方案。

原文摘要 · Abstract (English)

Large language models (LLMs) excel at clinical information extraction but their computational demands limit practical deployment. Knowledge distillation--the process of transferring knowledge from larger to smaller models--offers a potential solution. We evaluate the performance of distilled BERT models, which are approximately 1,000 times smaller than modern LLMs, for clinical named entity recognition (NER) tasks. We leveraged state-of-the-art LLMs (Gemini and OpenAI models) and medical ontologies (RxNorm and SNOMED) as teacher labelers for medication, disease, and symptom extraction. We applied our approach to over 3,300 clinical notes spanning five publicly available datasets, comparing distilled BERT models against both their teacher labelers and BERT models fine-tuned on human labels. External validation was conducted using clinical notes from the MedAlign dataset. For disease extraction, F1 scores were 0.82 (teacher model), 0.89 (BioBERT trained on human labels), and 0.84 (BioBERT-distilled). For medication, F1 scores were 0.84 (teacher model), 0.91 (BioBERT-human), and 0.87 (BioBERT-distilled). For symptoms: F1 score of 0.73 (teacher model) and 0.68 (BioBERT-distilled). Distilled BERT models had faster inference (12x, 4x, 8x faster than GPT-4o, o1-mini, and Gemini Flash respectively) and lower costs (85x, 101x, 2x cheaper than GPT-4o, o1-mini, and Gemini Flash respectively). On the external validation dataset, the distilled BERT model achieved F1 scores of 0.883 (medication), 0.726 (disease), and 0.699 (symptom). Distilled BERT models were up to 101x cheaper and 12x faster than state-of-the-art LLMs while achieving similar performance on NER tasks. Distillation offers a computationally efficient and scalable alternative to large LLMs for clinical information extraction.

大模型蒸馏临床信息抽取轻量化模型医疗文本

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