MMBERT: Multimodal BERT Pretraining for Improved Medical VQA

Yash Khare,Viraj Bagal,Minesh Mathew,Adithi Devi,U Deva Priyakumar,CV Jawahar
DOI: https://doi.org/10.48550/arXiv.2104.01394
2021-04-03
Abstract:Images in the medical domain are fundamentally different from the general domain images. Consequently, it is infeasible to directly employ general domain Visual Question Answering (VQA) models for the medical domain. Additionally, medical images annotation is a costly and time-consuming process. To overcome these limitations, we propose a solution inspired by self-supervised pretraining of Transformer-style architectures for NLP, Vision and Language tasks. Our method involves learning richer medical image and text semantic representations using Masked Language Modeling (MLM) with image features as the pretext task on a large medical image+caption dataset. The proposed solution achieves new state-of-the-art performance on two VQA datasets for radiology images -- VQA-Med 2019 and VQA-RAD, outperforming even the ensemble models of previous best solutions. Moreover, our solution provides attention maps which help in model interpretability. The code is available at <a class="link-external link-https" href="https://github.com/VirajBagal/MMBERT" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition,Computation and Language,Machine Learning
What problem does this paper attempt to address?