On Architectures for Including Visual Information in Neural Language Models for Image Description

  • 2019-11-09 17:07:23
  • Marc Tanti, Albert Gatt, Kenneth P. Camilleri
  • 0

Abstract

A neural language model can be conditioned into generating descriptions forimages by providing visual information apart from the sentence prefix. Thisvisual information can be included into the language model through differentpoints of entry resulting in different neural architectures. We identify fourmain architectures which we call init-inject, pre-inject, par-inject, andmerge. We analyse these four architectures and conclude that the best performing oneis init-inject, which is when the visual information is injected into theinitial state of the recurrent neural network. We confirm this using bothautomatic evaluation measures and human annotation. We then analyse how much influence the images have on each architecture. Thisis done by measuring how different the output probabilities of a model are whena partial sentence is combined with a completely different image from the oneit is meant to be combined with. We find that init-inject tends to quicklybecome less influenced by the image as more words are generated. A differentarchitecture called merge, which is when the visual information is merged withthe recurrent neural network's hidden state vector prior to output, losesvisual influence much more slowly, suggesting that it would work better forgenerating longer sentences. We also observe that the merge architecture can have its recurrent neuralnetwork pre-trained in a text-only language model (transfer learning) ratherthan be initialised randomly as usual. This results in even better performancethan the other architectures, provided that the source language model is nottoo good at language modelling or it will overspecialise and be less effectiveat image description generation. Our work opens up new avenues of research in neural architectures,explainable AI, and transfer learning.

 

Quick Read (beta)

loading the full paper ...