Measuring and Controlling Solution Degeneracy across Task-Trained Recurrent Neural Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Huang, Ann, Singh, Satpreet H., Martinelli, Flavio, Rajan, Kanaka
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915787204198400
author Huang, Ann
Singh, Satpreet H.
Martinelli, Flavio
Rajan, Kanaka
author_facet Huang, Ann
Singh, Satpreet H.
Martinelli, Flavio
Rajan, Kanaka
contents Task-trained recurrent neural networks (RNNs) are widely used in neuroscience and machine learning to model dynamical computations. To gain mechanistic insight into how neural systems solve tasks, prior work often reverse-engineers individual trained networks. However, different RNNs trained on the same task and achieving similar performance can exhibit strikingly different internal solutions, a phenomenon known as solution degeneracy. Here, we develop a unified framework to systematically quantify and control solution degeneracy across three levels: behavior, neural dynamics, and weight space. We apply this framework to 3,400 RNNs trained on four neuroscience-relevant tasks: flip-flop memory, sine wave generation, delayed discrimination, and path integration, while systematically varying task complexity, learning regime, network size, and regularization. We find that higher task complexity and stronger feature learning reduce degeneracy in neural dynamics but increase it in weight space, with mixed effects on behavior. In contrast, larger networks and structural regularization reduce degeneracy at all three levels. These findings empirically validate the Contravariance Principle and provide practical guidance for researchers seeking to tune the variability of RNN solutions, either to uncover shared neural mechanisms or to model the individual variability observed in biological systems. This work provides a principled framework for quantifying and controlling solution degeneracy in task-trained RNNs, offering new tools for building more interpretable and biologically grounded models of neural computation.
format Preprint
id arxiv_https___arxiv_org_abs_2410_03972
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Measuring and Controlling Solution Degeneracy across Task-Trained Recurrent Neural Networks
Huang, Ann
Singh, Satpreet H.
Martinelli, Flavio
Rajan, Kanaka
Machine Learning
Information Theory
Neural and Evolutionary Computing
Neurons and Cognition
Task-trained recurrent neural networks (RNNs) are widely used in neuroscience and machine learning to model dynamical computations. To gain mechanistic insight into how neural systems solve tasks, prior work often reverse-engineers individual trained networks. However, different RNNs trained on the same task and achieving similar performance can exhibit strikingly different internal solutions, a phenomenon known as solution degeneracy. Here, we develop a unified framework to systematically quantify and control solution degeneracy across three levels: behavior, neural dynamics, and weight space. We apply this framework to 3,400 RNNs trained on four neuroscience-relevant tasks: flip-flop memory, sine wave generation, delayed discrimination, and path integration, while systematically varying task complexity, learning regime, network size, and regularization. We find that higher task complexity and stronger feature learning reduce degeneracy in neural dynamics but increase it in weight space, with mixed effects on behavior. In contrast, larger networks and structural regularization reduce degeneracy at all three levels. These findings empirically validate the Contravariance Principle and provide practical guidance for researchers seeking to tune the variability of RNN solutions, either to uncover shared neural mechanisms or to model the individual variability observed in biological systems. This work provides a principled framework for quantifying and controlling solution degeneracy in task-trained RNNs, offering new tools for building more interpretable and biologically grounded models of neural computation.
title Measuring and Controlling Solution Degeneracy across Task-Trained Recurrent Neural Networks
topic Machine Learning
Information Theory
Neural and Evolutionary Computing
Neurons and Cognition
url https://arxiv.org/abs/2410.03972