Domain-adapted large language models for classifying nuclear medicine reports

Zachary Huemann,Changhee Lee,Junjie Hu,Steve Y. Cho,Tyler Bradshaw
2023-03-01
Abstract:With the growing use of transformer-based language models in medicine, it is unclear how well these models generalize to nuclear medicine which has domain-specific vocabulary and unique reporting styles. In this study, we evaluated the value of domain adaptation in nuclear medicine by adapting language models for the purpose of 5-point Deauville score prediction based on clinical 18F-fluorodeoxyglucose (FDG) PET/CT reports. We retrospectively retrieved 4542 text reports and 1664 images for FDG PET/CT lymphoma exams from 2008-2018 in our clinical imaging database. Deauville scores were removed from the reports and then the remaining text in the reports was used as the model input. Multiple general-purpose transformer language models were used to classify the reports into Deauville scores 1-5. We then adapted the models to the nuclear medicine domain using masked language modeling and assessed its impact on classification performance. The language models were compared against vision models, a multimodal vision language model, and a nuclear medicine physician with seven-fold Monte Carlo cross validation, reported are the mean and standard deviations. Domain adaption improved all language models. For example, BERT improved from 61.3% five-class accuracy to 65.7% following domain adaptation. The best performing model (domain-adapted RoBERTa) achieved a five-class accuracy of 77.4%, which was better than the physician's performance (66%), the best vision model's performance (48.1), and was similar to the multimodal model's performance (77.2). Domain adaptation improved the performance of large language models in interpreting nuclear medicine text reports.
Computation and Language,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to evaluate the applicability and improvement methods of large - scale language models in the field of nuclear medicine, especially how these models can improve the accuracy of Deauville score prediction in nuclear medicine reports through domain adaptation. Nuclear medicine reports contain domain - specific vocabulary and a unique reporting style, which may make general - purpose language models unable to understand and interpret these reports well. Therefore, the researchers used 4,542 clinical 18F - fluorodeoxyglucose (FDG) positron emission tomography/computed tomography (PET/CT) reports and 1,664 images. After removing the Deauville scores from the reports, they used the remaining text as input to test the ability of multiple general - purpose Transformer language models (such as BERT, RoBERTa, etc.) to classify the reports into Deauville scores of 1 - 5. In addition, the study also explored the impact of domain adaptation through Masked Language Modeling (MLM) on classification performance and compared the performance of language models with that of visual models, multimodal visual - language models, and nuclear medicine physicians. The study hypothesizes that domain adaptation can improve the performance of the downstream task - predicting the Deauville score. Through this method, the study aims to explore how to improve the efficiency and accuracy of large - scale language models in processing nuclear medicine text reports through domain adaptation techniques, thereby providing more accurate support tools for clinical applications.