STLLaVA-Med: Self-Training Large Language and Vision Assistant for Medical Question-Answering

Guohao Sun, Can Qin, Huazhu Fu, Linwei Wang, Zhiqiang Tao
6/28/2024
cs.CVcs.LG

No official code found. Best implementation: heliossun/STLLaVA-Med (16 stars, MIT, last commit 2024-11-03).

Abstract

Large Vision-Language Models (LVLMs) have shown significant potential in assisting medical diagnosis by leveraging extensive biomedical datasets. However, the advancement of medical image understanding and reasoning critically depends on building high-quality visual instruction data, which is costly and labor-intensive to obtain, particularly in the medical domain. To mitigate this data-starving issue, we introduce Self-Training Large Language and Vision Assistant for Medicine (STLLaVA-Med). The proposed method is designed to train a policy model (an LVLM) capable of auto-generating medical visual instruction data to improve data efficiency, guided through Direct Preference Optimization (DPO). Specifically, a more powerful and larger LVLM (e.g., GPT-4o) is involved as a biomedical expert to oversee the DPO fine-tuning process on the auto-generated data, encouraging the policy model to align efficiently with human preferences. We validate the efficacy and data efficiency of STLLaVA-Med across three major medical Visual Question Answering (VQA) benchmarks, demonstrating competitive zero-shot performance with the utilization of only 9% of the medical data.

View on arXivView PDF

Code Implementations(1)

MIT

Self-training LLaVA for medical

161CSS, HTMLJun 26, 20241 years agoMIT

Similar Papers

Cite this paper

@article{sun2024stllavamed,
  title  = {STLLaVA-Med: Self-Training Large Language and Vision Assistant for Medical Question-Answering},
  author = {Guohao Sun and Can Qin and Huazhu Fu and Linwei Wang and Zhiqiang Tao},
  year   = {2024},
  eprint = {2406.19973},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.19973v2}
}

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/2406.19973.svg)](https://codeofpaper.com/paper/2406.19973)

Discussion