1 Introduction

Striatum is a critical sub-cortical structure that connects the cortex and other basal ganglia structures. It is an essential part of the cortico-striatal-thalamo-cortical (CSTC) network that regulates human emotion and behaviors [1]. With T1-weighted MRI, shape analysis has been applied to study morphometry changes of the striatum in neurological [2] and mental disorders [3] by dividing the striatum into three parts: the caudate, putamen, and accumbens nuclei. Recent connectivity research [4, 5] and animal studies [6], however, indicate that striatum functions do not follow such subdivisions and it is more natural to map the striatum as a holistic structure. In this work, we follow this line of research in neuroscience and develop a novel striatum surface mapping method for studying its connectivity and function.

The novel surface mapping method we develop is based on Riemannian metric optimization on surfaces (RMOS) in the Laplace-Beltrami (LB) embedding space we recently proposed in [7]. As illustrated in Fig. 1, our metric optimization approach eliminates the large metric distortion during the parameterization step of conventional spherical registration methods [8,9,10]. This fundamental advantage stems from the isometry of the LB embedding [11]. This is especially important for mapping the striatum as a holistic surface because of its drastically different geometry as compared to the unit sphere. To drive the striatum surface mapping in the LB embedding space, we also develop novel anatomical contextual features using the relation of neighboring brain structures. This helps establish anatomically meaningful maps and remove sign ambiguity in the LB eigenfunctions, which is a critical problem in shape analysis using the LB spectrum [12, 13]. In our experimental results, we compare our novel striatum mapping method with two spherical registration algorithms from FreeSurfer [8] and FSL [10], and apply it to a large-scale dataset from Human Connectome Project (HCP) [14] for the construction of a surface-based connectivity atlas of the striatum.

Fig. 1.
figure 1

A conceptual comparison of (A) the spherical registration method and (B) our metric optimization method in the Laplace-Beltrami embedding space for the mapping of two striatum surfaces \(M_1\) and \(M_2\). The fundamental advantage of the metric optimization framework is that there is no extra distortion induced by the parameterization process. All the metric changes are induced to match the two surfaces. On the other hand, the spherical mapping process introduces large metric distortions purely for the parameterization step, which can lead to large errors in the final maps.

2 Methods

Striatal Surface Reconstruction. To perform striatal surface mapping, we first reconstruct a triangular mesh representation of the striatum in each hemisphere of the human brain from T1-weighted MRI (Fig. 2). Because publicly available tools typically segment out the caudate, putamen, and accumbens nuclei of the striatum, we merge them into one mask and apply the surface reconstruction method in our MOCA software tool [15] on NITRC. This algorithm ensures all surfaces have genus-zero topology, and removes segmentation artifacts without volume shrinkage. Finally we decimate each mesh to 1000 vertices and 2994 edges for all mapping tasks.

Fig. 2.
figure 2

Surface representation of (A) subdivision of the striatum; (B) the whole striatum.

Metric Optimization in LB Embedding Space. Let \(\mathcal {M}_1\) and \(\mathcal {M}_2\) denote the mesh representation of two striatum surfaces, \(\xi _1^j\) and \(\xi _2^j (j=1,\cdots ,L)\) denote the L features defined on the two surfaces to guide their mapping, and \(W_1\) and \(W_2\) denote their Riemannian metrics, i.e., the edge weights. Given the Riemannian metrics, we can compute the LB embeddings of \(\mathcal {M}_i (i=1,2)\) as:

$$\begin{aligned} I_{\mathcal {M}_i}^{\varPhi _i}(x)&= (\frac{f_1^i(x)}{\sqrt{\lambda _1^i}},\frac{f_2^i(x)}{\sqrt{\lambda _2^i}} \cdots ,\frac{f_n^i(x)}{\sqrt{\lambda _n^i}},\cdots ) ~~~\forall x \in \mathcal {M}_i , \end{aligned}$$
(1)

where \(\lambda _n^i\) and \(f_n^i\) denote the n-th eigenvalue and eigenfunction of \(\mathcal {M}_i\) [7]. To compute the surface maps \(u_1: \mathcal {M}_1 \rightarrow \mathcal {M}_2\) and \(u_2: \mathcal {M}_2 \rightarrow \mathcal {M}_1\), we will minimize an energy function via metric optimization as follows:

$$\begin{aligned} E(W_1{,}W_2{,}u_1{,}u_2)&=\underbrace{\sum _{j=1}^L\left[ \int _{\mathcal {M}_1}C_j\left( \xi _1^j,\xi _2^j{\circ }u_1\right) d\mathcal {M}_1{+}\int _{\mathcal {M}_2}C_j\left( \xi _2^j,\xi _1^j{\circ }u_2\right) d\mathcal {M}_2\right] }_{\text {Feature Term}\, E_F}\nonumber \\&\quad + \underbrace{\gamma \left( \text {Reg}\,(W_1)+Reg(W_2)\right) }_{\text {Regularization Term}\, E_R} \end{aligned}$$
(2)

There are two terms in the energy function: feature term \(E_F\), and regularization term \(E_R\) with the coefficient \(\gamma \) for the smoothness of the Riemannian metrics. The feature term measures the agreement of corresponding features. For the j-th feature, the cost function \(C_j\) measures the similarity of the same feature at the corresponding locations of the two surfaces induced by the surface maps \(u_1\) and \(u_2\). In this work, we define the cost function as the \(l_2\) energy, but more general choices such as mutual information are also possible. The regularization energy is designed to avoid overly irregular distortion during the mapping process. In this work, we define \(E_R\) at each edge as the \(l_2\) difference between its metric ratio and the mean value of its neighbors.

Fig. 3.
figure 3

The main algorithmic steps to compute the striatal surface maps.

As shown in Fig. 3, there are three main steps in the energy minimization. Given the current metrics, we compute the LB eigen-system and construct their LB embeddings in Step 1. Also, the sign and order ambiguities in the eigenfunctions are resolved in this step by searching for the optimal embedding from possible combinations of the eigenfunctions. To better estimate the right combination, we compute the feature energy \(E_F\) in Eq. 2 with the nearest point map for every possible combination of the eigenfunctions, and search for the optimal embedding only from the combinations for which the feature energy is sufficiently low (below the lower 10 percentile in this work). To link the mismatch of features with the metrics, we then compute the \(\beta \)-maps \(\tilde{u}_i^\beta \) along the gradient descent direction of \(E_F\) and convert it into a distance energy \(\tilde{E}_F\) in Step 2. This is a critical step as the distance energy in the embedding space is a function of the eigen-systems as defined in [13], so it allows the computation of the gradient of the final energy E with respect to the metrics. This information is sent back to the image space to update the metrics \(W_1\) and \(W_2\) in Step 3. After that, the three steps are repeated until convergence. The final maps are obtained via composition of the embeddings and the maps in the embedding space. Because both surfaces are treated equally in the RMOS framework, the maps are symmetric.

As shown in Fig. 1(B), this computational framework uses the high dimensional LB embedding space as a canonical space instead of the 2D sphere (Fig. 1(A)). The isometry of the LB embedding ensures that the numerical calculation we perform in the embedding space stays faithful to the original surfaces. The surface registration algorithm in the framework is also fundamentally different from conventional nonlinear mesh deformation on the sphere [8, 9]. To calculate a general diffeomorphism between surfaces, iterative optimization of the metrics on mesh edges was conducted instead of deforming the meshes in the embedding space. This means that we only need to optimize a set of scalar functions on the surfaces without the need of worrying about numerical difficulties such as the flipping of triangles or self-intersection. This numerical advantage is from the theoretical guarantee that the LB embedding is fully determined by the Riemannian metric on the surfaces and the Riemannian metric can be fully determined by a diffeomorphism via the pullback metric.

Anatomical Contextual Features for Striatal Surface Mapping. The metric optimization framework is flexible and can take general features to guide surface mapping in the LB embedding space. For the mapping of striatal surfaces, we will utilize contextual information for robustness. Compared with local features derived from curvatures, the context features provide a global characterization about “where to where” in the mapping process. Importantly, they also help resolve the sign ambiguity of the LB eigen-system by providing a unique description in different parts of a structure, especially when there exist some symmetries of the structure shape. Finally, the redundant characterizations from all the features will increase robustness in mapping results.

Fig. 4.
figure 4

An illustration of Anatomical Neighbor Context (ANC). (A) a striatum surface and its neighboring structures used for defining the ANC. Normalized distance features to the ventricle, insula, and pallidum are shown in (B), (C), and (D), respectively.

For striatal surface mapping, we develop anatomical neighbor context (ANC) features that characterize the geometric relation between neighboring brain structures. At each vertex of a surface, a contextual feature will be represented as a vector. Let \(d_i\) (\(i=1,2,\cdots ,C\)) denote the \(l_2\)-distance transform of its i-th neighbor. For each vertex \(V_j\in \mathcal {M}\), its ANC is defined as

$$\begin{aligned} ANC(V_j)&= \left[ \frac{d_1(V_j)}{\max \limits _{V_k\in \mathcal {M}}d_1(V_k)},\frac{d_2(V_j)}{\max \limits _{V_k\in \mathcal {M}}d_2(V_k)},\cdots ,\frac{d_C(V_j)}{\max \limits _{V_k\in \mathcal {M}}d_C(V_k)}\right] . \end{aligned}$$
(3)

By normalizing each distance transform by its maximal value on the surface, we make the ANC feature invariant to scale differences. As an example, we show in Fig. 4 the ANC features for a striatal surface with respect to three neighbors: the lateral ventricle, insular cortex and pallidum. These features provide complementary information in different parts of the striatum. Taken together, they give a highly informative description about the corresponding locations across different subjects.

3 Results

Curvature-Driven Surface Mapping. As a first experiment, we demonstrate curvature-driven mapping between two left striatal surfaces shown in Fig. 5(A) and (B) using RMOS and spherical registration. The surfaces are colored by mean curvature (MC) that is normalized by surface volume for surface mapping. We computed the RMOS map with the maximum of 60 eigenfunctions, and the regularization coefficient of 0.1. For spherical registration, the two surfaces were first parameterized to the unit sphere (Fig. 5(E) and (F)) using SPHARM-MAT [16], with the maximum SPHARM degree of 6 and other recommended parameter settings. Then the source sphere was registered to the target sphere by matching the MC using FreeSurfer’s mris_register [8] and FSL’s msm [10] under the default parameter settings.

Fig. 5.
figure 5

Curvature-driven mapping between two left striatal surfaces. (A) The source and (B) target surfaces are colored by mean curvature (MC). Using the maps computed by MC-driven RMOS, we pull back the MC of the target surface onto the source surface (C), and project the source onto the target surface (D). The two striatal surfaces in (A, B) were parametrized to (E, F) the unit spheres (colored by MC in (A, B)) using SPHARM-MAT, (E) the source sphere was registered to (F) the target sphere using (G) FreeSurfer and (H) FSL. Projection of the source striatal surface in (A) to the target surface by the spherical map is shown in (I, J), respectively.

Large metric distortions were introduced in spherical parameterization as shown in Fig. 5(E) and (F), which led to significant errors in the subsequent spherical registration from the source to target sphere as shown in Fig. 5(G) and (H) by FreeSurfer and FSL, respectively. It is more obviously observed, in projection of the source to target striatal surface via the maps computed by spherical registration (Fig. 5(I) and (J)), that spherical mapping failed in the establishment of correspondences especially in the putamen part. This failure was consistently observed in spherical mapping of other striatal surfaces. With MC-driven RMOS, we can see in Fig. 5(C) and (D) that high quality surface maps have been achieved with little metric distortion.

Fig. 6.
figure 6

RMOS mapping from (A) a left striatal surface to (B) a right striatal surface driven by (C) MC only and (D) ANC (with MC). (C) In projection of the left to right striatal surface by the map of MC-driven RMOS, the green patch on the left putamen in (A) was mapped onto the opposite side of the right putamen in (C). (D) ANC-driven RMOS correctly maps the patch onto the same side of the right putamen in (D). The surface views in (C) is opposite to (A, B, D) (lateral vs. medial views).

ANC-Driven RMOS Mapping. In this experiment, we demonstrate ANC-driven RMOS mapping between left and right striatal surfaces, which is more challenging because the symmetry between the left and right shapes makes it more difficult to resolve the sign ambiguity in the early mapping process. As shown in Fig. 6(C), MC-driven RMOS failed to establish anatomically correct correspondences (for example, the green patch on the putamen) for a pair of left and right striatal surfaces shown in Fig. 6(A) and (B). This is because MC did not provide a unique description in different parts of the striatum, especially in the putamen part. Using both the MC and ANC features developed in this work, we can see that RMOS is able to correctly map the corresponding anatomy (Fig. 6(D)).

Striatal Connectivity Atlas. In the final experiment, we constructed a surface-based connectivity atlas of the left striatum from 211 subjects of the Q1–Q3 release of HCP. We computed striatal connectivity using probabilistic tractography with fiber orientation distributions (FODs) reconstructed from the multi-shell diffusion MRI data [17]. We generated the connectivity maps to seven cortical regions: orbital-frontal (medial only), middle/inferior-frontal, superior-frontal, precentral, parietal, occipital, and temporal cortices [4] as done from the thalamus in [7].

For construction of the connectivity atlas, RMOS maps between one reference surface and the other 210 surfaces was computed using both the three ANC and MC features with the same RMOS parameters above. Then the connectivities on each surface were pulled back to the reference surface by the RMOS maps, and we have a vector of the 211 connectivities to each cortical region for every vertex on the reference surface. The probabilistic atlas of each cortical regions was defined as the number of the connectivities to the cortical region higher than 10% divided by the total number 211 for every vertex. Then we thresholded the probabilistic atlas at 25%, which are shown in Fig. 7(A). Finally, the labeled atlas was computed by assigning each vertex to the cortical region to which it had the highest probability of the connection.

Each cortical region is topographically connected to adjacent areas on the striatal surface, but there is little connection at the lateral parts of the striatal surface as observed in tractography around the striatum (Fig. 7(A)). This is possibly due to the difficulty of current tractography techniques in detecting extremely short connections to neighboring cortices. More importantly, the striato-cortical connectivity patterns do not follow the conventional subdivison from the FSL segmentation tool. In particular, the striatal connection to the orbito-frontal cortex not only occupies the nucleus accumbens but also extends to the ventral part of the putamen/caudate as shown in Fig. 7(C).

4 Conclusion

In this paper, we developed a novel holistic mapping method for studying striatum structure and connectivity. Using the data from 211 HCP subjects, we constructed a tractography-based striato-cortical connectivity atlas. We successfully established the detailed correspondences across the striatal surfaces using RMOS driven by the novel ANC features. The projected connectivities to each cortical region by the RMOS maps are highly clustered on functional ROIs that do not follow the traditional structural subdivision of the striatum. For future work, with our high-quality surface maps of the striatum, we will build a striatal connectivity atlas using resting-state fMRI that provides more accurate functional connections.

Fig. 7.
figure 7

Striato-cortical connectivity atlas of 211 subjects. From the connectivities on all the striatal surfaces projected to the reference surface with the RMOS maps, (A) a probabilistic atlas was computed and thresholded at 25% for each cortical region. The labeled atlas by the highest probability of connection is shown in (B) and (C). The accumbens boundary and the boundary between putamen and caudate are drawn on the surface in (C) as solid and dashed curves, respectively.