用公开医学目录训练小模型,提升德语肿瘤诊断编码准确率
Unlocking Public Catalogues: Instruction-Tuning LLMs for ICD Coding of German Tumor Diagnoses
- 基于公开医学词典构建问答数据集,指导小模型学习德语肿瘤编码
- 编码准确率从不足25%提升至58%,错误代码输出归零
- 适合医疗信息化、隐私敏感场景下需本地部署的AI编码系统
德国癌症结构化记录依赖ICD-10-GM和ICD-O-3编码。小型开源大模型虽利于隐私保护,但在德语肿瘤诊断文本上常表现不佳。本研究通过公共医学目录构建超50万条问答对,对来自Qwen、Llama、Mistral系列的8个7-70B参数模型进行指令微调。测试结果显示,ICD-10-GM精确编码准确率从1.4%-24%升至41%-58%,部分编码(三位码)准确率达73%-83%;ICD-O-3解剖部位编码精确率提升至22%-40%,部分编码达56%-67%。所有模型的错误代码输出降至0%,肿瘤诊断识别率达99%。模型规模越大性能越好,但微调后大小模型差距缩小。使用Qwen3推理模式性能低于微调且速度慢100倍以上。研究证实利用公开词典构建指令数据可有效提升小模型在医疗文档任务中的表现。完整数据集及最优模型权重已发布于HuggingFace。
原文摘要 · Abstract (English)
Accurate coding of tumor diagnoses with ICD-10-GM and ICD-O-3 is essential for structured cancer documentation in Germany. Smaller open-weight LLMs are appealing for privacy-preserving automation but often struggle with coding accuracy in German-language contexts. This study investigates whether instruction-based fine-tuning on public datasets improves the coding accuracy of open-weight LLMs for German tumor diagnosis texts. The evaluation uses coded diagnoses from the local tumor documentation system as test data. In a systematic data quality assessment, the upper limit for ICD-10 coding performance was estimated at 60-79% for exact and 81-94% for partial (three-character codes only) derivation. As training data, over 500,000 question-answer pairs were created based on the ICD-10-GM, ICD-O-3, and OPS catalogues. Eight open-weight models from the Qwen, Llama, and Mistral families (7-70 B parameters) were fine-tuned. ICD-10-GM accuracy rose from 1.4-24% to 41-58%, and partial accuracy from 31-74% to 73-83%. The accuracy of ICD-O-3 topography coding also improved but started and remained considerably lower with an exact accuracy of 22-40% and a partial accuracy of 56-67% after fine-tuning. Malformed code outputs dropped to 0% for all models. Tumor-diagnosis recognition reached 99%. Accuracy correlated positively with model size, but gaps between small and large models narrowed after fine-tuning. The reasoning mode in Qwen3 generally yielded a lower performance than fine-tuning and was over 100 times slower. Our findings highlight the potential of leveraging public catalogues to build instruction datasets that improve LLMs in medical documentation tasks. The complete training dataset and the best-performing checkpoints of the fine-tuned models are available from https://huggingface.co/datasets/stefan-m-lenz/ICDOPS-QA-2024.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。