English

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

Computer Vision and Pattern Recognition 2024-10-28 v2 Machine Learning

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.

Keywords

Cite

@article{arxiv.2406.19973,
  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},
  journal= {arXiv preprint arXiv:2406.19973},
  year   = {2024}
}

Comments

EMNLP 2024 Main conference