pyhgf: A neural network library for predictive coding

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Legrand, Nicolas, Weber, Lilian, Waade, Peter Thestrup, Daugaard, Anna Hedvig Møller, Khodadadi, Mojtaba, Mikuš, Nace, Mathys, Chris
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908479555371008
author Legrand, Nicolas
Weber, Lilian
Waade, Peter Thestrup
Daugaard, Anna Hedvig Møller
Khodadadi, Mojtaba
Mikuš, Nace
Mathys, Chris
author_facet Legrand, Nicolas
Weber, Lilian
Waade, Peter Thestrup
Daugaard, Anna Hedvig Møller
Khodadadi, Mojtaba
Mikuš, Nace
Mathys, Chris
contents Bayesian models of cognition have gained considerable traction in computational neuroscience and psychiatry. Their scopes are now expected to expand rapidly to artificial intelligence, providing general inference frameworks to support embodied, adaptable, and energy-efficient autonomous agents. A central theory in this domain is predictive coding, which posits that learning and behaviour are driven by hierarchical probabilistic inferences about the causes of sensory inputs. Biological realism constrains these networks to rely on simple local computations in the form of precision-weighted predictions and prediction errors. This can make this framework highly efficient, but its implementation comes with unique challenges on the software development side. Embedding such models in standard neural network libraries often becomes limiting, as these libraries' compilation and differentiation backends can force a conceptual separation between optimization algorithms and the systems being optimized. This critically departs from other biological principles such as self-monitoring, self-organisation, cellular growth and functional plasticity. In this paper, we introduce \texttt{pyhgf}: a Python package backed by JAX and Rust for creating, manipulating and sampling dynamic networks for predictive coding. We improve over other frameworks by enclosing the network components as transparent, modular and malleable variables in the message-passing steps. The resulting graphs can implement arbitrary computational complexities as beliefs propagation. But the transparency of core variables can also translate into inference processes that leverage self-organisation principles, and express structure learning, meta-learning or causal discovery as the consequence of network structural adaptation to surprising inputs. The code, tutorials and documentation are hosted at: https://github.com/ilabcode/pyhgf.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09206
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle pyhgf: A neural network library for predictive coding
Legrand, Nicolas
Weber, Lilian
Waade, Peter Thestrup
Daugaard, Anna Hedvig Møller
Khodadadi, Mojtaba
Mikuš, Nace
Mathys, Chris
Neural and Evolutionary Computing
Artificial Intelligence
Machine Learning
Neurons and Cognition
Bayesian models of cognition have gained considerable traction in computational neuroscience and psychiatry. Their scopes are now expected to expand rapidly to artificial intelligence, providing general inference frameworks to support embodied, adaptable, and energy-efficient autonomous agents. A central theory in this domain is predictive coding, which posits that learning and behaviour are driven by hierarchical probabilistic inferences about the causes of sensory inputs. Biological realism constrains these networks to rely on simple local computations in the form of precision-weighted predictions and prediction errors. This can make this framework highly efficient, but its implementation comes with unique challenges on the software development side. Embedding such models in standard neural network libraries often becomes limiting, as these libraries' compilation and differentiation backends can force a conceptual separation between optimization algorithms and the systems being optimized. This critically departs from other biological principles such as self-monitoring, self-organisation, cellular growth and functional plasticity. In this paper, we introduce \texttt{pyhgf}: a Python package backed by JAX and Rust for creating, manipulating and sampling dynamic networks for predictive coding. We improve over other frameworks by enclosing the network components as transparent, modular and malleable variables in the message-passing steps. The resulting graphs can implement arbitrary computational complexities as beliefs propagation. But the transparency of core variables can also translate into inference processes that leverage self-organisation principles, and express structure learning, meta-learning or causal discovery as the consequence of network structural adaptation to surprising inputs. The code, tutorials and documentation are hosted at: https://github.com/ilabcode/pyhgf.
title pyhgf: A neural network library for predictive coding
topic Neural and Evolutionary Computing
Artificial Intelligence
Machine Learning
Neurons and Cognition
url https://arxiv.org/abs/2410.09206