Home / Publications / Local Large Language Models for Complex Structured Tasks.
Publication

Local Large Language Models for Complex Structured Tasks.

Bumgardner V K Cody, Mullen Aaron, Armstrong Samuel E, Hickey Caylin, Marek Victor, Talbert Jeff

Details

Journal AMIA Joint Summits on Translational Science proceedings. AMIA Joint Summits on Translational Science
Year 2024
Categories cs.CL, cs.AI
Note 12 pages, Preprint of an article submitted for consideration in Pacific Symposium on Biocomputing c{opyright} 2024 copyright World Scientific Publishing Company https://www.worldscientific.com/

Abstract

This paper introduces an approach that combines the language reasoning capabilities of large language models (LLMs) with the benefits of local training to tackle complex language tasks. The authors demonstrate their approach by extracting structured condition codes from pathology reports. The proposed approach utilizes local, fine-tuned LLMs to respond to specific generative instructions and provide structured outputs. Over 150k uncurated surgical pathology reports containing gross descriptions, final diagnoses, and condition codes were used. Different model architectures were trained and evaluated, including LLaMA, BERT, and LongFormer. The results show that the LLaMA-based models significantly outperform BERT-style models across all evaluated metrics. LLaMA models performed especially well with large datasets, demonstrating their ability to handle complex, multi-label tasks. Overall, this work presents an effective approach for utilizing LLMs to perform structured generative tasks on domain-specific language in the medical domain.