RAM-EHR: Retrieval Augmentation Meets Clinical Predictions on Electronic Health Records
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913448044003328 |
|---|---|
| author | Xu, Ran Shi, Wenqi Yu, Yue Zhuang, Yuchen Jin, Bowen Wang, May D. Ho, Joyce C. Yang, Carl |
| author_facet | Xu, Ran Shi, Wenqi Yu, Yue Zhuang, Yuchen Jin, Bowen Wang, May D. Ho, Joyce C. Yang, Carl |
| contents | We present RAM-EHR, a Retrieval AugMentation pipeline to improve clinical predictions on Electronic Health Records (EHRs). RAM-EHR first collects multiple knowledge sources, converts them into text format, and uses dense retrieval to obtain information related to medical concepts. This strategy addresses the difficulties associated with complex names for the concepts. RAM-EHR then augments the local EHR predictive model co-trained with consistency regularization to capture complementary information from patient visits and summarized knowledge. Experiments on two EHR datasets show the efficacy of RAM-EHR over previous knowledge-enhanced baselines (3.4% gain in AUROC and 7.2% gain in AUPR), emphasizing the effectiveness of the summarized knowledge from RAM-EHR for clinical prediction tasks. The code will be published at \url{https://github.com/ritaranx/RAM-EHR}. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2403_00815 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | RAM-EHR: Retrieval Augmentation Meets Clinical Predictions on Electronic Health Records Xu, Ran Shi, Wenqi Yu, Yue Zhuang, Yuchen Jin, Bowen Wang, May D. Ho, Joyce C. Yang, Carl Computation and Language Artificial Intelligence Information Retrieval Other Quantitative Biology We present RAM-EHR, a Retrieval AugMentation pipeline to improve clinical predictions on Electronic Health Records (EHRs). RAM-EHR first collects multiple knowledge sources, converts them into text format, and uses dense retrieval to obtain information related to medical concepts. This strategy addresses the difficulties associated with complex names for the concepts. RAM-EHR then augments the local EHR predictive model co-trained with consistency regularization to capture complementary information from patient visits and summarized knowledge. Experiments on two EHR datasets show the efficacy of RAM-EHR over previous knowledge-enhanced baselines (3.4% gain in AUROC and 7.2% gain in AUPR), emphasizing the effectiveness of the summarized knowledge from RAM-EHR for clinical prediction tasks. The code will be published at \url{https://github.com/ritaranx/RAM-EHR}. |
| title | RAM-EHR: Retrieval Augmentation Meets Clinical Predictions on Electronic Health Records |
| topic | Computation and Language Artificial Intelligence Information Retrieval Other Quantitative Biology |
| url | https://arxiv.org/abs/2403.00815 |