Retrieval augmented text-to-SQL generation for epidemiological question answering using electronic health records

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ziletti, Angelo, D'Ambrosi, Leonardo
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909450170793984
author Ziletti, Angelo
D'Ambrosi, Leonardo
author_facet Ziletti, Angelo
D'Ambrosi, Leonardo
contents Electronic health records (EHR) and claims data are rich sources of real-world data that reflect patient health status and healthcare utilization. Querying these databases to answer epidemiological questions is challenging due to the intricacy of medical terminology and the need for complex SQL queries. Here, we introduce an end-to-end methodology that combines text-to-SQL generation with retrieval augmented generation (RAG) to answer epidemiological questions using EHR and claims data. We show that our approach, which integrates a medical coding step into the text-to-SQL process, significantly improves the performance over simple prompting. Our findings indicate that although current language models are not yet sufficiently accurate for unsupervised use, RAG offers a promising direction for improving their capabilities, as shown in a realistic industry setting.
format Preprint
id arxiv_https___arxiv_org_abs_2403_09226
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Retrieval augmented text-to-SQL generation for epidemiological question answering using electronic health records
Ziletti, Angelo
D'Ambrosi, Leonardo
Computation and Language
Electronic health records (EHR) and claims data are rich sources of real-world data that reflect patient health status and healthcare utilization. Querying these databases to answer epidemiological questions is challenging due to the intricacy of medical terminology and the need for complex SQL queries. Here, we introduce an end-to-end methodology that combines text-to-SQL generation with retrieval augmented generation (RAG) to answer epidemiological questions using EHR and claims data. We show that our approach, which integrates a medical coding step into the text-to-SQL process, significantly improves the performance over simple prompting. Our findings indicate that although current language models are not yet sufficiently accurate for unsupervised use, RAG offers a promising direction for improving their capabilities, as shown in a realistic industry setting.
title Retrieval augmented text-to-SQL generation for epidemiological question answering using electronic health records
topic Computation and Language
url https://arxiv.org/abs/2403.09226