ToyArchitecture: Unsupervised learning of interpretable models of the environment.

Journal: PloS one
Published Date:

Abstract

Research in Artificial Intelligence (AI) has focused mostly on two extremes: either on small improvements in narrow AI domains, or on universal theoretical frameworks which are often uncomputable, or lack practical implementations. In this paper we attempt to follow a big picture view while also providing a particular theory and its implementation to present a novel, purposely simple, and interpretable hierarchical architecture. This architecture incorporates the unsupervised learning of a model of the environment, learning the influence of one's own actions, model-based reinforcement learning, hierarchical planning, and symbolic/sub-symbolic integration in general. The learned model is stored in the form of hierarchical representations which are increasingly more abstract, but can retain details when needed. We demonstrate the universality of the architecture by testing it on a series of diverse environments ranging from audio/visual compression to discrete and continuous action spaces, to learning disentangled representations.

Authors

  • Jaroslav Vítků
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Petr Dluhoš
    Behavioural and Social Neuroscience Group, CEITEC - Central European Institute of Technology, Masaryk University, Brno, Czech Republic; Department of Psychiatry, University Hospital Brno and Masaryk University, Brno, Czech Republic. Electronic address: dluhos@mail.muni.cz.
  • Joseph Davidson
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Matěj Nikl
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Simon Andersson
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Přemysl Paška
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Jan Šinkora
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Petr Hlubuček
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Martin Stránský
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Martin Hyben
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Martin Poliak
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Jan Feyereisl
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.
  • Marek Rosa
    GoodAI Research s.r.o., Karolinská, Prague, Czech Republic.