Timezone: »

A Universal Abstraction for Hierarchical Hopfield Networks
Benjamin Hoover · Duen Horng Chau · Hendrik Strobelt · Dmitry Krotov

Fri Dec 09 08:50 AM -- 09:05 AM (PST) @
Event URL: https://openreview.net/forum?id=SAv3nhzNWhw »

Conceptualized as Associative Memory, Hopfield Networks (HNs) are powerful models which describe neural network dynamics converging to a local minimum of an energy function. HNs are conventionally described by a neural network with two layers connected by a matrix of synaptic weights. However, it is not well known that the Hopfield framework generalizes to systems in which many neuron layers and synapses work together as a unified Hierarchical Associative Memory (HAM) model: a single network described by memory retrieval dynamics (convergence to a fixed point) and governed by a global energy function. In this work we introduce a universal abstraction for HAMs using the building blocks of neuron layers (nodes) and synapses (edges) connected within a hypergraph. We implement this abstraction as a software framework, written in JAX, whose autograd feature removes the need to derive update rules for the complicated energy-based dynamics. Our framework, called HAMUX (HAM User eXperience), enables anyone to build and train hierarchical HNs using familiar operations like convolutions and attention alongside activation functions like Softmaxes, ReLUs, and LayerNorms. HAMUX is a powerful tool to study HNs at scale, something that has never been possible before. We believe that HAMUX lays the groundwork for a new type of AI framework built around dynamical systems and energy-based associative memories.

Author Information

Benjamin Hoover (IBM Research; GA Tech)
Duen Horng Chau (Georgia Tech)
Hendrik Strobelt (IBM Research)
Dmitry Krotov (IBM Research)

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors