A Sensing Whole Brain Zebrafish Foundation Model for Neuron Dynamics and Behavior

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Vegas, Sam Fatehmanesh, Thomson, Matt, Gornet, James, Prober, David
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917052721135616
author Vegas, Sam Fatehmanesh
Thomson, Matt
Gornet, James
Prober, David
author_facet Vegas, Sam Fatehmanesh
Thomson, Matt
Gornet, James
Prober, David
contents Neural dynamics underlie behaviors from memory to sleep, yet identifying mechanisms for higher-order phenomena (e.g., social interaction) is experimentally challenging. Existing whole-brain models often fail to scale to single-neuron resolution, omit behavioral readouts, or rely on PCA/conv pipelines that miss long-range, non-linear interactions. We introduce a sparse-attention whole-brain foundation model (SBM) for larval zebrafish that forecasts neuron spike probabilities conditioned on sensory stimuli and links brain state to behavior. SBM factorizes attention across neurons and along time, enabling whole-brain scale and interpretability. On a held-out subject, it achieves mean absolute error <0.02 with calibrated predictions and stable autoregressive rollouts. Coupled to a permutation-invariant behavior head, SBM enables gradient-based synthesis of neural patterns that elicit target behaviors. This framework supports rapid, behavior-grounded exploration of complex neural phenomena.
format Preprint
id arxiv_https___arxiv_org_abs_2510_27366
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Sensing Whole Brain Zebrafish Foundation Model for Neuron Dynamics and Behavior
Vegas, Sam Fatehmanesh
Thomson, Matt
Gornet, James
Prober, David
Neurons and Cognition
Neural dynamics underlie behaviors from memory to sleep, yet identifying mechanisms for higher-order phenomena (e.g., social interaction) is experimentally challenging. Existing whole-brain models often fail to scale to single-neuron resolution, omit behavioral readouts, or rely on PCA/conv pipelines that miss long-range, non-linear interactions. We introduce a sparse-attention whole-brain foundation model (SBM) for larval zebrafish that forecasts neuron spike probabilities conditioned on sensory stimuli and links brain state to behavior. SBM factorizes attention across neurons and along time, enabling whole-brain scale and interpretability. On a held-out subject, it achieves mean absolute error <0.02 with calibrated predictions and stable autoregressive rollouts. Coupled to a permutation-invariant behavior head, SBM enables gradient-based synthesis of neural patterns that elicit target behaviors. This framework supports rapid, behavior-grounded exploration of complex neural phenomena.
title A Sensing Whole Brain Zebrafish Foundation Model for Neuron Dynamics and Behavior
topic Neurons and Cognition
url https://arxiv.org/abs/2510.27366