HGFX: a validated Python/JAX reproduction of the Hierarchical Gaussian Filter toolbox
Ahmadkhanloo, M.
Show abstract
The Hierarchical Gaussian Filter (HGF) is a hierarchical Bayesian model of learning under uncertainty and volatility, with a widely used MATLAB implementation. We present HGFX 1.0.0, a Python/JAX toolbox whose primary objective is validated behavioral and numerical compatibility with a frozen HGF Toolbox 8.2.0 reference, while removing MATLAB as a user-runtime dependency. Reimplementation is treated as a validation problem rather than source translation: configuration semantics, parameter transforms, forward trajectories, observation likelihoods, objectives, fitting and statistical surfaces, simulation workflows, official demo behavior, model selection, and backend agreement are compared against the pinned oracle under predeclared tolerances. Two official MATLAB demo workflows reproduce the reference at frozen trajectory tolerances, including a regime in which classic HGF encounters negative posterior precision while eHGF completes successfully. In paired model-selection validation, all 36 BIC winners agree between MATLAB and HGFX. Physical NVIDIA GPU applicability was confirmed on two Tesla T4 devices, with a maximum CPU-versus-GPU final-objective difference of 1.42e-14 against a frozen 1e-7 criterion. Historical parameter recovery results that did not meet predeclared criteria, together with exact MATLAB/HGFX limitation matches, are preserved and are not reclassified as scientific success. A prospectively gated comparison with pyhgf 0.3.2 shows binary64-scale agreement on mapped perceptual trajectories in one authorized three-level binary-HGF cell, while participant-response negative log-likelihood is retained as not directly comparable. HGFX therefore provides a MATLAB-independent Python implementation with an explicit evidence model that separates direct parity, matched reference limitations, backend applicability, and future performance claims.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Modernizing the NEURON Simulator for Sustainability, Portability, and Performance 93%
- A scalable implementation of the recursive least-squares algorithm for training spiking neural networks 93%
- Efficient parameter calibration and real-time simulation of large scale spiking neural networks with GeNN and NEST 93%
Similar papers in this journal
- Compressive Big Data Analytics: An Ensemble Meta-Algorithm for High-dimensional Multisource Datasets 93%
- CBGTPy: An extensible cortico-basal ganglia-thalamic framework for modeling biological decision making 92%
- Small hand-designed convolutional neural networks outperform transfer learning in automated cell shape detection in confluent tissues 92%
Similar papers in this journal
- Class imbalance should not throw you off balance: Choosing the right classifiers and performance metrics for brain decoding with imbalanced data 92%
- Post-hoc modification of linear models: combining machine learning with domain information to make solid inferences from noisy data 91%
- Mixtures of large-scale dynamic functional brain network modes 91%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.