Population Template-Based Brain Graph Augmentation for Improving One-Shot Learning Classification

by   Oben Özgür, et al.

The challenges of collecting medical data on neurological disorder diagnosis problems paved the way for learning methods with scarce number of samples. Due to this reason, one-shot learning still remains one of the most challenging and trending concepts of deep learning as it proposes to simulate the human-like learning approach in classification problems. Previous studies have focused on generating more accurate fingerprints of the population using graph neural networks (GNNs) with connectomic brain graph data. Thereby, generated population fingerprints named connectional brain template (CBTs) enabled detecting discriminative bio-markers of the population on classification tasks. However, the reverse problem of data augmentation from single graph data representing brain connectivity has never been tackled before. In this paper, we propose an augmentation pipeline in order to provide improved metrics on our binary classification problem. Divergently from the previous studies, we examine augmentation from a single population template by utilizing graph-based generative adversarial network (gGAN) architecture for a classification problem. We benchmarked our proposed solution on AD/LMCI dataset consisting of brain connectomes with Alzheimer's Disease (AD) and Late Mild Cognitive Impairment (LMCI). In order to evaluate our model's generalizability, we used cross-validation strategy and randomly sampled the folds multiple times. Our results on classification not only provided better accuracy when augmented data generated from one sample is introduced, but yields more balanced results on other metrics as well.


page 5

page 13

page 14


Predicting Brain Multigraph Population From a Single Graph Template for Boosting One-Shot Classification

A central challenge in training one-shot learning models is the limited ...

BrainNetGAN: Data augmentation of brain connectivity using generative adversarial network for dementia classification

Alzheimer's disease (AD) is the most common age-related dementia. It rem...

On the use of Pairwise Distance Learning for Brain Signal Classification with Limited Observations

The increasing access to brain signal data using electroencephalography ...

Deep Graph Normalizer: A Geometric Deep Learning Approach for Estimating Connectional Brain Templates

A connectional brain template (CBT) is a normalized graph-based represen...

Manifold learning for brain connectivity

Human brain connectome studies aim at extracting and analyzing relevant ...

Simultaneous Matrix Diagonalization for Structural Brain Networks Classification

This paper considers the problem of brain disease classification based o...

Please sign up or login with your details

Forgot password? Click here to reset