Saved in:
Bibliographic Details
Main Authors: Ebrahimi, M. Reza, Memisevic, Roland
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.21749
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914110523834368
author Ebrahimi, M. Reza
Memisevic, Roland
author_facet Ebrahimi, M. Reza
Memisevic, Roland
contents The role of hidden units in recurrent neural networks is typically seen as modeling memory, with research focusing on enhancing information retention through gating mechanisms. A less explored perspective views hidden units as active participants in the computation performed by the network, rather than passive memory stores. In this work, we revisit bilinear operations, which involve multiplicative interactions between hidden units and input embeddings. We demonstrate theoretically and empirically that they constitute a natural inductive bias for representing the evolution of hidden states in state tracking tasks. These are the simplest type of tasks that require hidden units to actively contribute to the behavior of the network. We also show that bilinear state updates form a natural hierarchy corresponding to state tracking tasks of increasing complexity, with popular linear recurrent networks such as Mamba residing at the lowest-complexity center of that hierarchy.
format Preprint
id arxiv_https___arxiv_org_abs_2505_21749
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Revisiting Bi-Linear State Transitions in Recurrent Neural Networks
Ebrahimi, M. Reza
Memisevic, Roland
Machine Learning
Computation and Language
The role of hidden units in recurrent neural networks is typically seen as modeling memory, with research focusing on enhancing information retention through gating mechanisms. A less explored perspective views hidden units as active participants in the computation performed by the network, rather than passive memory stores. In this work, we revisit bilinear operations, which involve multiplicative interactions between hidden units and input embeddings. We demonstrate theoretically and empirically that they constitute a natural inductive bias for representing the evolution of hidden states in state tracking tasks. These are the simplest type of tasks that require hidden units to actively contribute to the behavior of the network. We also show that bilinear state updates form a natural hierarchy corresponding to state tracking tasks of increasing complexity, with popular linear recurrent networks such as Mamba residing at the lowest-complexity center of that hierarchy.
title Revisiting Bi-Linear State Transitions in Recurrent Neural Networks
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2505.21749