Search papers, labs, and topics across Lattice.
We introduce conditional flow matching to learn a generative model for single-neuron electrophysiology, addressing the limitations of deterministic parametrizations and existing ML surrogates that map stimuli to single voltage responses. This approach effectively captures the multimodal nature of neuronal responses, including coexisting firing and non-firing states near threshold and distinct low- and high-firing modes at depolarization block, outperforming a neural operator baseline. The model accurately reproduces electrophysiological feature distributions, spike-time structure, and excitability profiles, matching both biophysically detailed simulations and experimental recordings.
Single-neuron models can now capture multimodal voltage responses, including coexisting firing/non-firing states and depolarization block modes, a feat previously requiring massive ensembles or failing with neural operators.
Neurons of the brain exhibit a rich repertoire of electrophysiology dynamics with the same repeated stimulus eliciting very different voltage responses from the same cell. One common approach in biophysically detailed models is to capture this variability through ensembles of deterministic parametrizations, at a cost of hundreds of thousands of CPU hours. Existing machine learning surrogates inherit the same limitation, where a stimulus is mapped to a single voltage response. We address this challenge by learning a conditional generative model for single-neuron electrophysiology, using flow matching with a velocity field conditioned on the input current. On biophysically detailed models of two human cortical interneuron types, the generated responses closely reproduce the electrophysiological feature distributions, spike-time structure, and excitability profiles, even matching the experimental recordings from the corresponding human cortical neurons. Near the firing threshold, firing and non-firing responses coexist at the same stimulus amplitude, and at high amplitudes, ensembles may split into low- and high-firing modes near depolarization block. We show that our model recovers both modes in each case, while a neural operator baseline suppresses spiking near threshold and blurs the gap between modes at depolarization block.