用改进的BERT模型从脑部影像报告中自动识别损伤,准确率高达98%。
Neu-RadBERT for Enhanced Diagnosis of Brain Injuries and Conditions
- 基于BERT架构,结合掩码语言建模和数据过采样优化
- 对急性脑损伤诊断准确率达98.0%,显著优于基线模型
- 适合医疗AI研发与临床辅助诊断系统开发者参考
目标:开发一种分类算法,从接受有创机械通气的急性呼吸衰竭(ARF)患者脑部影像的自由文本放射科报告中提取诊断信息。方法:我们构建并微调了Neu-RadBERT,一种基于BERT的模型,用于分类非结构化放射科报告。从MIMIC-IV数据库中提取所有因ARF进行的脑部影像检查(计算机断层扫描和磁共振成像)报告,对部分报告进行初始人工标注,随后采用三种策略微调Neu-RadBERT:1)基线RadBERT;2)带掩码语言建模(MLM)预训练的Neu-RadBERT;3)带MLM预训练和过采样以缓解数据偏斜的Neu-RadBERT。将该模型性能与Llama-2-13B(自回归大语言模型)对比。结果:经过采样优化的Neu-RadBERT在脑部异常分类上表现显著优于基线RadBERT,急性脑损伤诊断准确率最高达98.0%。而Llama-2-13B性能相对较低,二分类准确率峰值为67.5%。这表明当前自回归大模型在特定分类任务中存在局限性,尽管更大模型或进一步微调可能提升效果。结论:通过目标领域预训练与过采样技术增强的Neu-RadBERT,可成为从放射科报告中精准、可靠诊断神经疾病的重要工具。本研究凸显了基于Transformer的自然语言处理模型在自动解析自由文本报告方面的潜力,具有广泛的研究与临床应用前景。
原文摘要 · Abstract (English)
Objective: We sought to develop a classification algorithm to extract diagnoses from free-text radiology reports of brain imaging performed in patients with acute respiratory failure (ARF) undergoing invasive mechanical ventilation. Methods: We developed and fine-tuned Neu-RadBERT, a BERT-based model, to classify unstructured radiology reports. We extracted all the brain imaging reports (computed tomography and magnetic resonance imaging) from MIMIC-IV database, performed in patients with ARF. Initial manual labelling was performed on a subset of reports for various brain abnormalities, followed by fine-tuning Neu-RadBERT using three strategies: 1) baseline RadBERT, 2) Neu-RadBERT with Masked Language Modeling (MLM) pretraining, and 3) Neu-RadBERT with MLM pretraining and oversampling to address data skewness. We compared the performance of this model to Llama-2-13B, an autoregressive LLM. Results: The Neu-RadBERT model, particularly with oversampling, demonstrated significant improvements in diagnostic accuracy compared to baseline RadBERT for brain abnormalities, achieving up to 98.0% accuracy for acute brain injuries. Llama-2-13B exhibited relatively lower performance, peaking at 67.5% binary classification accuracy. This result highlights potential limitations of current autoregressive LLMs for this specific classification task, though it remains possible that larger models or further fine-tuning could improve performance. Conclusion: Neu-RadBERT, enhanced through target domain pretraining and oversampling techniques, offered a robust tool for accurate and reliable diagnosis of neurological conditions from radiology reports. This study underscores the potential of transformer-based NLP models in automatically extracting diagnoses from free text reports with potential applications to both research and patient care.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。