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

TL;DR

This paper introduces a novel data augmentation method using a graph-based generative adversarial network to improve one-shot learning classification of brain connectome data, demonstrating enhanced accuracy and balanced metrics.

Contribution

It presents the first approach to augment brain graph data from a single population template using gGAN, improving classification performance in neurological disorder datasets.

Findings

Enhanced classification accuracy with augmented data.

More balanced metrics across evaluation criteria.

Effective generalization demonstrated through cross-validation.

Abstract

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.