R2GenGPT: Radiology Report Generation with Frozen LLMs

Zhanyu Wang, Lingqiao Liu, Lei Wang, Luping Zhou
9/18/2023
cs.CV

Official code: wang-zhanyu/R2GenGPT (132 stars, BSD-3-Clause, last commit 2024-04-19).

Abstract

Large Language Models (LLMs) have consistently showcased remarkable generalization capabilities when applied to various language tasks. Nonetheless, harnessing the full potential of LLMs for Radiology Report Generation (R2Gen) still presents a challenge, stemming from the inherent disparity in modality between LLMs and the R2Gen task. To bridge this gap effectively, we propose R2GenGPT, which is a novel solution that aligns visual features with the word embedding space of LLMs using an efficient visual alignment module. This innovative approach empowers the previously static LLM to seamlessly integrate and process image information, marking a step forward in optimizing R2Gen performance. R2GenGPT offers the following benefits. First, it attains state-of-the-art (SOTA) performance by training only the lightweight visual alignment module while freezing all the parameters of LLM. Second, it exhibits high training efficiency, as it requires the training of an exceptionally minimal number of parameters while achieving rapid convergence. By employing delta tuning, our model only trains 5M parameters (which constitute just 0.07\% of the total parameter count) to achieve performance close to the SOTA levels. Our code is available at https://github.com/wang-zhanyu/R2GenGPT.

View on arXivView PDF

Code Implementations(1)

BSD-3-Clause

Radiology Report Generation with Frozen LLMs

13216Shell, PythonAug 7, 20232 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{wang2023rgengpt,
  title  = {R2GenGPT: Radiology Report Generation with Frozen LLMs},
  author = {Zhanyu Wang and Lingqiao Liu and Lei Wang and Luping Zhou},
  year   = {2023},
  eprint = {2309.09812},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2309.09812v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2309.09812.svg)](https://codeofpaper.com/paper/2309.09812)

Discussion