用大模型自动标注波兰语医学文本,训练出小型高效分类器。
ADMEDTAGGER: an annotation framework for distillation of expert knowledge for the Polish medical language
- 用多语言大模型自动生成波兰医学文本标签
- 小型分类器在所有类别上F1超0.80,3类超0.93
- 模型体积小500倍,推理快数百倍,适合部署
本文提出一种标注框架,利用预训练于大规模语料的多语言Llama3.1模型作为教师模型,对波兰语医学文本进行知识蒸馏。该工作是更大项目ADMEDVOICE的一部分,收集了涵盖放射科、肿瘤学、心脏病学、高血压和病理学五个临床类别的大量医学文本。由于人工标注资源有限,我们使用Llama3.1自动标注全文本,并仅对部分标签进行人工验证,构建测试集。基于这些标注数据,训练了三种基于BERT架构的分类器:压缩版DistilBERT、在医学数据上微调的BioBERT,以及在波兰语语料上微调的HerBERT。实验结果表明,其中DistilBERT表现最佳,在所有临床类别上F1分数均超过0.80,三个类别超过0.93。所获模型体积仅为大型语言模型的约1/500,显存占用降低约1/300,推理速度提升数百倍,为医疗文本分类提供轻量高效替代方案。
原文摘要 · Abstract (English)
In this work, we present an annotation framework that demonstrates how a multilingual LLM pretrained on a large corpus can be used as a teacher model to distill the expert knowledge needed for tagging medical texts in Polish. This work is part of a larger project called ADMEDVOICE, within which we collected an extensive corpus of medical texts representing five clinical categories - Radiology, Oncology, Cardiology, Hypertension, and Pathology. Using this data, we had to develop a multi-class classifier, but the fundamental problem turned out to be the lack of resources for annotating an adequate number of texts. Therefore, in our solution, we used the multilingual Llama3.1 model to annotate an extensive corpus of medical texts in Polish. Using our limited annotation resources, we verified only a portion of these labels, creating a test set from them. The data annotated in this way were then used for training and validation of 3 different types of classifiers based on the BERT architecture - the distilled DistilBERT model, BioBERT fine-tuned on medical data, and HerBERT fine-tuned on the Polish language corpus. Among the models we trained, the DistilBERT model achieved the best results, reaching an F1 score > 0.80 for each clinical category and an F1 score > 0.93 for 3 of them. In this way, we obtained a series of highly effective classifiers that represent an alternative to large language models, due to their nearly 500 times smaller size, 300 times lower GPU VRAM consumption, and several hundred times faster inference.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。