Automatic Personalized Impression Generation for PET Reports Using Large Language Models 📄✍
Authored by: Xin Tie, Muheon Shin, Ali Pirasteh, Nevein Ibrahim, Zachary Huemann, Sharon M. Castellino, Kara Kelly, John Garrett, Junjie Hu, Steve Y. Cho, Tyler J. Bradshaw
📑 Model Description
This is the fine-tuned PEGASUS model for summarizing findings in PET reports.
To check our fine-tuned large language models (LLMs) for PET report summarization:
📑 Abstract
Purpose: To determine if fine-tuned large language models (LLMs) can generate accurate, personalized impressions for whole-body PET reports.
Materials and Methods: Twelve language models were trained on a corpus of PET reports using the teacher-forcing algorithm, with the report findings as input and the clinical impressions as reference. An extra input token encodes the reading physician’s identity, allowing models to learn physician-specific reporting styles. Our corpus comprised 37,370 retrospective PET reports collected from our institution between 2010 and 2022. To identify the best LLM, 30 evaluation metrics were benchmarked against quality scores from two nuclear medicine (NM) physicians, with the most aligned metrics selecting the model for expert evaluation. In a subset of data, model-generated impressions and original clinical impressions were assessed by three NM physicians according to 6 quality dimensions and an overall utility score (5-point scale). Each physician reviewed 12 of their own reports and 12 reports from other physicians. Bootstrap resampling was used for statistical analysis.
Results: Of all evaluation metrics, domain-adapted BARTScore and PEGASUSScore showed the highest Spearman’s ρ correlations (0.568 and 0.563) with physician preferences. Based on these metrics, the fine-tuned PEGASUS model was selected as the top LLM. When physicians reviewed PEGASUS-generated impressions in their own style, 89% were considered clinically acceptable, with a mean utility score of 4.08/5. Physicians rated these personalized impressions as comparable in overall utility to the impressions dictated by other physicians (4.03, P=0.41).
Conclusion: Personalized impressions generated by PEGASUS were clinically useful, highlighting its potential to expedite PET reporting.
🚀 Usage
1finetuned_model = "xtie/PEGASUS-PET-impression"
2tokenizer = AutoTokenizer.from_pretrained(finetuned_model)
3model = AutoModelForSeq2SeqLM.from_pretrained(finetuned_model, ignore_mismatched_sizes=True).eval()
4
5findings_info =
6"""
7Description: PET CT WHOLE BODY
8Radiologist: James
9Findings:
10Head/Neck: xxx Chest: xxx Abdomen/Pelvis: xxx Extremities/Musculoskeletal: xxx
11Indication:
12The patient is a 60-year old male with a history of xxx
13"""
14
15inputs = tokenizer(findings_info.replace('\n', ' '),
16 padding="max_length",
17 truncation=True,
18 max_length=1024,
19 return_tensors="pt")
20input_ids = inputs.input_ids.to("cuda")
21attention_mask = inputs.attention_mask.to("cuda")
22outputs = model.generate(input_ids,
23 attention_mask=attention_mask,
24 max_new_tokens=512,
25 num_beam_groups=1,
26 num_beams=4,
27 do_sample=False,
28 diversity_penalty=0.0,
29 num_return_sequences=1,
30 length_penalty=2.0,
31 no_repeat_ngram_size=3,
32 early_stopping=True
33 )
34# get the generated impressions
35output_str = tokenizer.decode(outputs,
36 skip_special_tokens=True)
📊 Performance Metrics
For detailed evaluation results, please refer to our paper.
- ROUGE-1: 53.8
- ROUGE-2: 30.9
- ROUGE-L: 40.0
- BLEU: 24.7
- BERTScore: 0.744
💡 Highlights
- The fine-tuned large language model provides clinically useful, personalized impressions based on PET findings.
- To our knowledge, this is the first attempt to automate impression generation for whole-body PET reports.
🖥️ Hardware
The models were trained on NVIDIA A100 GPUs.
📁 Additional Resources
- Finetuned from model: Google's PEGASUS Large
- Codebase for training and inference: GitHub Repository
