Representational Distance Learning for Deep Neural Networks

被引:31
作者
McClure, Patrick [1 ]
Kriegeskorte, Nikolaus [1 ]
机构
[1] MRC, Cognit & Brain Sci Unit, Cambridge, England
基金
欧洲研究理事会; 英国医学研究理事会;
关键词
neural networks; transfer learning; distance matrices; visual perception; computational neuroscience; GRADIENT; MODELS;
D O I
10.3389/fncom.2016.00131
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
Deep neural networks (DNNs) provide useful models of visual representational transformations. We present a method that enables a DNN (student) to learn from the internal representational spaces of a reference model (teacher), which could be another DNN or, in the future, a biological brain. Representational spaces of the student and the teacher are characterized by representational distance matrices (RDMs). We propose representational distance learning (RDL), a stochastic gradient descent method that drives the RDMs of the student to approximate the RDMs of the teacher. We demonstrate that RDL is competitive with other transfer learning techniques for two publicly available benchmark computer vision datasets (MNIST and CIFAR-100), while allowing for architectural differences between student and teacher. By pulling the student's RDMs toward those of the teacher, RDL significantly improved visual classification performance when compared to baseline networks that did not use transfer learning. In the future, RDL may enable combined supervised training of deep neural networks using task constraints (e.g., images and category labels) and constraints from brain-activity measurements, so as to build models that replicate the internal representational spaces of biological brains.
引用
收藏
页数:10
相关论文
共 31 条
  • [1] [Anonymous], 2015, DEEP LEARNING WORKSH
  • [2] [Anonymous], 2012, NEURAL NETWORKS TRIC
  • [3] [Anonymous], 2014, Deeply Supervised Nets
  • [4] [Anonymous], 2015, ARXIV PREPRINT ARXIV
  • [5] [Anonymous], 2015, ARXIV150502496
  • [6] [Anonymous], 2007, Large Scale Kernel Machines
  • [7] [Anonymous], 2014, Advances in neural information processing ,systems, DOI 10.5555/2969033.2969123
  • [8] [Anonymous], BIGLEARN NIPS WORKSH
  • [9] [Anonymous], 2014, P 2 INT C LEARNING R
  • [10] [Anonymous], 2014, Advances in neural information processing systems