Skip to main navigation Skip to search Skip to main content

Where is the Model Looking At? - Concentrate and Explain the Network Attention

  • Wenjia Xu
  • , Jiuniu Wang*
  • , Yang Wang
  • , Guangluan Xu*
  • , Daoyu Lin
  • , Wei Dai
  • , Yirong Wu
  • *Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

Abstract

Image classification models have achieved satisfactory performance on many datasets, sometimes even better than human. However, The model attention is unclear since the lack of interpretability. This paper investigates the fidelity and interpretability of model attention. We propose an Explainable Attribute-based Multi-task (EAT) framework to concentrate the model attention on the discriminative image area and make the attention interpretable. We introduce attributes prediction to the multi-task learning network, helping the network to concentrate attention on the foreground objects. We generate attribute-based textual explanations for the network and ground the attributes on the image to show visual explanations. The multi-model explanation can not only improve user trust but also help to find the weakness of network and dataset. Our framework can be generalized to any basic model. We perform experiments on three datasets and five basic models. Results indicate that the EAT framework can give multi-modal explanations that interpret the network decision. The performance of several recognition approaches is improved by guiding network attention.
Original languageEnglish
Pages (from-to)506-516
JournalIEEE Journal on Selected Topics in Signal Processing
Volume14
Issue number3
DOIs
Publication statusPublished - Mar 2020
Externally publishedYes

Research Keywords

  • Explainable artificial intelligence
  • Multi-task learning
  • Attributes

Fingerprint

Dive into the research topics of 'Where is the Model Looking At? - Concentrate and Explain the Network Attention'. Together they form a unique fingerprint.

Cite this