Source-linked AI summary
Connectome-to-Function: Conditional Generative Latent Representations for Reservoir Computing
Zhuolin Yu, Xingyu Liu, Yuanhao Jia, Yunhang Xiao, Hairuo Xue, Feihan Sun, Guozhang Chen
TL;DR
Mapping connectome structure to computation remains difficult, and existing approaches do not provide a generative representation that also organizes functional variation. This paper learns a conditional latent representation from biological connectivity and finds that its coordinates predict reservoir performance, with task-specific structural associations.
Problem
It remains difficult to represent connectomes in a generative form that also relates biological wiring to computational function.
Source: Introduction · p. 1Method
A conditional generative framework encodes connectome structure using neuronal cell type and spatial organization as conditions for reconstruction and generation.
Source: Introduction · p. 2, Conditional VAE for Connectome · p. 3Results
The latent space predicts reservoir-performance variation across memory, dynamic prediction, and classification, with memory associated with reciprocal connectivity and prediction and classification with spectral properties.
Source: Conclusion · p. 8Takeaways & Limitations
Sparse connectome graphs can be organized into a latent space connecting circuit topology with measured computational behavior.
Source: Conclusion · p. 8Experiments use local cortical microcircuits from MICrONS, and generalization to other brain regions, species, or spatial scales remains unestablished.
Source: Conclusion · p. 8
Abstract
from arXiv · showhide
Connectomes, graph-level maps of neurons and their synaptic connections, provide a structural basis for understanding how brain circuits support function and computation. However, mapping connectome structure to computation remains difficult because these graphs are high-dimensional, sparse, and sensitive to local structural variation. Existing approaches often depend on hand-crafted structural descriptors or task-specific predictors, which limits their ability to represent connectomes in a form that is both generative and functionally meaningful. We propose a conditional generative latent framework that encodes connectome graphs into a compact structural space while using available node-level conditions to guide reconstruction and generation. From this space, the model can reconstruct observed connectivity with a mean edge-reconstruction AUC up to 0.910 and generate new candidate connectomes, enabling a unified analysis of graph structure and computational behavior. Using connectome-derived graphs as recurrent computational substrates, we found that the learned latent space captures functional variation across reservoir-computing experiments, with cross-validated $R^2$ values up to approximately 0.87. Interpretability analysis further revealed task-specific structural mechanisms: in our examples, memory performance is associated with reciprocal recurrent connectivity, whereas prediction and classification are more strongly associated with spectral properties of the recurrent network. These findings suggest an AI-for-science approach to linking neural connectivity to computation and provide a generative and interpretable basis for studying how distinct structural mechanisms shape computational capacity.
Introduction
The paper asks whether measured connectome structure can be organized into a continuous representation that helps explain variation in circuit computation. It proposes a conditional generative latent framework and evaluates whether its coordinates relate to reservoir performance across tasks.
Introduction
- A key gap is a generative representation learned from biological connectivity alone that can reveal computational organization without using function to train the representation.Prior work either emphasizes structural reproduction, task-trained artificial networks, or fixed connectomes and rewired variants.Source: Introduction · p. 2
- The framework conditions connectome generation on neuronal cell type and spatial organization, supporting reconstruction and controlled sampling of candidate connectomes.It encodes local circuits into compact structural coordinates.Source: Introduction · p. 2
- The learned latent coordinates predict reservoir performance across memory, dynamic prediction, and classification tasks.The reported cross-validated R2 values are approximately 0.46–0.87.Source: Introduction · p. 2
- Interpretability analysis associates memory with reciprocal recurrent connectivity, while prediction and classification are more strongly associated with recurrent-network spectral properties.These task-specific mechanisms link functional directions in the learned space to distinct structural descriptors.Source: Introduction · p. 2
Related Work
The paper situates its approach between descriptive graph statistics, mechanistic graph generators, and connectome-based reservoir studies. It then describes a conditional graph representation and tests its structural fidelity and relation to reservoir function.
Related work
- Existing descriptors are predefined summaries, mechanistic generators lack continuous data-driven latent spaces, and general graph generators are difficult to apply to sparse neural microcircuits with spatial and cell-identity features.Prior connectome-reservoir studies mainly evaluate fixed empirical networks or simple rewired variants rather than learning a generative latent space.Source: Related Work · p. 2
Problem setup
- The model represents each local circuit as an attributed directed graph, with node annotations comprising neuronal soma coordinates and one-hot neuron types.The latent variable encodes graph structure conditioned on observed node features.Source: Conditional VAE for Connectome · p. 3
Architecture
- The encoder combines spatial and node attributes with adjacency information, then uses a permutation-invariant graph-level representation to parameterize the latent distribution.PointNet++ fuses geometric features, a GAT integrates adjacency, and a Transformer-based encoder uses a learned CLS token.Source: Conditional VAE for Connectome · p. 4
- The decoder combines node-condition embeddings with the latent code and predicts directed edge probabilities that are sampled into binary adjacency matrices.Decoded representations are used to generate candidate connectomes under node-level conditions.Source: Conditional VAE for Connectome · p. 4
Connectome-based reservoir computing
- Connectome graphs are evaluated as fixed Dale-signed reservoirs, with recurrent connectivity normalized to a fixed spectral radius and only the readout fitted to task targets.The evaluation uses recurrent state updates and linear readout; the recurrent matrix and random input weights remain fixed during training.Source: Linking Latent Structure to Function · p. 4
Latent-to-function regression
- To test latent-to-function prediction, the analysis decodes latent codes under shared node conditions and evaluates generated graphs as reservoirs; reliable cross-validated prediction is taken as evidence that the representation contains function-informative structural variation.The shared condition template is repeated across 10 choices, and the functional score averages 10 generated adjacency matrices across 3 random seeds per latent code.Source: Linking Latent Structure to Function · p. 4
Reconstruction and latent organization
- Figure 3 presents representative original, naive-VAE, and full-CVAE adjacency matrices alongside a PCA view of latent vectors colored by graph descriptors.Table 1 compares full-CVAE and naive-VAE reconstruction using edge-level AUC and graph-topology relative errors; the supplied caption does not give cell values.Source: Figure 3 · p. 5, Table 1 · p. 5
Experiments
Experiments test how connectome graph structure is represented and how latent variation relates to reservoir performance across three tasks. They report topology preservation and task-specific structural associations.
- Experiments use MICrONS mouse-cortex local cylindrical circuits (100–330 nodes, padded to 330), split spatially, and test copy memory, Lorenz prediction, and sequential MNIST.Source: Experimental Setup · p. 5, Connectome Latents Predict Reservoir… · p. 6
- The full CVAE reduces relative errors in modularity, transitivity, and clustering coefficient, retaining higher-order modular and transitive structure beyond marginal edge density.Source: Connectome Reconstruction and Latent… · pp. 5–6
- Across 498 samples, the first two principal components explain 56.13% of variance, while clustering coefficient and mean degree change smoothly across the PC1–PC2 plane.Source: Connectome Reconstruction and Latent… · p. 6
- Sequential MNIST reaches best cross-validated R2 of approximately 0.87, versus Lorenz system prediction at approximately 0.46; Ridge is close to the best nonlinear models across tasks.Source: Table 3 · p. 6, Connectome Latents Predict Reservoir… · pp. 6–7, Figure 4 · p. 6
- Interpretability projects standardized latent vectors onto task-specific Ridge gradients, then tests descriptors as partial mediators through their associations and attenuation after adjustment.Source: Interpreting the Latent Space · p. 7
- Copy memory is associated with mixed excitatory-inhibitory reciprocal connections and length-5 closed walks, while prediction and classification are linked to distinct spectral features; the analysis is observational.Source: Task-specific Structural Mechanisms · pp. 7–8
Conclusion
The framework links compact connectome representations to reservoir-computing performance and task-specific structural associations. The experiments are limited to MICrONS local cortical microcircuits and reservoir computing as a proxy for circuit computation.
Conclusion
- The latent space predicts functional variation across memory, dynamic prediction, and classification, with reciprocal-core associations for memory and Dale-signed spectral associations for prediction and classification.The memory-associated cores are mixed excitatory–inhibitory; the spectral properties are global.Source: Conclusion · p. 8
- The framework learns compact connectome latent representations conditioned on neuronal cell type and spatial organization.Generated connectomes are used as recurrent reservoirs to evaluate computational behavior.Source: Conclusion · p. 8
- The experiments use MICrONS local cortical microcircuits, and generalization to other brain regions, species, or spatial scales remains unestablished.Reservoir computing is a controlled, interpretable proxy that does not capture the full range of biological neural dynamics or learning mechanisms; synthesized circuits have not been experimentally validated.Source: Conclusion · p. 8