ADHD Classification with GCN via Joint Feature Learning among Nodes and Edges.
Authors
Abstract
Brain functional connectivity networks (FCNs) derived from resting-state functional magnetic resonance imaging (rs-fMRI) data have been widely used to identify altered brain network patterns in attention-deficit/hyperactivity disorder (ADHD). Current graph neural network (GNN) approaches using FCNs predominantly emphasize node features while underutilizing edge information. Moreover, these GNN-based methods also inadequately represent dynamic interdependencies among evolving node features across network layers, limiting their diagnostic performance. We present a graph convolutional network via joint feature learning between nodes and edges (JNEL-GCN) that integrates neuroimaging features for ADHD classification and biomarker discovery. Our framework constructs dual graph representations: (1) a node graph using amplitude of low-frequency fluctuations (ALFF) measures across multiple frequency bands as nodal features, along with functional connectivity (FC) and node feature relationship matrices as edge attributes; (2) an edge graph derived through line graph theory, enabling the interchange of node and edge roles. By leveraging the dual-graph design, our model implements an alternating feature update mechanism with optimized graph convolution operations, facilitating feature hierarchical learning of node-edge relationships across network layers. Extensive experiments demonstrate remarkable performance, achieving 97.3% accuracy on ADHD200 and 97.1% on ABIDE-I datasets, significantly outperforming current benchmarks. Meanwhile, gradient-based biomarker analysis identifies significant regions in bilateral limbic and default mode networks associated with ADHD, aligning with the findings in existing literature. Therefore, this dual-graph approach advances neuroimaging-based diagnosis by comprehensively capturing dynamic network interactions, while providing interpretable biomarkers for clinical neuroscience applications.