Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

KM-BART: Knowledge Enhanced Multimodal BART for Visual Commonsense Generation

About

We present Knowledge Enhanced Multimodal BART (KM-BART), which is a Transformer-based sequence-to-sequence model capable of reasoning about commonsense knowledge from multimodal inputs of images and texts. We adapt the generative BART architecture to a multimodal model with visual and textual inputs. We further develop novel pretraining tasks to improve the model performance on the Visual Commonsense Generation (VCG) task. In particular, our pretraining task of Knowledge-based Commonsense Generation (KCG) boosts model performance on the VCG task by leveraging commonsense knowledge from a large language model pretrained on external commonsense knowledge graphs. To the best of our knowledge, we are the first to propose a dedicated task for improving model performance on the VCG task. Experimental results show that our model reaches state-of-the-art performance on the VCG task by applying these novel pretraining tasks.

Yiran Xing, Zai Shi, Zhao Meng, Gerhard Lakemeyer, Yunpu Ma, Roger Wattenhofer• 2021

Related benchmarks

TaskDatasetResultRank
Visual Commonsense GenerationVCG (val)
BLEU-214.21
6
Commonsense GenerationVisual Commonsense Graphs (VCG) (val)
Before Score61.3
4
Showing 2 of 2 rows

Other info

Code

Follow for update