微调长文本模型提升临床文本分类准确率
Enhancing Clinical Text Classification via Fine-Tuned DRAGON Longformer Models
- 调整序列长度、学习率和训练轮数,结合医学术语优化
- 分类准确率从72.0%提升至85.2%,F1达85.2%
- 适合医疗NLP研究者与临床辅助系统开发者
本研究针对临床文本分类任务,优化了基于DRAGON Longformer基础模型的性能,聚焦于医疗病例描述的二分类。使用包含500例结构化医学观察的临床数据集,其中400例用于训练,100例用于验证。对预训练模型joeranbosma/dragon-longformer-base-mixed-domain进行了超参数调优、领域特定预处理及架构调整,关键修改包括将序列长度从512增至1024个标记,学习率从1e-05调至5e-06,训练轮数从5增至8,并引入专业医学术语。优化后模型性能显著提升:准确率由72.0%增至85.2%,精确率从68.0%升至84.1%,召回率从75.0%增至86.3%,F1分数从71.0%提高到85.2%。统计分析显示改进具有高度显著性(p < .001)。模型在解析医学术语、解剖测量和临床观察方面表现更佳。研究成果为领域专用语言模型提供支持,并具实际临床自然语言处理应用价值。
原文摘要 · Abstract (English)
This study explores the optimization of the DRAGON Longformer base model for clinical text classification, specifically targeting the binary classification of medical case descriptions. A dataset of 500 clinical cases containing structured medical observations was used, with 400 cases for training and 100 for validation. Enhancements to the pre-trained joeranbosma/dragon-longformer-base-mixed-domain model included hyperparameter tuning, domain-specific preprocessing, and architectural adjustments. Key modifications involved increasing sequence length from 512 to 1024 tokens, adjusting learning rates from 1e-05 to 5e-06, extending training epochs from 5 to 8, and incorporating specialized medical terminology. The optimized model achieved notable performance gains: accuracy improved from 72.0% to 85.2%, precision from 68.0% to 84.1%, recall from 75.0% to 86.3%, and F1-score from 71.0% to 85.2%. Statistical analysis confirmed the significance of these improvements (p < .001). The model demonstrated enhanced capability in interpreting medical terminology, anatomical measurements, and clinical observations. These findings contribute to domain-specific language model research and offer practical implications for clinical natural language processing applications. The optimized model's strong performance across diverse medical conditions underscores its potential for broad use in healthcare settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。