Prompt-tuning GatorTronGPT-20B gives efficient, higher-scoring clinical dialogue summaries

March 19, 20246 min

Overview

Decision SnapshotNeeds Validation

Scores reflect test-set improvements and lower compute cost from prompt tuning, tempered by reliance on automatic metrics and limited public release details.

Citations1

Evidence Strength0.70

Confidence0.90

Risk Signals10

Trust Signals

Findings with numeric evidence: 5/5

Findings with evidence refs: 5/5

Results with explicit delta: 3/5

Reproducibility

Status: Partial assets available

Open source: Unknown

At A Glance

Cost impact: 70%

Production readiness: 60%

Novelty: 50%

Authors

Mengxian Lyu, Cheng Peng, Xiaohan Li, Patrick Balian, Jiang Bian, Yonghui Wu

Links

Abstract / PDF

Why It Matters For Business

Prompt tuning lets teams deploy clinical summarization with much lower compute and faster turnaround than full fine-tuning while often improving quality if you have a large domain LLM.

Who Should Care

Summary TLDR

The authors use prompt tuning (trainable soft prompts) to steer clinical LLMs (GatorTronGPT-5B and -20B) to summarize doctor–patient dialogues into clinical notes. On the MTS-DIALOG benchmark, prompt-tuned GatorTronGPT-20B outperforms a fine-tuned T5-Large across ROUGE, BERTScore and BLEU (e.g., Rouge-1: 0.3628 vs 0.3425). Prompt tuning updates a small subset of parameters (70M–302M) while keeping the LLM frozen, cutting fine-tuning time (2h10m–4h23m vs 9h34m) and compute. Soft prompt length 128 tokens worked best. Few-shot tuning improves with sample size but still trails full-data performance.

Problem Statement

Clinical documentation is time-consuming and causes burnout. We need cost-efficient automatic summarization that produces concise, clinically useful notes from doctor–patient dialogues without expensive full-model fine-tuning.

Main Contribution

Designed prompt-tuning (soft prompts) pipeline for clinical dialogue summarization using GatorTronGPT.

Compared initialization strategies (LSTM vs MLP) and soft-prompt lengths; found LSTM and 128 virtual tokens best.

Key Findings

GatorTronGPT-20B prompt-tuned outperformed fine-tuned T5-Large on automatic metrics.

NumbersRouge-1 0.3628 vs 0.3425; BERTScore 0.7309 vs 0.6765 (Table 4)

Practical UseIf you have access to a large clinical LLM, prompt tuning can yield better automatic-summary scores than fine-tuning a T5 baseline on the same data.

Evidence RefTable 4

Prompt tuning updates far fewer parameters and runs faster than full fine-tuning.

NumbersGatorTronGPT-20B prompt-tuning: 302M params, 4h23m vs T5 fine-tune: 770M params, 9h34m (Table 4)

Practical UseUse prompt tuning to cut compute and wall-clock time when you must keep base model weights frozen.

Evidence RefTable 4

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Rouge-1 (test)0.3628T5-Large 0.3425+0.0203MTS-DIALOG testTable 4 reports GatorTronGPT-20B 0.3628 vs T5-Large 0.3425Table 4
BERTScore (test)0.7309T5-Large 0.6765+0.0544MTS-DIALOG testTable 4 reports BERTScore 0.7309 for GatorTronGPT-20B vs 0.6765 for T5-LargeTable 4

What To Try In 7 Days

Run soft-prompt tuning on an available clinical LLM; start with 128 virtual tokens and LSTM init.

Compare trainable-parameter budget and wall-clock time vs fine-tuning a small encoder-decoder model.

Validate summaries with a few dozen clinician checks and BERTScore/ROUGE to detect obvious gaps or hallucinations.

Optimization Features

Token Efficiency
Found 128 virtual tokens best for this task
Training Optimization
Prompt tuning (update soft prompts, freeze base weights)

Reproducibility

Code AvailableNo
Data AvailableYes
Open Source StatusUnknown
LicenseUnknown

Risks & Boundaries

Limitations

Evaluation relies on automatic metrics (ROUGE/BERTScore) that may not match clinician judgement.

Prompt tuning matched or beat fine-tuning only for large generative LLMs; smaller LLMs lag.

When Not To Use

When you need verified, human-reviewed clinical summaries without automated checks.

If you lack access to a large pre-trained clinical LLM (20B+ scale) required for parity.

Failure Modes

Hallucinated facts or omitted critical clinical details not caught by n-gram metrics.

Performance drop in specialties underrepresented in MTS-DIALOG.

Core Entities

Models

GatorTronGPT-5BGatorTronGPT-20BT5-Large

Metrics

ROUGE-1ROUGE-2ROUGE-LBLEUBERTScore

Datasets

MTS-DIALOG