Automatic discovery of cell types and microcircuitry from neural connectomics
Abstract
Neural connectomics has begun producing massive amounts of data, necessitating new analysis methods to discover the biological and computational structure. It has long been assumed that discovering neuron types and their relation to microcircuitry is crucial to understanding neural function. Here we developed a nonparametric Bayesian technique that identifies neuron types and microcircuitry patterns in connectomics data. It combines the information traditionally used by biologists in a principled and probabilistically coherent manner, including connectivity, cell body location, and the spatial distribution of synapses. We show that the approach recovers known neuron types in the retina and enables predictions of connectivity, better than simpler algorithms. It also can reveal interesting structure in the nervous system of Caenorhabditis elegans and an old manmade microprocessor. Our approach extracts structural meaning from connectomics, enabling new approaches of automatically deriving anatomical insights from these emerging datasets.
https://doi.org/10.7554/eLife.04250.001eLife digest
The human brain is made up of billions of neurons, which are organised into networks via trillions of connections. The study of the nature of these connections will be central to understanding how the brain works. In recent years, a number of new methods for imaging the brain have made it possible to visualise and map these connections, generating striking images and creating an additional field of neuroscience known as ‘connectomics’.
However, the sheer volume of data generated by connectomics is now beginning to exceed the capacity of researchers to analyse it. Just as the advent of genome sequencing required the development of statistical techniques to analyse the resulting data, so the emergence of connectomics has created a need for similarly powerful mathematical models in neuroscience.
Jonas and Kording have developed one such algorithm that can classify the component units of circuits, both biological and manmade, and identify the connections between them. When applied to connectomics data for 950 neurons in the mouse retina, the algorithm generated predictions regarding cell types and patterns of connectivity. The predicted cell types agreed closely with those identified by human neuroanatomists. Results were similarly convincing when the algorithm was applied to the nervous system of the nematode worm and genetic model organism, Caenorhabditis elegans, and even when it was asked to classify electronic components and connectivity patterns in a manmade microprocessor.
Algorithms such as that developed by Jonas and Kording will soon be essential for making sense of the vast quantities of data generated by connectomic studies of the human brain. At present, an analysis of 950 neurons requires several hours, thus refinements that make the process faster will likely be required prior to the analysis of larger human datasets. Such algorithms will open up a range of possibilities for examining the structure of the healthy brain, as well as the changes triggered by developmental abnormalities and disease.
https://doi.org/10.7554/eLife.04250.002Introduction
Emerging connectomics techniques (Zador et al., 2012; Morgan and Lichtman, 2013) promise to quantify the location and connectivity of each neuron within a tissue volume. These massive datasets will far exceed the capacity of neuroanatomists to manually trace small circuits, thus necessitating computational, quantitative, and automatic methods for understanding neural circuit structure. The impact of this kind of highthroughput transition has been seen before—the rise of sequencing techniques necessitated the development of novel computational methods to understand genomic structure, ushering in bioinformatics as an independent discipline (Koboldt et al., 2013).
The brain consists of multiple kinds of neurons, each of which is hypothesized to have a specific role in overall computation. Neuron types differ in many ways, for example, chemical or morphological, but they also differ in the way they connect to one another (Seung and Sümbül, 2014). In fact, the idea of well defined, typedependent local connectivity patterns (microcircuits) has a long history (Passingham, 2002), and is prominent in many areas, from sensory (e.g., retina; Masland, 2001) to processing (e.g., neocortex; Mountcastle, 1997) to movement (e.g., spinal cord; Grillner et al., 2005). These types of repeated computing patterns are a common feature of computing systems, even occurring in manmade computing circuits. It remains an important challenge to develop algorithms to use connectivitybased anatomical data (connectomics) to automatically back out underlying microcircuitry.
The discovery of structure is a crucial aspect of network science. Early approaches focused on global graph properties, such as the types of scaling present in the network (Watts and Strogatz, 1998). While this approach leads to an understanding of the global network, more recent work aims at identifying very smallscale repeat patterns, or motifs, in networks (Milo et al., 2002). These motifs are defined not between different node types, but rather represent repeated patterns of topology.
The discovery of structure in probabilistic graphs is a wellknown problem in machine learning. Commonly used algorithms include communitybased detection methods (Girvan and Newman, 2002) and stochastic block models (Nowicki and Snijders, 2001). While these approaches can incorporate the probabilistic nature of neural connections (Hill et al., 2012), they do not incorporate the additional richer structure present in connectomics data—the location of cell bodies, the spatial distribution of synapses, and the distances between neurons. It is of particular importance that the probability of connections has a strong spatial component, a factor that is hard to reconcile with many other methods. A model attempting to fully capture the variation in the nervous system should take into account the broad set of available features.
When it comes to neuroscience and other computing systems, we expect patterns of connectivity much more complex than traditional motifs, exhibiting a strong spatial dependence arising from the complex genetic, chemical, and activitybased neural development processes.
To address these challenges, here we describe a Bayesian nonparametric model that can discover circuit structure automatically from connectomics data: the cell types, their spatial patterns of interconnection, and the locations of somata and synapses. We show that by incorporating this additional information, our model both accurately predicts the connection as well as agrees with human neuroanatomists as to the identification of cell types. We take as inspiration previous work on identifying cell types automatically from morphology (Guerra et al., 2011) and electrophysiology (Druckmann et al., 2013).
We primarily focus on the recently released mouse retina connectome (Helmstaedter et al., 2013), but additionally examine the Caenorhabditis elegans connectome (White et al., 1986). Comparing the cell types discovered by the algorithms with those obtained manually by human anatomists reveals a high degree of agreement. We thus present a scalable probabilistic approach to infer microcircuitry from connectomics data available today and in the future.
Model
We build a structured probabilistic model which begins with the generic notion of a cell being a member of a single type—and these types affect soma depth, distribution of synapses, as well as a cell type and distancedependent connection probability. For example, retinal ganglion cells may synapse on nearby, but not far away, amacrine cells, with bipolar cells clearly tessellating space and synapsing on both. In machine learning parlance, our method is unsupervised—it seeks to discover structure in data and make predictions in the absence of training data. Rather than taking in examples of types annotated by human neuroanatomists, we instead start with the weakest possible assumption in an attempt to algorithmically discover this structure. We contrast this with the supervised approaches taken in Guerra et al. (2011), where there is high confidence in the (morphologically defined) types and then a supervised classifier is built, as our goal here is explicit discovery of types.
From these assumptions (priors) we develop a generative Bayesian model that estimates the underlying cell types and how they connect. We take as input (Figure 1A) the connectivity matrix of cells (Figure 1B), a matrix of the distance between cells (Figure 1C), the percell soma depth (Figure 1D), and the depth profile of the cell's synapses (Figure 1E). We perform joint probabilistic inference to automatically learn the number of cell types, which cells belong to which type, their typespecific connectivity, and how connections between types vary with distance. We also simultaneously learn the soma depth associated with each type and the typical synaptic density profile (Figure 1F–H).
We start with a model for connectivity, the iSBM (Kemp et al., 2006; Xu et al., 2006), which has been shown to meaningfully cluster connection graphs while learning the number of hidden groups, or types. We extend this approach by adding distance dependence to model salient aspects of microcircuitry via logistic and exponential distancelink functions. We form a unimodial model of cell body depth and a multimodal synapse density profile model (see ‘Materials and methods’ for mathematical details).
As an illustrative example, consider a network with only three cell types, labeled A, B, and C. Assume these cells are uniformly distributed in space, and that the probability of connection between any two cells, c_{i} and c_{j}, depends only on their type and their distance, according to a logistic (sigmoidal) function. Let A connect only to nearby B and C cells, but B connect to any C regardless of distance. This is the prior intuition our model is designed to capture.
For the basic linkdistance model, we take as input a connectivity matrix R defining the connections between cell e_{i} and e_{j}, as well as a distance function d(e_{i}, e_{j}) representing a (physical) distance between adjacent cells. See the supplemental material for extension to multiple connectivity matrices. We assume there exist an unknown number K of latent (unobserved) cell types, $k\in \left\{1,2,3,\dots ,K\right\}$, and that each cell e_{i} belongs to a single cell type. We indicate a cell e_{i} is of type k using the assignment vector (c), so c_{i} = k. The observed connectivity between two cells R(e_{i}, e_{j}) then depends only on their latent type and their distance through a link function f(⋅, d(e_{i}, e_{j})). We assume f is parameterized based on the latent type, c_{i} = m and c_{j} = n, via a parameter η_{mn}, as well as a set of global hyperparameters θ, such that the link function is f(d(e_{i}, e_{j})η_{mn}, θ).
We then jointly infer the posterior distribution of the class assignment vector (c) = {c_{i}}, the parameter matrix η_{mn}, and the global model hyperparameters θ:
Our subsequent analysis uses both the full posterior distribution over these parameters as well as the most probable, or maximum a posteriori (MAP), estimate.
For the retina data, we then extend the model with the additional features indicated. Cell soma depth is modeled as a celltypedependent Gaussian distribution with latent (unknown) pertype mean and variance. Similarly, each cell has some number N_{i} of synapses, each of which is drawn from a celltypespecific density profile with up to three modes.
Inference is performed via MCMC via three composable transition kernels—one for structural, one for pertype parameters, and one for global parameters and hyperparameters. Details of data preprocessing, inference parameters, and runtime can be found in the 'Methods section.
Metrics
To evaluate the quality of the model fit, we need to use information that quantifies aspects of the data for which we have ground truth information. We focus on two aspects of performance. First, if the model works well, then the probability that a pair of neurons is of the same type should be high if the neurons actually are of the same type. Second, the model should assign a high probability of connection between two cells if they have a connection in the underlying data. We term these two factors clustering accuracy and linkprediction accuracy.
To assess the accuracy of a clustering compared to that determined by neuroanatomists, we employ three metrics—clustering homogeneity, clustering completeness, and the ARI. All metrics equal 1.0 when two clusterings completely agree. Homogeneity reflects the degree to which a found cluster or type contains only a single true type. Completeness measures how much of a true type is contained within a single identified type—a completeness of 1.0 means no true type is split into multiple subtypes. ARI is a metric that reflects both measures (see the supplemental material for more information).
To assess the accuracy of the model for connections, we use link prediction accuracy. If our model accurately captures the true structure of the data, it should be good at predicting if a link exists. We thus train the model on the data with a subset of the links marked as unobserved and thus compute our predictive accuracy. We perform 10way crossvalidation on a given dataset (Guerra et al., 2011), learn the resulting model, and use that model to predict the missing synapses. Each potential link between cells is assigned a probability, and we compute the AUC for the resulting ROC curve. An AUC of 1.0 means that we perfectly predict the presence and absence of the missing synapses. We use link prediction accuracy to quantify how good the model is at discovering the underlying connectivity.
Results
We will first establish that our algorithm works properly and try to understand its properties using simulated data. Subsequently, we will analyze in detail a dataset on the retina. Lastly, we will briefly discuss the analysis of data from the worm C. elegans and from an old manmade microprocessor.
Validation with simulated data where ground truth is known
To validate our model, we performed a series of simulations to test if the model can accurately recover the true underlying network structure and cell type identity. We thus simulate data for which we know the correct structure and compare the estimated structure based on the algorithm (see ‘Materials and methods’) with the one we used for simulation. We find that the model does a good job of recovering the correct number of cell types (Figure 2A), the cell identities (Figure 2B), and the spatial extent of each type (Figure 2C). For comparison, we show the results using the infinite stochastic block model (iSBM) instead (Figure 2A–C, black line) which assumes that only cell type matters, and thus finds small neighborhoods of connected nodes (instead of global connectivity patterns). This contrast shows that while the regular block model can not correctly deal with distancedependent connectivity, our model can. Our model converges relatively quickly (see 'Mixing of Markov chains') to an estimate of the most probable values for the cell types, which is enabled by using a combination of simulated annealing and parallelized Markovchain Monte Carlo (MCMC) (see ‘Materials and methods’ for details). Thus our model at least is promising for application to biological datasets.
Model mismatch
We next analyze how our model performs in cases where the data are generated with assumptions different from ours. To understand the properties of our model, we attempt connectivity inference on four sets of synthetic data. This helps us understand what our model would do if the data do not obey our assumptions.
We thus generate 10 sets of synthetic data from each of four existing models. The distancedependent stochastic block model assumes type depends on distance, the traditional stochastic block model has no notion of distance, the mixed membership block model assumes type is combinatorial, and the latent position cluster model assumes that type is clusteredbutcontinuous.
If the data are sampled from our model, inference according to our model, unsurprisingly, is good by all measures. It correctly estimates the number of cell types, it is good at predicting connectivity (high area under the curve, AUC), it agrees with human classification (Rand index), it discovers all types, and leads to homogeneous estimates (Figure 3, first row). If the data come from a block model without distance dependence, we see that it still does well on all meaningful measures (Figure 3, second row). This is unsurprising, as our model learns the distance dependence, even its absence. For the mixed membership model (Figure 3, third row), the model grossly overestimates the number of types, by basically allocating a type for each combination of memberships. Otherwise, it still performs relatively well. Lastly, for the latent position clustering model (Figure 3, fourth row), the model does poorly. If type is continuous instead of discrete, then our model is basically trying to cover a continuous set with a discrete scenario leading to rather poor performance. However, as we do expect cell types to have a discrete biological basis, we might expect our model to do well with real data.
Sensitivity to edge effects
Connectomic efforts so far have reconstructed only small sections of neural tissue. Consequently, many connections to cells outside that tissue volume will be lost. We are concerned that this selective elimination of connectivity along the boundary might give the appearance of distancedependent connectivity when there is none. We thus performed simulations to check if edge effects could destroy spatial structure and if edge effects could introduce artificial, spurious spatial structure. We measure the degree to which distancedependent effects can arise from selecting regions that are smaller than the ‘scale’ of connectivity (Figure 4). We do this by generating two collections of synthetic datasets—one with distancedependent connectivity and one without. We then in each dataset randomly examine contiguous circular regions with area varying from zero to the entire volume, and empirically calculate the spatial variance in typedependent connectivity. We find that, if there is no distance dependence, edge effects do not artificially introduce distance dependence. However, if the section we are examining is too small, our model can miss the distance dependence. Thus with respect to distancedependent connectivity inference, our model errs on the side of caution. But we also find that for spatial extent that is similar to the currently available datasets, the effects of this are quite limited.
Learning types and circuitry in the retina
The mouse retina (Masland, 2001) is a neural circuit which we expect to have connectivity patterns that are well approximated by our generative model. It is known that there are multiple classes of cells that can be broadly grouped into: ganglion cells that transmit information to the rest of the brain; bipolar cells that connect between different cells; and amacrine cells that feed into the ganglion cells. Recent research (Helmstaedter et al., 2013) has produced a large dataset containing both the types of cells from orthogonal approaches, and also the connectivity matrix between all reconstructed cells (Figure 5A).
The algorithm took 8 hr to perform inference, dividing neurons into a set of cell types which reflect known neuroanatomical distinctions (Figure 5 shows the MAP result). For each pair of neurons there is a specific distancedependent connection probability (Figure 5D), which is well approximated by the model fit. Moreover, each type of cell is rather isotropically distributed across space (Figure 5C) as should be expected for true cell types.
Comparing the results of the algorithm to other information sources allows evaluation of the quality of the type determination. Our types closely reflect the (anatomistdetermined) segmentation of cells into retinal ganglion, narrow amacrine, medium/wide amacrine, and bipolar cells (Figure 6B). We find that the types we find tend to reflect the known laminar distribution in the retina (Figure 6C) as well as the known synaptic density profiles.
The algorithm yields a separation of neurons into a smaller number of types than the fully granular listing of 71 types found by the original authors of the paper (Helmstaedter et al., 2013), although it is still highly correlated with those finer type distinctions (see section ‘Mouse retina’). It is our expectation that, with larger datasets, even closer agreement would be found.
Our fully Bayesian model produces a distribution over probable clusterings. Figure 6 shows this posterior distribution as a cell–cell coassignment matrix, sorted to find maximum block structure. Each large, dark block represents a collection of cells believed with strong probability to be of the same type. When we plot (Figure 6B) the anatomistderived cell types along the left, we can see that each block consists of a roughly homogeneous collection of types.
We evaluate our model along three sets of parameters (Figure 6): how closely does our clustering agree with neuroanatomists' knowledge? Given two cells, how accurately can our model predict the link between them? And how closely does the spatial extent (within a layer) of our identified types agree with the spatial extent of types identified by neuroanatomists?
For our model we show the receiver operating characteristic (ROC) curve (Figure 6D) which shows how the true and false positive rates trade off. We plot the posterior distribution of the area under this curve in Figure 6E. We then plot the posterior distribution for cluster agreement metrics—completeness, homogeneity, and adjusted Rand index (ARI) (Figure 6F). We see that our model tends to overcluster—cells which are of distinct type (at the finest granularity of neuroanatomistidentified type) are grouped as a single type by our model.
We compare linkprediction accuracy across the methods, including our own (Figure 6G, AUC, red). We find that given the dataset, many techniques allow for good linkpredictive accuracy. All the methods allow decent link prediction with an AUC in the 0.9 range. However, our algorithm clearly outperforms the simple statistical models that only use connectivity.
As a second measure we compare linkprediction accuracy across the methods (Figure 6G, ARI, blue). We find that our algorithm far outperforms the controls. We also find that when it is based on more of the same information used by anatomists, then it gets better at agreeing with these anatomists. In particular, using connectivity, distance, synapse distribution, and soma depth leads to the highest ARI. When using the available information, the algorithm produces a good fit to human anatomist judgments.
Finally we look at the spatial extent of the discovered types both within a layer and between layers (Figure 6H). We see that, in the absence of distance information, mere connectivity information results in types which only span a small region of space—essentially local cliques. Incorporation of distance information results in types which span the entire extent of the layer. The depth variance of all models continues to be substantially larger than that predicted by human anatomists—future directions of work include attempting to more strongly encode this prior belief of laminarity.
Recovering spatial connectivity in multiple graphs simultaneously
Having shown our model to work on the repeating tessellated, laminar structure of the mammalian retina, we then apply our model to a structurally very different connectome—the whole body of a small roundworm: C. elegans is a model system in developmental neuroscience (White et al., 1986), with the location and connectivity of each of 302 neurons developmentally determined, leading to early measurement of the connectome. Unlike the retina, only the motor neurons in C. elegans exhibit regular distribution in space—along the body axis. Most interneurons are concentrated in various ganglia that project throughout the entire animal, and the sensory neurons are primarily located in a small number of anterior ganglia. C. elegans also differs from the retina in that the measured connectome is actually two separate graphs—one of directed chemical synapses and another of undirected electrical synapses. As this is a very different connectome, it allows an interesting generalization test: how well will our model work on such a distinct dataset?
Using both the chemical and electrical connectivity (see ‘Materials and methods’), we determined the underlying cell types explained by connectivity and distance (Figure 7A). A superficial inspection of the results shows clustering into groups consisting roughly homogeneously of motor neurons, sensory neurons, and interneurons. Closer examination reveals agreement with the classifications originally outlined by White in 1986 (White et al., 1986).
Note our clustering does not perfectly reflect known divisions—several combinations of head and sensory neurons are combined, and a difficulttoexplain group of mostly VB and DB motor neuron types, with VC split between various groups. Our identified cell types thus reflect a ‘coarsening’ of known types, based entirely on connectivity and distance information, even when the organism exhibits substantially less spatial regularity than the retina.
Types and connectivity in artificial structures
To show the applicability of our method to other connectomestyle datasets, we obtained the spatial location and interconnectivity of the transistors in a classic microprocessor, the MOS Technology 6502 (used in the Apple II) (James et al., 2010). Computer architects use common patterns of transistors when designing circuits, with each transistor having a ‘type’ in the circuit. We identified a region of the processor with complex but known structure containing the primary 8bit registers X, Y, and S (Figure 8).
Our algorithm identifies areas of spatial homogeneity that mirror the known structure in the underlying architecture of the circuit, segmenting transistor types recognizable to computer architects. Using the original schematics, we see that one identified type contains the ‘clocked’ transistors, which retain digital state. Two other types contain transistors with pins C1 or C2 connected to ground, mostly serving as inverters. An additional identified type controls the behavior of the three registers of interest (X, Y, and S) with respect to the SB data bus, either allowing them to latch or drive data from the bus. The repeat patterns of spatial connectivity are visible in Figure 8C, showing the manmade horizontal and vertical layout of the same types of transistors.
Discussion
We have presented a machine learning technique that allows cell types and microcircuitry to be discovered from connectomics data. We have shown its applicability to regularly structured laminar neural circuits like the retina, as well as a less structured whole neuronal organism (C. elegans) and a classic processor. When compared to existing methods, we show how the incorporation of all of this data yields results that combine both high linkprediction accuracy and high agreement with human anatomists. We have found that combining the available data types allows us to discover cell types and microcircuitry that were known to exist in the systems based on decades of previous research and allows good prediction of connectivity.
For our probabilistic models, no known solution exists to exactly find the most probable parsing of the neurons into cell types and connectivity patterns. We employ a collection of MCMC techniques (see ‘Materials and methods’), but while different initializations converge to similar ultimate values, we can never realistically obtain the global optimum. There are a broad range of techniques that may offer good approximations to the global optimum and future work could adapt them to find more precise solutions to our problem.
For our probabilistic model, inference becomes slower as the amount of data increases. Our algorithm required several hours for 1000 neurons. Scaling this class of probabilistic model is an active area of research, and recent results in both variational methods (Hoffman et al., 2013) and spectral learning (Anandkumar et al., 2012) and future work could adapt them to find faster approximate solutions to our problem.
Larger datasets will allow algorithms to distinguish more distinct types and we expect closer agreement with existing anatomical knowledge as more data become available. Moreover, in general, for such problems precision increases with the size of the dataset and the cells that we have are not sufficient to statistically distinguish all the cell types known in anatomy (such as the ∼70 in the retina). Still, using only connectivity and distance, it is possible to meaningfully divide neurons into types.
Our small collection of handselected distancedependent likelihood functions is clearly nonexhaustive, and assumes monotonicity of connectivity probability—for a given class, closer cells are never less likely to connect. This is known to be insufficient for various neural systems. Future models could incorporate a wider variety of likelihood functions, or even learn the global functional form from the data.
There are a range of previous approaches to the discovery of neural microcircuitry (Mountcastle, 1957; Douglas and Martin, 1991; Freund and Buzsáki, 1998; Barthó et al., 2004). These generally involve a great deal of manual labor and ad hoc determination of what constitutes a type of cell—to this day there are disagreements in the literature as to the true types in the mammalian retina. Much as phylogenomics has changed our understanding of animal ontologies, modern large scale data will allow the efficient unbiased discovery of cell types and circuits. The sheer amount of available data demands the introduction of algorithmic approaches.
The development of automatic identification and quantification of cell type may also provide a new computational phenotype for quantifying the effect of disease, genetic interventions, and developmentally experienced neural activity. Our method can in principle identify neuron types across nonconnected graphs, for example, across animals. For example, the types of neurons in one animal can be associated with the types of neurons in another animal, in the same way as this is already possible through molecular markers (Brown and Hestrin, 2009). This could be particularly important if cell types appear that are due to properties of the stimuli and experience as opposed to just the molecular properties of cells, such as color and orientation selective types in primary visual cortex (Lennie and Movshon, 2005; Sincich and Horton, 2005). This would allow comparative quantitative anatomy across animals, and aid the search for the ultimate causes of connectivity.
Our model combines connectivity, cellular and synaptic properties, and suggests the way towards combining even richer data. Distinct cell types differ in morphology, connectivity, transcriptomics, relation to behavior or stimuli, and many other ways. Algorithms combining these data and type information may allow us to synthesize all the available information from one experiment or even across experiments into a joint model of brain structure and function.
Our work shows how rich probabilistic models can contribute to computational neuroanatomy. Eventually, algorithms will have to become a central tool for anatomists, as it will progressively become impossible for humans to parse the huge datasets. This transition may follow a similar transition to that of molecular biology (with genefinding algorithms) and evolutionary biology (with computational phylogenetics). Ultimately, computational approaches may help resolve the significant disagreements across human anatomists.
Methods
Probabilistic model
Request a detailed protocolOur model is a extension of the iSBM (Kemp et al., 2006; Xu et al., 2006) to incorporate spatial relations between entities, inspired by attempts to extend these models with arbitrary discriminative functions (Murphy, 2012).
We take as input a connectivity matrix R defining the connections between cell e_{i} and e_{j}, as well as a distance function d(e_{i}, e_{j}) representing a (physical) distance between adjacent cells. See the supplemental material for extension to multiple connectivity matrices. We assume there exist an unknown number K of latent (unobserved) cell types, $k\in \left\{1,2,3,\dots ,K\right\}$, and that each cell e_{i} belongs to a single cell type. We indicate a cell e_{i} is of type k using the assignment vector (c), so c_{i} = k. The observed connectivity between two cells R(e_{i}, e_{j}) then depends only on their latent type and their distance through a link function f(⋅, d(e_{i}, e_{j})). We assume f is parameterized based on the latent type, c_{i} = m and c_{j} = n, via a parameter η_{mn}, as well as a set of global hyperparameters θ, such that the link function is f(d(e_{i}, e_{j})η_{mn}, θ).
We then jointly infer the MAP estimate of the class assignment vector (c) = {c_{i}}, the parameter matrix η_{mn}, and the global model hyperparameters θ:
We describe the spatial ‘logisticdistance Bernoulli’ function here, and others in the supplemental material.
The ‘logisticdistance Bernoulli’ spatial model assumes that, if cell e_{i} is of type m and cell e_{j} is of type n, then η_{mn} = (μ_{mn}, λ_{mn}), and the probability that two cells e_{i} and e_{j} are connected is given by
where p_{max} and p_{min} are global pergraph parameters.
We place exponential priors on the latent parameters:
using λ^{hp} and μ^{hp} as global pergraph hyperparameters.
We use a Dirichletprocess prior on class assignments, which allows the number of classes to be determined automatically. In brief, for N total cells, the probability of a cell belonging to a class is proportional to the number of data points already in that class, N_{k}, such that $p({c}_{i}=k)\propto \frac{{m}_{k}}{N+\alpha}$ and the probability of the cell belonging to a new class k′ is $p({c}_{i}=k\prime )\propto \frac{\alpha}{N+\alpha}$. α is the global concentration parameter—larger values of α make the model more likely to propose new classes. We grid the parameter α and allow the best value to be learned from the data.
Where we model cell depth, we assume that each cell type has a typical depth, and thus a Gaussian distribution of s_{i}. We assume ${s}_{i}\sim N\left({\mu}_{k}^{\left(s\right)},{\sigma}_{k}^{2\left(s\right)}\right)$, where the (s) superscript indicates these model parameters are associated with the somadepth portion of our model. We use a conjugate prior for $\left({\mu}_{k}^{\left(s\right)},{\sigma}_{k}^{2\left(s\right)}\right)$ with ${\mu}_{k}^{\left(s\right)}\sim N\left({\mu}_{hp}^{\left(s\right)},{\sigma}_{k}^{2\left(s\right)}/{\kappa}_{hp}^{\left(s\right)}\right)$ and ${\sigma}_{k}^{2\left(s\right)}\sim {\chi}^{1}({\sigma}_{hp}^{2\left(s\right)},{\nu}_{hp}^{\left(s\right)}$. The use of conjugacy simplifies inference while allowing for each cell type to have its own depth mean and distribution.
We model synapse depth profile that each cell type has a characteristic depth distribution of synaptic contact points, and mixture of Gaussian distributions over cell is N_{i} contact points, g^{i}. We do this by assuming the ${g}_{j}^{i}$ are drawn from an M = 3component mixture of Gaussians. Thus associated with each cell type k is a vector of M Gaussian means $\left({\mu}_{k,1}^{g},\cdots ,{\mu}_{k,M}^{g}\right)$, and a mixture vector π_{k}. This representation can thus model depth distributions of contact points that have up to three modes, an assumption that is well matched in the bulk of anatomical studies of celltypedependent connectivity.
Inference
Request a detailed protocolWe perform posterior inference via MCMC, annealing on the global likelihood during the traditional burnin phase. MCMC transition kernels for different parts of the state space can be chained together to construct a kernel whose ergodic distribution is the target ergodic distribution over the entire state space.
Our first transition kernel (‘structural’) performs Gibbs sampling of the assignment vector $p\left(\mathbf{c}\eta ,\theta ,\alpha \right)$. The lack of conjugacy in our likelihood model makes an explicit evaluation of the conditional assignment probabilities impossible, motivating us to use an auxiliary variable method (Neal, 2000) in which a collection of ephemeral classes is explicitly represented for the duration of the Gibbs scan.
We then employ a transition kernel to update the percomponent parameter values η_{mn}. Conditioned on the assignment vector c and the model hyperparameters θ, α the individual η_{mn} are independent. We slice sample (Neal, 2003) each component's parameters, choosing the slice width as a function of the global hyperparameter range.
The global hyperparameters, both α and θ, are allowed to take on a discrete set of possible values. As θ is often a tuple of possible values, we explore the Cartesian product of all possible values. We then Gibbs sample (our final transition kernel), which is always possible in a small, finite, discrete state space.
We chain these three kernels together, and then globally anneal on the likelihood from a temperature of T = 64 down to T = 1 over 900 iterations unless otherwise indicated, and then run the chain for another 100 iterations. We then generate at least 20 samples, each taken from the end of a single Markov chain initialized from different random initial points in the state space. For visualization we pick the chain with the highest log likelihood, but for all numerical comparisons (including link probability and cluster accuracy) we use this full collection of samples from the posterior distribution to estimate the resulting statistics.
Link prediction
Request a detailed protocolTo compute linkprediction accuracy, we compute the probability of a link between two cells using each model, trained via 10fold crossvalidation. We use a full collection of posterior samples when computing the link probability, and then compute the area under the ROC curve for each.
We compare our model with a standard network clustering model, the latentposition clustering model. This model assumes each cell belongs to one of K clusters, and each cluster is associated with a ddimensional Gaussian distribution. The probability of a link is then a function of the distance between the data points in this continuous space. We use a variational implementation provided in R (SalterTownshend and Murphy, 2013), parametrically varying the number of latent dimensions and the number of requested groups. While this model provides reasonable linkpredictive accuracy, the clusterings dramatically disagree with those from human anatomists.
Parameters
Hierarchical generative models can be sensitive to hyperparameter settings, thus for most hyperparameters we perform inference. In cases where we cannot, we run separate collections of Markov chains at separate settings and show the results across all pooled parameters. For the case of the mouse retina data, we consider maximum link probability ${p}_{max}\in \left\{0.95,0.9,0.7\right\}$, variance scales for the synapse density profile of ${\sigma}^{2}\in \left\{0.01,0.1,1.0\right\}$ (of normalized depth), and $K\in \left\{2,3\right\}$ possible synapse density profile mixture components. For the connectivitydistanceonly model we actually perform inference over both p_{max} and p_{min}.
Mixing of our Markov chains
Request a detailed protocolEvaluating whether or not approximate inference methods, such as MCMC, produce samples which are valid approximations of the posterior distribution is an ongoing area of research in the computational statistics community. We use a rough proxy here—synthetic likelihood evaluation. For synthetic datasets of sizes comparable to our real data size, do we recover known ground truth information after running our Markov chains for the appropriate amount of time?
Figures 9 and 10 show the cluster accuracy (ARI) to ground truth and the total log score as a function of runtime. We see dramatic changes in log score initially as we vary the temperature, stabilizing as runtime progresses, for each chain. Then we see the characteristic jumps between nearby modes towards the end of the run, in both log score and ARI. Importantly, regardless of whether our model over or underestimates the exact posterior variance about the network, we find points in the latent variable space that are both predictive and parsimonious, largely agreeing with the human anatomists and predicting existing connections.
Dataset details
Mouse retina
Request a detailed protocolDense serial electron microscopy of a 114 μm × 80 μm area in the mouse retina by Helmstaedter et al. (2013) yielded a listing of places where neurons come into contact. There were over 1000 cells originally, and we selected the 950 for which the location of the soma could be reconstructed from the provided cell plots (soma locations were not provided by the study's authors in machinereadable form). The result was a matrix of the total synapselike contact area between all pairs of 950 cells. Area was thresholded at 0.1 μm, determined by hand, to yield a 950 × 950 entry matrix that served as input to our algorithm. We measured the distance between cells using the reconstructed soma centers, and used the logisticdistance spatial relation. Hyperprior griddings are shown in the ‘Hyperprior grids and hyperprior inference’ section.
C. elegans
Request a detailed protocolWe obtained the connectome of C. elegans from data published previously (Varshney et al., 2011), and isolated the 279 nonpharyngeal neurons, with a total of 6393 chemical synapses and 890 gap junctions originally cleaned up in Chen et al. (2006). A cell's position was its distance along the anterior–posterior axis normalized between 0 and 1. We used both networks, the chemical network as a directed graph and the electrical network as an undirected graph. We use the synapse counts with the logisticdistance Poisson likelihood, scaling the counts by 4.0 to compensate for the Poisson's overdispersion.
Microprocessor
Request a detailed protocolWe extracted the connection graph for the transistors in the MOS 6502 (James et al., 2010). Each transistor has three terminals (gate, source, drain), but the methods of the original dataset were unable to consistently resolve which of the C1 and C2 terminals were source and drain, leading to ambiguity in our encoding. We identified a region consisting of three registers X, Y, and S via visual inspection and focused our efforts there. We created a total of six connectivity matrices by examining possible terminal pairings. For example, one graph encodes the connectivity between pins $\text{g}$ and ${\text{c}}_{1}.$ We then have, ${R}^{g{c}_{1}}\left({e}_{i},{e}_{j}\right)=1$ if transistor e_{j} and e_{j} are connected via pins g and c_{1}.
Other likelihoods
Request a detailed protocolWe reparameterized the logisticdistance Bernoulli likelihood to better capture the microprocessor data structure. We are explicitly setting the maximum probability p of the logistic function on a percomponent basis, drawing from a global $p\sim \text{Beta}\left({\alpha}_{hp},{\beta}_{hp}\right)$. Then λ is set for each component as a global hyperparameter, λ.
The ‘logisticdistance Poisson’ spatial model is used to explicitly model the count of synapses, c, between two neurons. The probability of c synapses between two neurons is distributed $c\sim \text{Poisson}\left(cr\right)$, where r (the ‘rate’) is generated by a scaled logistic function (the logistic function has range [0, 1]). For each component η_{mn} we learn both the threshold μ_{mn} and the rate scaling factor r_{mn}. Thus if cells m and n are likely to have on average 20 synapses if they are closer than 5 μm, then μ_{mn} = 5 and r_{mn} = 20.
Thus the probability of R(e_{i}, e_{j}) = c synapses between two cells e_{i} and e_{j} is given by:
where λ and r_{min} are pergraph parameters and we have percomponent parameters ${\mu}_{mn}\sim \text{Exp}\left(\mu {\mu}^{hp}\right)$ and ${r}_{mn}\sim \text{Exp}\left({r}_{mn}{r}_{scale}^{hp}\right)$.
Source code and data
Request a detailed protocolAll source code and materials for running experiments can be obtained from the project website, at http://ericmjonas.github.io/connectodiscovery/.
All preprocessed data has been made publically available as well.
Extension to multiple graphs
Request a detailed protocolThe model can handle multiple graphs R^{q} simultaneously with a shared clustering by extending the likelihood to include the product of the likelihoods of the individual graphs.
Hyperprior grids and hyperprior inference
Request a detailed protocolFor the mouse retina logisticdistance Bernoulli model, we gridded μ^{hp} and λ^{hp} into 40 log_{10}spaced points 1.0 and 80.
For the C. elegans data with the logisticdistance Poisson model, we gridded μ_{hp} and λ into 20 log_{10}spaced points between 0.2 and 2.0, and the ratescale^{hp} parameter into 20 log_{10}spaced points between 2.0 and 20.0. We globally set rate_{min} = 0.01.
For the microprocessor with the logisticdistance with fixed lambda parameter and Bernoulli likelihood, we gridded mu_{hp} into 50 log_{10}spaced points between 10 and 500 and set λ = μ_{hp}/10. ${p}_{min}\in \left\{0.001,0.01,0.02\right\}$ and both p_{α} and ${p}_{\beta}\in \left\{0.1,1.0,2.0\right\}$.
Measuring clustering similarity
Request a detailed protocolWe compare discovered types to known types via cluster comparison metrics: cluster homogeneity, cluster completeness, and the ARI. Homogeneity measures how many true types are in a given found type. If every cell type is split into two types, each subtype is still completely homogeneous. Completeness measures how many members of a given true type are split across found types.
ARI takes into account both effects (Hubert and Arabie, 1985)—two identical clusterings have an ARI of 1.0, while progressively more dissimilar clusters have lower ARIs, becoming slightly negative as the clustering gets anticorrelated.
Figure 11 shows the result of taking 20 different clusters and moving data points between them according to the following operations.
•distribute: take a class and distribute its elements uniformly among the remaining types.
•merge: take a type and merge it into another existing type.
•split: take a type and split it into two distinct types.
We can see the impact on ARI, completeness, and homogeneity as we perform these operations on more of the original 20 types. In all cases, ‘distribution’ of one type among the others is detrimental to the metric. Splitting impacts completeness but not homogeneity, and merging impacts homogeneity but not completeness.
Data availability

Connectomics datasetsCanonical, cleanedup datasets, publicly available at GitHub, originally published in Varshney et al., 2011 (http://dx.doi.org/10.1371/journal.pcbi.1001066) and Helmstaedter et al., 2013 (http://dx.doi.org/10.1038/nature12346).
References

Tensor decompositions for learning latent variable modelsJournal of Machine Learning Research 15:1–55.

Celltype identity: a key to unlocking the function of neocortical circuitsCurrent Opinion in Neurobiology 19:415–421.https://doi.org/10.1016/j.conb.2009.07.011

Wiring optimization can relate neuronal structure and functionProceedings of the National Academy of Sciences of USA 103:4723–4728.https://doi.org/10.1073/pnas.0506806103

A functional microcircuit for cat visual cortexThe Journal of Physiology 440:735–769.https://doi.org/10.1113/jphysiol.1991.sp018733

Community structure in social and biological networksProceedings of the National Academy of Sciences of USA 99:7821–7826.https://doi.org/10.1073/pnas.122653799

Microcircuits in action from CPGs to neocortexTrends in Neurosciences 28:525–533.https://doi.org/10.1016/j.tins.2005.08.003

Comparison between supervised and unsupervised classifications of neuronal cell types: a case studyDevelopmental Neurobiology 71:71–82.https://doi.org/10.1002/dneu.20809

Statistical connectivity provides a sufficient foundation for specific functional connectivity in neocortical neural microcircuitsProceedings of the National Academy of Sciences of USA 109:E2885–E2894.https://doi.org/10.1073/pnas.1202128109

ACM SIGGRAPH 2010 Talks1, ACM SIGGRAPH 2010 Talks, New York, USA, ACM Press, 10.1145/1837026.1837061.

Proceedings of the 21st National Conference on Artificial Intelligence381–388, Proceedings of the 21st National Conference on Artificial Intelligence, volume 1, Palo Alto, California, AAAI Press, 10.1145/1837026.1837061.

Coding of color and form in the geniculostriate visual pathway (invited review)Journal of the Optical Society of America. A, Optics, Image Science, and Vision 22:2013–2033.https://doi.org/10.1364/JOSAA.22.002013

The fundamental plan of the retinaNature Neuroscience 4:877–886.https://doi.org/10.1038/nn0901877

Modality and topographic properties of single neurons of cat's somatic sensory cortexJournal of Neurophysiology 20:408–434.

Markov chain sampling methods for Dirichlet process mixture modelsJournal of Computational and Graphical Statistics 9:249.

Estimation and prediction for stochastic blockstructuresJournal of the American Statistical Association 96:1077–1087.https://doi.org/10.1198/016214501753208735

The frontal cortex: does size matter?Nature Neuroscience 5:190–192.https://doi.org/10.1038/nn0302190

Variational Bayesian inference for the latent position cluster model for network dataComputational Statistics & Data Analysis 57:661–671.https://doi.org/10.1016/j.csda.2012.08.004

The circuitry of V1 and V2: integration of color, form, and motionAnnual Review of Neuroscience 28:303–326.https://doi.org/10.1146/annurev.neuro.28.061604.135731

Structural properties of the Caenorhabditis elegans neuronal networkPLOS Computational Biology 7:e1001066.https://doi.org/10.1371/journal.pcbi.1001066

The structure of the nervous system of the nematode Caenorhabditis elegansPhilosophical Transactions of the Royal Society of London. Series B, Biological Sciences 314:1–340.https://doi.org/10.1098/rstb.1986.0056

Proceedings of the TwentySecond Conference on Uncertainty in Artificial IntelligenceProceedings of the TwentySecond Conference on Uncertainty in Artificial Intelligence, Arlington, Virginia, AUAI Press.
Article and author information
Author details
Funding
University of California Berkeley (University of California, Berkeley) (AWS in Education grant)
 Eric Jonas
National Science Foundation (NSF CISE Expeditions Award CCF1139158)
 Eric Jonas
Lawrence Berkely National Laboratory (Award 7076018)
 Eric Jonas
Defense Advanced Research Projects Agency (XData Award FA87501220331)
 Eric Jonas
National Institutes of Health (R01NS074044)
 Konrad Kording
National Institutes of Health (R01NS063399)
 Konrad Kording
The funders had no role in study design, data collection and interpretation, or the decision to submit the work for publication.
Acknowledgements
We thank Josh Vogelstein for discussions and reading of the manuscript, Finale DoshiVelez for early discussions on the model, and Erica Peterson, Jonathan Glidden, and Yarden Katz for extensive manuscript review. Funding for compute time was provided by Amazon Web Services ‘AWS in Education’ grants.
Version history
 Received: August 5, 2014
 Accepted: March 24, 2015
 Version of Record published: April 30, 2015 (version 1)
Copyright
© 2015, Jonas and Kording
This article is distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use and redistribution provided that the original author and source are credited.
Metrics

 3,921
 views

 788
 downloads

 45
 citations
Views, downloads and citations are aggregated across all versions of this paper published by eLife.
Download links
Downloads (link to download the article as PDF)
Open citations (links to open the citations from this article in various online reference manager services)
Cite this article (links to download the citations from this article in formats compatible with various reference manager tools)
Further reading

 Computational and Systems Biology
 Neuroscience
Normal aging leads to myelin alterations in the rhesus monkey dorsolateral prefrontal cortex (dlPFC), which are positively correlated with degree of cognitive impairment. It is hypothesized that remyelination with shorter and thinner myelin sheaths partially compensates for myelin degradation, but computational modeling has not yet explored these two phenomena together systematically. Here, we used a twopronged modeling approach to determine how agerelated myelin changes affect a core cognitive function: spatial working memory. First, we built a multicompartment pyramidal neuron model fit to monkey dlPFC empirical data, with an axon including myelinated segments having paranodes, juxtaparanodes, internodes, and tight junctions. This model was used to quantify conduction velocity (CV) changes and action potential (AP) failures after demyelination and subsequent remyelination. Next, we incorporated the single neuron results into a spiking neural network model of working memory. While complete remyelination nearly recovered axonal transmission and network function to unperturbed levels, our models predict that biologically plausible levels of myelin dystrophy, if uncompensated by other factors, can account for substantial working memory impairment with aging. The present computational study unites empirical data from ultrastructure up to behavior during normal aging, and has broader implications for many demyelinating conditions, such as multiple sclerosis or schizophrenia.

 Neuroscience
How is new information organized in memory? According to latent state theories, this is determined by the level of surprise, or prediction error, generated by the new information: a small prediction error leads to the updating of existing memory, large prediction error leads to encoding of a new memory. We tested this idea using a protocol in which rats were first conditioned to fear a stimulus paired with shock. The stimulus was then gradually extinguished by progressively reducing the shock intensity until the stimulus was presented alone. Consistent with latent state theories, this gradual extinction protocol (small prediction errors) was better than standard extinction (large prediction errors) in producing longterm suppression of fear responses, and the benefit of gradual extinction was due to updating of the conditioning memory with information about extinction. Thus, prediction error determines how new information is organized in memory, and latent state theories adequately describe the ways in which this occurs.