Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluation
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , , , , , , , , , , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866912304388374528 |
|---|---|
| author | Chaves, Juan Manuel Zambrano Huang, Shih-Cheng Xu, Yanbo Xu, Hanwen Usuyama, Naoto Zhang, Sheng Wang, Fei Xie, Yujia Khademi, Mahmoud Yang, Ziyi Awadalla, Hany Gong, Julia Hu, Houdong Yang, Jianwei Li, Chunyuan Gao, Jianfeng Gu, Yu Wong, Cliff Wei, Mu Naumann, Tristan Chen, Muhao Lungren, Matthew P. Chaudhari, Akshay Yeung-Levy, Serena Langlotz, Curtis P. Wang, Sheng Poon, Hoifung |
| author_facet | Chaves, Juan Manuel Zambrano Huang, Shih-Cheng Xu, Yanbo Xu, Hanwen Usuyama, Naoto Zhang, Sheng Wang, Fei Xie, Yujia Khademi, Mahmoud Yang, Ziyi Awadalla, Hany Gong, Julia Hu, Houdong Yang, Jianwei Li, Chunyuan Gao, Jianfeng Gu, Yu Wong, Cliff Wei, Mu Naumann, Tristan Chen, Muhao Lungren, Matthew P. Chaudhari, Akshay Yeung-Levy, Serena Langlotz, Curtis P. Wang, Sheng Poon, Hoifung |
| contents | The scaling laws and extraordinary performance of large foundation models motivate the development and utilization of such models in biomedicine. However, despite early promising results on some biomedical benchmarks, there are still major challenges that need to be addressed before these models can be used in real-world clinics. Frontier general-domain models such as GPT-4V still have significant performance gaps in multimodal biomedical applications. More importantly, less-acknowledged pragmatic issues, including accessibility, model cost, and tedious manual evaluation make it hard for clinicians to use state-of-the-art large models directly on private patient data. Here, we explore training open-source small multimodal models (SMMs) to bridge competency gaps for unmet clinical needs in radiology. To maximize data efficiency, we adopt a modular approach by incorporating state-of-the-art pre-trained models for image and text modalities, and focusing on training a lightweight adapter to ground each modality to the text embedding space, as exemplified by LLaVA-Med. For training, we assemble a large dataset of over 697 thousand radiology image-text pairs. For evaluation, we propose CheXprompt, a GPT-4-based metric for factuality evaluation, and demonstrate its parity with expert evaluation. For best practice, we conduct a systematic ablation study on various choices in data engineering and multimodal training. The resulting LlaVA-Rad (7B) model attains state-of-the-art results on standard radiology tasks such as report generation and cross-modal retrieval, even outperforming much larger models such as GPT-4V and Med-PaLM M (84B). The inference of LlaVA-Rad is fast and can be performed on a single V100 GPU in private settings, offering a promising state-of-the-art tool for real-world clinical applications. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2403_08002 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluation Chaves, Juan Manuel Zambrano Huang, Shih-Cheng Xu, Yanbo Xu, Hanwen Usuyama, Naoto Zhang, Sheng Wang, Fei Xie, Yujia Khademi, Mahmoud Yang, Ziyi Awadalla, Hany Gong, Julia Hu, Houdong Yang, Jianwei Li, Chunyuan Gao, Jianfeng Gu, Yu Wong, Cliff Wei, Mu Naumann, Tristan Chen, Muhao Lungren, Matthew P. Chaudhari, Akshay Yeung-Levy, Serena Langlotz, Curtis P. Wang, Sheng Poon, Hoifung Computation and Language Computer Vision and Pattern Recognition The scaling laws and extraordinary performance of large foundation models motivate the development and utilization of such models in biomedicine. However, despite early promising results on some biomedical benchmarks, there are still major challenges that need to be addressed before these models can be used in real-world clinics. Frontier general-domain models such as GPT-4V still have significant performance gaps in multimodal biomedical applications. More importantly, less-acknowledged pragmatic issues, including accessibility, model cost, and tedious manual evaluation make it hard for clinicians to use state-of-the-art large models directly on private patient data. Here, we explore training open-source small multimodal models (SMMs) to bridge competency gaps for unmet clinical needs in radiology. To maximize data efficiency, we adopt a modular approach by incorporating state-of-the-art pre-trained models for image and text modalities, and focusing on training a lightweight adapter to ground each modality to the text embedding space, as exemplified by LLaVA-Med. For training, we assemble a large dataset of over 697 thousand radiology image-text pairs. For evaluation, we propose CheXprompt, a GPT-4-based metric for factuality evaluation, and demonstrate its parity with expert evaluation. For best practice, we conduct a systematic ablation study on various choices in data engineering and multimodal training. The resulting LlaVA-Rad (7B) model attains state-of-the-art results on standard radiology tasks such as report generation and cross-modal retrieval, even outperforming much larger models such as GPT-4V and Med-PaLM M (84B). The inference of LlaVA-Rad is fast and can be performed on a single V100 GPU in private settings, offering a promising state-of-the-art tool for real-world clinical applications. |
| title | Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluation |
| topic | Computation and Language Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2403.08002 |