arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.20719cs.LGcs.AI

通过层次与语义引导的 Transformer 生成临床真实的 EHR 数据

Generating Clinically Realistic EHR Data via a Hierarchy- and Semantics-Guided Transformer

University of Queensland(昆士兰大学)

Guanglin Zhou, Sebastiano Barbieri

AI总结:

本文提出 HiSGT 框架,融合临床代码层次图嵌入与 ClinicalBERT 语义嵌入以生成高保真 EHR,并在 MIMIC-III/IV 上验证其统计一致性和下游任务效用。

摘要:

生成逼真的合成电子健康记录(EHR)在加速医疗研究、促进 AI 模型开发和增强患者隐私方面具有巨大前景。然而,现有生成方法通常将 EHR 视为离散医疗代码的扁平序列。这种做法忽视了两个关键方面:临床编码系统固有的层次组织结构,以及代码描述所提供的丰富语义上下文。因此,合成患者序列往往缺乏较高的临床保真度,并且在下游临床任务中的效用有限。本文提出层次与语义引导的 Transformer(Hierarchy- and Semantics-Guided Transformer,HiSGT),这是一种在生成过程中同时利用层次信息和语义信息的新颖框架。HiSGT 构建层次图以编码临床代码之间的父子关系和兄弟关系,并采用图神经网络获得具有层次感知能力的嵌入。随后,这些嵌入与从预训练临床语言模型(例如 ClinicalBERT)中提取的语义嵌入相融合,使基于 Transformer 的生成器能够更准确地建模真实 EHR 中固有的细微临床模式。在 MIMIC-III 和 MIMIC-IV 数据集上的大量实验表明,HiSGT 显著提升了合成数据与真实患者记录之间的统计一致性,并支持慢性病分类等稳健的下游应用。通过克服传统基于原始代码的生成模型的局限,HiSGT 代表了迈向临床高保真合成数据生成的重要一步,也提供了一种适用于可解释医疗代码表示的通用范式,并在数据增强和隐私保护型医疗分析中具有宝贵应用价值。

英文摘要:

Generating realistic synthetic electronic health records (EHRs) holds tremendous promise for accelerating healthcare research, facilitating AI model development and enhancing patient privacy. However, existing generative methods typically treat EHRs as flat sequences of discrete medical codes. This approach overlooks two critical aspects: the inherent hierarchical organization of clinical coding systems and the rich semantic context provided by code descriptions. Consequently, synthetic patient sequences often lack high clinical fidelity and have limited utility in downstream clinical tasks. In this paper, we propose the Hierarchy- and Semantics-Guided Transformer (HiSGT), a novel framework that leverages both hierarchical and semantic information for the generative process. HiSGT constructs a hierarchical graph to encode parent-child and sibling relationships among clinical codes and employs a graph neural network to derive hierarchy-aware embeddings. These are then fused with semantic embeddings extracted from a pre-trained clinical language model (e.g., ClinicalBERT), enabling the Transformer-based generator to more accurately model the nuanced clinical patterns inherent in real EHRs. Extensive experiments on the MIMIC-III and MIMIC-IV datasets demonstrate that HiSGT significantly improves the statistical alignment of synthetic data with real patient records, as well as supports robust downstream applications such as chronic disease classification. By addressing the limitations of conventional raw code-based generative models, HiSGT represents a significant step toward clinically high-fidelity synthetic data generation and a general paradigm suitable for interpretable medical code representation, offering valuable applications in data augmentation and privacy-preserving healthcare analytics.

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

↑