Amortized Inference of Neuron Parameters on Analog Neuromorphic Hardware

· Source: cs.NE updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Emerging Technologies & Innovation, Neuromorphic Computing · Depth: Expert, extended

Summary

This work introduces a non-sequential simulation-based inference (SBI) algorithm to create an amortized neural density estimator (NDE) for approximating the posterior distribution of seven parameters of the adaptive exponential integrate-and-fire (AdEx) neuron model on the analog neuromorphic BrainScaleS-2 (BSS-2) substrate. The researchers constrained the extensive parameter space by training a binary classifier to identify parameter combinations that produce moderate spike counts (1-70 kHz), increasing relevant samples from 2% to 38%. They compared two NDEs: one using handcrafted summary statistics and another incorporating a summary network trained jointly with the NDE. The summary network approach yielded a more focused posterior distribution and generated posterior predictive traces that more accurately captured membrane potential dynamics, whereas handcrafted statistics showed deviations in exact dynamics despite matching included features. The results validate amortized SBI for parameterizing analog neuron circuits, despite observed biases and miscalibration in posteriors.

Key takeaway

For research scientists working with neuromorphic hardware or complex biological models, this study demonstrates that amortized simulation-based inference (SBI) can efficiently parameterize analog neuron circuits. You should prioritize integrating summary networks over handcrafted summary statistics when training neural density estimators (NDEs) to achieve more accurate membrane potential dynamics and focused posterior distributions, especially when dealing with high-dimensional observation data. This approach can significantly reduce the effort in identifying suitable model parameters.

Key insights

Amortized SBI effectively parameterizes analog neuron circuits on neuromorphic hardware, with summary networks outperforming handcrafted statistics.

Principles

Method

A non-sequential SBI algorithm, BayesFlow, was used to train an amortized NDE. A binary classifier pre-filtered parameter space for moderate spike counts. Two NDEs were compared: one with handcrafted summary statistics and another with a jointly trained summary network.

In practice

Topics

Best for: Research Scientist, AI Researcher, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.NE updates on arXiv.org.