Joyce C. Ho
2024
BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrievers
Ran Xu
|
Wenqi Shi
|
Yue Yu
|
Yuchen Zhuang
|
Yanqiao Zhu
|
May Dongmei Wang
|
Joyce C. Ho
|
Chao Zhang
|
Carl Yang
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Developing effective biomedical retrieval models is important for excelling at knowledge-intensive biomedical tasks but still challenging due to the lack of sufficient publicly annotated biomedical data and computational resources. We present BMRetriever, a series of dense retrievers for enhancing biomedical retrieval via unsupervised pre-training on large biomedical corpora, followed by instruction fine-tuning on a combination of labeled datasets and synthetic pairs. Experiments on 5 biomedical tasks across 11 datasets verify BMRetriever’s efficacy on various biomedical applications. BMRetriever also exhibits strong parameter efficiency, with the 410M variant outperforming baselines up to 11.7 times larger, and the 2B variant matching the performance of models with over 5B parameters. The training data and model checkpoints are released at https://huggingface.co/BMRetriever to ensure transparency, reproducibility, and application to new domains.
EHRAgent: Code Empowers Large Language Models for Few-shot Complex Tabular Reasoning on Electronic Health Records
Wenqi Shi
|
Ran Xu
|
Yuchen Zhuang
|
Yue Yu
|
Jieyu Zhang
|
Hang Wu
|
Yuanda Zhu
|
Joyce C. Ho
|
Carl Yang
|
May Dongmei Wang
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Clinicians often rely on data engineers to retrieve complex patient information from electronic health record (EHR) systems, a process that is both inefficient and time-consuming. We propose EHRAgent, a large language model (LLM) agent empowered with accumulative domain knowledge and robust coding capability. EHRAgent enables autonomous code generation and execution to facilitate clinicians in directly interacting with EHRs using natural language. Specifically, we formulate a multi-tabular reasoning task based on EHRs as a tool-use planning process, efficiently decomposing a complex task into a sequence of manageable actions with external toolsets. We first inject relevant medical information to enable EHRAgent to effectively reason about the given query, identifying and extracting the required records from the appropriate tables. By integrating interactive coding and execution feedback, EHRAgent then effectively learns from error messages and iteratively improves its originally generated code. Experiments on three real-world EHR datasets show that EHRAgent outperforms the strongest baseline by up to 29.6% in success rate, verifying its strong capacity to tackle complex clinical tasks with minimal demonstrations.
Search
Co-authors
- Ran Xu 2
- Wenqi Shi 2
- Yue Yu 2
- Yuchen Zhuang 2
- May Dongmei Wang 2
- show all...