Radiology Report Summarization
This model summarizes radiology findings into accurate, informative impressions to improve radiologist-clinician communication.
Model Highlights
- Model name: Radiology_Bart
- Author: Muhammad Bilal
- Model type: Sequence-to-sequence model
- Library: PyTorch, Transformers
- Language: English
Parent Model
This model is a version of pretrained BioBart-v2-base model further finetuned on 70,000 radiology reports to generate radiology impressions. It produces concise, coherent summaries while preserving key findings.
Model Architecture
Radiology_Bart is built on the BioBart architecture, a sequence-to-sequence model which is pre-trained on biomedical-text-data
PubMed. The encoder-decoder structure allows it to compress radiology findings into impression statements.
Key components:
- Encoder: Maps input text to contextualized vector representations
- Decoder: Generates output text token-by-token
- Attention: Aligns relevant encoder and decoder hidden states
Data
The model was trained on 70,000 deidentified radiology reports split into training (52,000), validation (8,000), and test (10,000) sets. The data covers diverse anatomical regions and imaging modalities (X-ray, CT, MRI).
Training
- Optimization: AdamW
- Batch size: 16
- Learning rate: 5.6e-5
- Epochs: 4
The model was trained to maximize the similarity between generated and reference impressions using ROUGE metrics.
Performance
Evaluation Metrics
| ROUGE-1 score | ROUGE-2 score | ROUGE-L score | ROUGELSUM score |
|---|
| 44.857 | 29.015 | 42.032 | 42.038 |
Demonstrating high overlap with human references.
Usage
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, Pipeline
2
3# Sample findings
4findings = "There is a small lung nodule in the right upper lobe measuring 6 mm. The heart size is normal. No pleural effusion or pneumothorax."
5
6# Load model & tokenizer
7summarizer = pipeline("summarization", model="Mbilal755/Radiology_Bart")
8tokenizer = AutoTokenizer.from_pretrained("Mbilal755/Radiology_Bart")
9
10# Tokenize findings
11inputs = tokenizer(findings, return_tensors="pt")
12
13# Generate summary
14summary = summarizer(findings)[0]['summary_text']
15
16# Print outputs
17print(f"Findings: {findings}")
18print(f"Summary: {summary}")
Limitations
This model is designed solely for radiology report summarization. It should not be used for clinical decision-making or other NLP tasks.
Check Demo
Model Card Contact
