We present a new text-to-SQL dataset for electronic health records (EHRs). The utterances were collected from 222 hospital staff, including physicians, nurses, insurance review and health records teams, and more. To construct the QA dataset on structured EHR data, we conducted a poll at a university hospital and templatized the responses to create seed questions. Then, we manually linked them to two open-source EHR databases, MIMIC-III and eICU, and included them with various time expressions and held-out unanswerable questions in the dataset, which were all collected from the poll. Our dataset poses a unique set of challenges: the model needs to 1) generate SQL queries that reflect a wide range of needs in the hospital, including simple retrieval and complex operations such as calculating survival rate, 2) understand various time expressions to answer time-sensitive questions in healthcare, and 3) distinguish whether a given question is answerable or unanswerable based on the prediction confidence. We believe our dataset, EHRSQL, could serve as a practical benchmark to develop and assess QA models on structured EHR data and take one step further towards bridging the gap between text-to-SQL research and its real-life deployment in healthcare. EHRSQL is available at https://github.com/glee4810/EHRSQL.
翻译:我们提出了一个面向电子健康记录(EHR)的新型文本到SQL数据集。话语数据来自222名医院工作人员,包括医师、护士、保险审核与健康记录团队等。为了构建基于结构化EHR数据的问答数据集,我们在大学医院开展问卷调查,将回答模板化以生成种子问题。随后,我们手动将这些问题关联至两个开源EHR数据库(MIMIC-III与eICU),并在数据集中纳入不同时间表达式及来自问卷的不可回答的留存问题。本数据集提出了一系列独特挑战:模型需要1)生成反映医院多样化需求的SQL查询语句,包括简单检索及计算存活率等复杂操作;2)理解多种时间表达式以回答医疗保健领域中的时序敏感性问题;3)基于预测置信度区分给定问题是否可回答。我们相信,EHRSQL数据集可作为开发和评估结构化EHR数据问答模型的实用基准,向弥合文本到SQL研究与医疗实际部署之间的鸿沟迈出关键一步。EHRSQL数据集可通过https://github.com/glee4810/EHRSQL获取。