针对医疗代码表示难题,提出新架构提升大模型对结构化病历数据的理解能力。
Representation Learning of Structured Data for Medical Foundation Models

- 设计专用分词方法,适配医疗编码的结构特点。
- 在百亿级数据上训练,性能提升最高达23%。
- 适合需要融合文本与编码数据的医疗AI研究者。
大型语言模型在多个领域表现出色,但在处理如ICD-10或SNOMED-CT等非文本型医疗代码时表现受限,暴露于近期研究中。本文分析了当前分词方法在处理医疗代码时的不足,提出UniStruct架构,通过专门适配结构化医疗代码的子词分词技术,构建兼顾非结构化文本与结构化数据的多模态医疗基础模型。模型在内部超大规模医疗数据库(超过10亿个标记)上预训练,并在公开的结构化医疗记录库中验证。结果显示,该模型在评估指标上最高提升23%,其中约2%来自所提分词方法;在仅使用1/1000预训练数据的情况下,于EHRSHOT基准测试中超过42%的下游任务表现更优。该方法不仅增强了患者为中心模型的表征与泛化能力,也填补了现有模型在复杂结构化医疗数据处理上的关键空白。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have demonstrated remarkable performance across various domains, including healthcare. However, their ability to effectively represent structured non-textual data, such as the alphanumeric medical codes used in records like ICD-10 or SNOMED-CT, is limited and has been particularly exposed in recent research. This paper examines the challenges LLMs face in processing medical codes due to the shortcomings of current tokenization methods. As a result, we introduce the UniStruct architecture to design a multimodal medical foundation model of unstructured text and structured data, which addresses these challenges by adapting subword tokenization techniques specifically for the structured medical codes. Our approach is validated through model pre-training on both an extensive internal medical database and a public repository of structured medical records. Trained on over 1 billion tokens on the internal medical database, the proposed model achieves up to a 23% improvement in evaluation metrics, with around 2% gain attributed to our proposed tokenization. Additionally, when evaluated on the EHRSHOT public benchmark with a 1/1000 fraction of the pre-training data, the UniStruct model improves performance on over 42% of the downstream tasks. Our approach not only enhances the representation and generalization capabilities of patient-centric models but also bridges a critical gap in representation learning models' ability to handle complex structured medical data, alongside unstructured text.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。