Learning deep representations by mutual information estimation and maximization

R. Devon Hjelm,A. Fedorov,Samuel Lavoie-Marchildon,Karan Grewal,Adam Trischler,Yoshua Bengio

Published 2018 in International Conference on Learning Representations

ABSTRACT

This work investigates unsupervised learning of representations by maximizing mutual information between an input and the output of a deep neural network encoder. Importantly, we show that structure matters: incorporating knowledge about locality in the input into the objective can significantly improve a representation’s suitability for downstream tasks. We further control characteristics of the representation by matching to a prior distribution adversarially. Our method, which we call Deep InfoMax (DIM), outperforms a number of popular unsupervised learning methods and compares favorably with fully-supervised learning on several classification tasks in with some standard architectures. DIM opens new avenues for unsupervised learning of representations and is an important step towards flexible formulations of representation learning objectives for specific end-goals.

PUBLICATION RECORD

  • Publication year

    2018

  • Venue

    International Conference on Learning Representations

  • Publication date

    2018-08-20

  • Fields of study

    Mathematics, Computer Science

  • Identifiers
  • External record

    Open on Semantic Scholar

  • Source metadata

    Semantic Scholar

CITATION MAP

EXTRACTION MAP

CLAIMS

  • No claims are published for this paper.

CONCEPTS

  • No concepts are published for this paper.

REFERENCES

Showing 1-77 of 77 references · Page 1 of 1

CITED BY

Showing 1-100 of 2938 citing papers · Page 1 of 30