AI Navigate

Exploring the Dimensions of a Variational Neuron

arXiv cs.LG / 3/17/2026

📰 NewsIdeas & Deep AnalysisModels & Research

Key Points

  • The paper introduces EVE (Elemental Variational Expanse), a variational distributional neuron with an explicit prior, an amortized posterior, and unit-level variational regularization.
  • It relocates probabilistic structure to the neuron level, enabling the neuron to be locally observable and controllable rather than relying on global latent variables.
  • The study explores how changing the neuron's latent dimensionality k (from 1 to higher dimensions) interacts with local capacity control and a neuron-level autoregressive extension, supported by diagnostics like effective KL and drift indicators.
  • Across forecasting and tabular tasks, the work shows some neuron-level variables are measurable and predictive of downstream behavior, offering an initial map of the design space for variational neurons.

Abstract

We introduce EVE (Elemental Variational Expanse), a variational distributional neuron formulated as a local probabilistic computational unit with an explicit prior, an amortized posterior, and unit-level variational regularization. In most modern architectures, uncertainty is modeled through global latent variables or parameter uncertainty, while the computational unit itself remains scalar. EVE instead relocates probabilistic structure to the neuron level, making it locally observable and controllable. In this paper, the term dimensions refers primarily to the neuron's internal latent dimensionality, denoted by k. We study how varying k, from the atomic case k = 1 to higher-dimensional latent spaces, changes the neuron's learned operating regime. We then examine how this main axis interacts with two additional structural properties: local capacity control and temporal persistence through a neuron-level autoregressive extension. To support this study, EVE is instrumented with internal diagnostics and constraints, including effective KL, a target band on mu^2, out-of-band fractions, and indicators of drift and collapse. Across selected forecasting and tabular settings, we show that latent dimensionality, control, and temporal extension shape the neuron's internal regime, and that some neuron-level variables are measurable, informative, and related to downstream behavior. Overall, the paper provides an experimentally grounded first map of the design space opened by a variational neuron.