arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04024eess.IVcs.CV
收藏

利用隐私保护大语言模型和多类型标注增强胸部X射线数据集:一种用于改进分类的数据驱动方法

Enhancing chest X-ray datasets with privacy-preserving large language models and multi-type annotations: a data-driven approach for improved classification

Imaging Biomarkers and Computer-Aided Diagnosis Laboratory, Department of Radiology and Imaging Sciences, National Institutes of Health Clinical Center(影像生物标志物与计算机辅助诊断实验室,放射学与成像科学系,国家卫生研究院临床中心)

Ricardo Bigolin Lanfredi, Pritam Mukherjee, Ronald Summers

AI总结:

提出基于本地可执行大语言模型的MAPLEZ方法,从胸部X射线报告中提取包含位置、严重程度等多类型标注,显著提升标签质量和下游分类模型性能。

摘要:

在胸部X射线(CXR)图像分析中,通常采用基于规则的系统从报告中提取标签以用于数据集发布。然而,标签质量仍有提升空间。这些标注器通常仅输出存在性标签,有时带有二元不确定性指标,这限制了它们的实用性。尽管也开发了有监督深度学习模型用于报告标注,但它们与基于规则的系统类似,缺乏适应性。在本研究中,我们提出了MAPLEZ(使用快速零样本回答的隐私保护大语言模型医疗报告标注),这是一种利用本地可执行大语言模型(LLM)提取并增强CXR报告上发现标签的新方法。MAPLEZ不仅提取指示发现存在与否的二元标签,还提取位置、严重程度以及放射科医生对该发现的不确定性。在来自五个测试集的八种异常上,我们表明,与竞争标注器相比,我们的方法在分类存在性标注的宏观F1分数上提高了3.6个百分点(pp),在位置标注的F1分数上提高了20 pp以上。此外,在分类监督中结合使用改进的标注和多类型标注,我们证明了模型质量取得了实质性进展,与使用最佳替代方法标注训练的模型相比,AUROC提高了1.1 pp。我们共享了代码和标注。

英文摘要:

In chest X-ray (CXR) image analysis, rule-based systems are usually employed to extract labels from reports for dataset releases. However, there is still room for improvement in label quality. These labelers typically output only presence labels, sometimes with binary uncertainty indicators, which limits their usefulness. Supervised deep learning models have also been developed for report labeling but lack adaptability, similar to rule-based systems. In this work, we present MAPLEZ (Medical report Annotations with Privacy-preserving Large language model using Expeditious Zero shot answers), a novel approach leveraging a locally executable Large Language Model (LLM) to extract and enhance findings labels on CXR reports. MAPLEZ extracts not only binary labels indicating the presence or absence of a finding but also the location, severity, and radiologists' uncertainty about the finding. Over eight abnormalities from five test sets, we show that our method can extract these annotations with an increase of 3.6 percentage points (pp) in macro F1 score for categorical presence annotations and more than 20 pp increase in F1 score for the location annotations over competing labelers. Additionally, using the combination of improved annotations and multi-type annotations in classification supervision, we demonstrate substantial advancements in model quality, with an increase of 1.1 pp in AUROC over models trained with annotations from the best alternative approach. We share code and annotations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑