Back

The Effect of Node Features on GCN-based Brain Network Classification: An Empirical Study

Wang, G.; Zhang, L.; Qiao, L.

2022-11-03 neuroscience
10.1101/2022.11.01.514695 bioRxiv
Show abstract

Brain functional network (BFN) analysis has become a popular technique for identifying neurological/mental diseases. Due to the fact that BFN is a graph, graph convolutional network (GCN) can be naturally used in the classification of BFN. Different from traditional methods that directly use the adjacency matrices of BFNs to train a classifier, GCN requires an additional input -- node features. To our best knowledge, however, there is no systematic study to analyze their influence on the performance of GCN-based brain disorder classification. Therefore, in this per, we conduct an empirical study on various node feature measures, including 1) original fMRI signals, 2) one-hot encoding, 3) node statistics, 4) node correlation, and 5) their combination. Experimental results on two benchmark databases show that different node feature inputs to GCN significantly affect the brain disease classification performance, and node correlation usually contributes higher accuracy compared to original signals and manually extracted statistical features.

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.