Learning Cortical Parcellations Using Graph Neural Networks

Jumping knowledge graph attention network for resource allocation in wireless cellular system

Next-generation wireless networks are characterized by two essential features: ubiquitous connectivity and high-speed data transmission. The realization of these features hinges on the development of rational resource allocation strategies to optimize the utilization of radio resources. This study addresses the beamforming design problem for downlink transmission in multi-cell cellular networks, with a focus on maximizing user data rates while adhering to stringent power constraints. To tackle this challenge, we propose a novel graph learning-based optimization framework that learns the mapping from channel states to beamforming vectors in an unsupervised manner. At the core of this framework is an attention-based graph neural network (GNN), which efficiently captures complex inter-node relationships by dynamically computing the importance of neighboring nodes. Furthermore, a jumping knowledge network is integrated to enhance structural representation learning, enabling the model to adaptively capture diverse neighborhood ranges for each node and mitigate the issue of over-smoothing. Extensive simulations demonstrate that the proposed algorithm significantly outperforms existing benchmark methods, exhibiting robust performance and strong generalization capabilities across a wide range of system parameter configurations.

Fine-scale striatal parcellation using diffusion MRI tractography and graph neural networks

The striatum, a crucial part of the basal ganglia, plays a key role in various brain functions through its interactions with the cortex. The complex structural and functional diversity across subdivisions within the striatum highlights the necessity for precise striatal segmentation. In this study, we introduce a novel deep clustering pipeline for automated, fine-scale parcellation of the striatum using diffusion MRI (dMRI) tractography. Initially, we employ a voxel-based probabilistic fiber tractography algorithm combined with a fiber-tract embedding technique to capture intricate dMRI connectivity patterns. To maintain critical inter-voxel relationships, our approach employs Graph Neural Networks (GNNs) to create accurate graph representations of the striatum. This involves encoding probabilistic fiber bundle characteristics as node attributes and refining edge weights using activation functions to enhance the graph's interpretability and accuracy. The methodology incorporates a Transformer-based GraphConv autoencoder in the pre-training phase to extract critical spatial features while minimizing reconstruction loss. In the fine-tuning phase, a novel joint loss mechanism markedly improves segmentation precision and anatomical fidelity. Integration of traditional clustering techniques with multi-head self-attention mechanisms further elevates the accuracy and robustness of our segmentation approach. This methodology provides new insights into the striatum's role in cognition and behavior and offers potential clinical applications for neurological disorders.

Geometric deep learning with adaptive full-band spatial diffusion for accurate, efficient, and robust cortical parcellation

Cortical parcellation delineates the cerebral cortex into distinct regions according to their distinctiveness in anatomy and/or function, which is a fundamental preprocess in brain cortex analysis and can influence the accuracy and specificity of subsequent neuroscientific research and clinical diagnosis. Conventional methods for cortical parcellation involve spherical mapping and multiple morphological feature computation, which are time-consuming and prone to error due to the spherical mapping process. Recent geometric learning approaches have attempted to automate this process by replacing the registration-based parcellation with deep learning-based methods. However, they have not fully addressed spherical mapping and cortical features quantification, making them sensitive to variations in mesh structures. In this work, to directly parcellate original surfaces in individual space with minimal preprocessing, we present a full-band spectral-accelerated spatial diffusion strategy for stable information propagation on highly folded cortical surfaces, contributing to adaptive learning of fine-grained geometric representations and the construction of a compact deep network (termed Cortex-Diffusion) for fully automatic parcellation. Using only raw 3D vertex coordinates and having merely 0.49 MB of learnable parameters, it demonstrates state-of-the-art parcellation accuracy, efficiency, and superior robustness to mesh resolutions and discretization patterns in both the cases of infant and adult brain imaging datasets.

fMRI-based spatio-temporal parcellations of the human brain

PURPOSE OF REVIEW

Human brain parcellation based on functional magnetic resonance imaging (fMRI) plays an essential role in neuroscience research. By segmenting vast and intricate fMRI data into functionally similar units, researchers can better decipher the brain's structure in both healthy and diseased states. This article reviews current methodologies and ideas in this field, while also outlining the obstacles and directions for future research.

RECENT FINDINGS

Traditional brain parcellation techniques, which often rely on cytoarchitectonic criteria, overlook the functional and temporal information accessible through fMRI. The adoption of machine learning techniques, notably deep learning, offers the potential to harness both spatial and temporal information for more nuanced brain segmentation. However, the search for a one-size-fits-all solution to brain segmentation is impractical, with the choice between group-level or individual-level models and the intended downstream analysis influencing the optimal parcellation strategy. Additionally, evaluating these models is complicated by our incomplete understanding of brain function and the absence of a definitive "ground truth".

SUMMARY

While recent methodological advancements have significantly enhanced our grasp of the brain's spatial and temporal dynamics, challenges persist in advancing fMRI-based spatio-temporal representations. Future efforts will likely focus on refining model evaluation and selection as well as developing methods that offer clear interpretability for clinical usage, thereby facilitating further breakthroughs in our comprehension of the brain.

Longitudinally consistent registration and parcellation of cortical surfaces using semi-supervised learning

Temporally consistent and accurate registration and parcellation of longitudinal cortical surfaces is of great importance in studying longitudinal morphological and functional changes of human brains. However, most existing methods are developed for registration or parcellation of a single cortical surface. When applying to longitudinal studies, these methods independently register/parcellate each surface from longitudinal scans, thus often generating longitudinally inconsistent and inaccurate results, especially in small or ambiguous cortical regions. Essentially, longitudinal cortical surface registration and parcellation are highly correlated tasks with inherently shared constraints on both spatial and temporal feature representations, which are unfortunately ignored in existing methods. To this end, we unprecedentedly propose a novel semi-supervised learning framework to exploit these inherent relationships from limited labeled data and extensive unlabeled data for more robust and consistent registration and parcellation of longitudinal cortical surfaces. Our method utilizes the spherical topology characteristic of cortical surfaces. It employs a spherical network to function as an encoder, which extracts high-level cortical features. Subsequently, we build two specialized decoders dedicated to the tasks of registration and parcellation, respectively. To extract more meaningful spatial features, we design a novel parcellation map similarity loss to utilize the relationship between registration and parcellation tasks, i.e., the parcellation map warped by the deformation field in registration should match the atlas parcellation map, thereby providing extra supervision for the registration task and augmented data for parcellation task by warping the atlas parcellation map to unlabeled surfaces. To enable temporally more consistent feature representation, we additionally enforce longitudinal consistency among longitudinal surfaces after registering them together using their concatenated features. Experiments on two longitudinal datasets of infants and adults have shown that our method achieves significant improvements on both registration/parcellation accuracy and longitudinal consistency compared to existing methods, especially in small and challenging cortical regions.

Review of the Brain's Behaviour after Injury and Disease for Its Application in an Agent-Based Model (ABM)

The brain is the most complex organ in the human body and, as such, its study entails great challenges (methodological, theoretical, etc.). Nonetheless, there is a remarkable amount of studies about the consequences of pathological conditions on its development and functioning. This bibliographic review aims to cover mostly findings related to changes in the physical distribution of neurons and their connections-the connectome-both structural and functional, as well as their modelling approaches. It does not intend to offer an extensive description of all conditions affecting the brain; rather, it presents the most common ones. Thus, here, we highlight the need for accurate brain modelling that can subsequently be used to understand brain function and be applied to diagnose, track, and simulate treatments for the most prevalent pathologies affecting the brain.

An Agent-Based Model to Reproduce the Boolean Logic Behaviour of Neuronal Self-Organised Communities through Pulse Delay Modulation and Generation of Logic Gates

The human brain is arguably the most complex "machine" to ever exist. Its detailed functioning is yet to be fully understood, let alone modelled. Neurological processes have logical signal-processing and biophysical aspects, and both affect the brain's structure, functioning and adaptation. Mathematical approaches based on both information and graph theory have been extensively used in an attempt to approximate its biological functioning, along with Artificial Intelligence frameworks inspired by its logical functioning. In this article, an approach to model some aspects of the brain learning and signal processing is presented, mimicking the metastability and backpropagation found in the real brain while also accounting for neuroplasticity. Several simulations are carried out with this model to demonstrate how dynamic neuroplasticity, neural inhibition and neuron migration can reshape the brain's logical connectivity to synchronise signal processing and obtain certain target latencies. This work showcases the importance of dynamic logical and biophysical remodelling in brain plasticity. Combining mathematical (agents, graph theory, topology and backpropagation) and biomedical ingredients (metastability, neuroplasticity and migration), these preliminary results prove complex brain phenomena can be reproduced-under pertinent simplifications-via affordable computations, which can be construed as a starting point for more ambitiously accurate simulations.

A joint parcellation and boundary network with multi-rate-shared dilated graph attention for cortical surface parcellation

Cortical surface parcellation aims to segment the surface into anatomically and functionally significant regions, which are crucial for diagnosing and treating numerous neurological diseases. However, existing methods generally ignore the difficulty in learning labeling patterns of boundaries, hindering the performance of parcellation. To this end, this paper proposes a joint parcellation and boundary network (JPBNet) to promote the effectiveness of cortical surface parcellation. Its core is developing a multi-rate-shared dilated graph attention (MDGA) module and incorporating boundary learning into the parcellation process. The former, in particular, constructs a dilated graph attention strategy, extending the dilated convolution from regular data to irregular graph data. We fuse it with different dilated rates to extract context information in various scales by devising a shared graph attention layer. After that, a boundary enhancement module and a parcellation enhancement module based on graph attention mechanisms are built in each layer, forcing MDGA to capture informative and valuable features for boundary detection and parcellation tasks. Integrating MDGA, the boundary enhancement module, and the parcellation enhancement module at each layer to supervise boundary and parcellation information, an effective JPBNet is formed by stacking multiple layers. Experiments on the public dataset reveal that the proposed method outperforms comparison methods and performs well on boundaries for cortical surface parcellation.

Predicting in vitro single-neuron firing rates upon pharmacological perturbation using Graph Neural Networks

Modern Graph Neural Networks (GNNs) provide opportunities to study the determinants underlying the complex activity patterns of biological neuronal networks. In this study, we applied GNNs to a large-scale electrophysiological dataset of rodent primary neuronal networks obtained by means of high-density microelectrode arrays (HD-MEAs). HD-MEAs allow for long-term recording of extracellular spiking activity of individual neurons and networks and enable the extraction of physiologically relevant features at the single-neuron and population level. We employed established GNNs to generate a combined representation of single-neuron and connectivity features obtained from HD-MEA data, with the ultimate goal of predicting changes in single-neuron firing rate induced by a pharmacological perturbation. The aim of the main prediction task was to assess whether single-neuron and functional connectivity features, inferred under baseline conditions, were informative for predicting changes in neuronal activity in response to a perturbation with Bicuculline, a GABA A receptor antagonist. Our results suggest that the joint representation of node features and functional connectivity, extracted from a baseline recording, was informative for predicting firing rate changes of individual neurons after the perturbation. Specifically, our implementation of a GNN model with inductive learning capability (GraphSAGE) outperformed other prediction models that relied only on single-neuron features. We tested the generalizability of the results on two additional datasets of HD-MEA recordings-a second dataset with cultures perturbed with Bicuculline and a dataset perturbed with the GABA A receptor antagonist Gabazine. GraphSAGE models showed improved prediction accuracy over other prediction models. Our results demonstrate the added value of taking into account the functional connectivity between neurons and the potential of GNNs to study complex interactions between neurons.