Customizing General-Purpose Foundation Models for Medical Report Generation
Authors: Bang Yang, Asif Raza, Yuexian Zou, Tong Zhang
AI-generated Key Points
- Medical report generation (MRG) involves automatically generating accurate and coherent captions for medical images.
- Scarcity of labeled medical image-report pairs poses challenges in developing deep and large-scale neural networks for MRG.
- The authors propose customizing off-the-shelf general-purpose large-scale pre-trained models, known as foundation models (FMs), for MRG.
- Their encoder-decoder based MRG model utilizes a lightweight query Transformer to connect two FMs: EVA-ViT-g (vision Transformer) and ChatGLM-6B (bilingual language model).
- Unfreezing EVA-ViT-g to learn medical image representations and parameter efficient training of ChatGLM 6B are crucial factors for optimal results.
- The authors achieved impressive rankings in the ImageCLEFmedical Caption 2023 competition based on BERTScore and ROUGE 1 metrics.
- Previous research on MRG has focused on cross modal alignment, reinforcement learning, architecture design, explicit loss constraints, retrieval, and knowledge augmented approaches.
- Foundation models have become a research hotspot in computer vision and natural language processing.
- Prompt engineering and parameter efficient transfer learning are popular techniques in leveraging foundation models.
- This work presents a novel approach to MRG by customizing off-the-shelf foundation models, with experimental results demonstrating its effectiveness.
Abstract: Medical caption prediction which can be regarded as a task of medical report generation (MRG), requires the automatic generation of coherent and accurate captions for the given medical images. However, the scarcity of labelled medical image-report pairs presents great challenges in the development of deep and large-scale neural networks capable of harnessing the potential artificial general intelligence power like large language models (LLMs). In this work, we propose customizing off-the-shelf general-purpose large-scale pre-trained models, i.e., foundation models (FMs), in computer vision and natural language processing with a specific focus on medical report generation. Specifically, following BLIP-2, a state-of-the-art vision-language pre-training approach, we introduce our encoder-decoder-based MRG model. This model utilizes a lightweight query Transformer to connect two FMs: the giant vision Transformer EVA-ViT-g and a bilingual LLM trained to align with human intentions (referred to as ChatGLM-6B). Furthermore, we conduct ablative experiments on the trainable components of the model to identify the crucial factors for effective transfer learning. Our findings demonstrate that unfreezing EVA-ViT-g to learn medical image representations, followed by parameter-efficient training of ChatGLM-6B to capture the writing styles of medical reports, is essential for achieving optimal results. Our best attempt (PCLmed Team) achieved the 4th and the 2nd, respectively, out of 13 participating teams, based on the BERTScore and ROUGE-1 metrics, in the ImageCLEFmedical Caption 2023 Caption Prediction Task competition.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.