Acta Universitatis Sapientiae: Informatica (Jul 2020)

Encouraging an appropriate representation simplifies training of neural networks

  • Buza Krisztian

DOI
https://doi.org/10.2478/ausi-2020-0007
Journal volume & issue
Vol. 12, no. 1
pp. 102 – 111

Abstract

Read online

A common assumption about neural networks is that they can learn an appropriate internal representation on their own, see e.g. end-to-end learning. In this work we challenge this assumption. We consider two simple tasks and show that the state-of-the-art training algorithm fails, although the model itself is able to represent an appropriate solution. We will demonstrate that encouraging an appropriate internal representation allows the same model to solve these tasks. While we do not claim that it is impossible to solve these tasks by other means (such as neural networks with more layers), our results illustrate that integration of domain knowledge in form of a desired internal representation may improve the generalization ability of neural networks.

Keywords