2D and 3D Deep Learning Models for MRI-based Parkinson's Disease Classification: A Comparative Analysis of Convolutional Kolmogorov-Arnold Networks, Convolutional Neural Networks, and Graph Convolutional Networks

Overview

• Early and accurate Parkinson's Disease (PD) diagnosis remains challenging.
• This study compares deep learning architectures for MRI-based PD classification.
• It introduces the first three-dimensional (3D) implementation of Convolutional Kolmogorov-Arnold Networks (ConvKANs), a new approach that combines convolutional layers with adaptive, spline-based activations.
• The study evaluates Convolutional Neural Networks (CNNs), ConvKANs, and Graph Convolutional Networks (GCNs) using three open-source datasets.

Plain English Explanation

Parkinson's Disease (PD) is a neurological condition that can be difficult to diagnose, especially in its early stages. This study looks at using MRI brain scans to help identify PD more accurately.

The researchers tested three different deep learning models - Convolutional Neural Networks (CNNs), Convolutional Kolmogorov-Arnold Networks (ConvKANs), and Graph Convolutional Networks (GCNs) - to see how well they could classify people with PD versus healthy controls. ConvKANs are a new type of model that combines traditional convolutional layers with more flexible, spline-based activation functions.

The models were trained and tested on data from 142 participants across three different datasets. The researchers looked at both 2D (slices) and 3D (whole volume) MRI scans to see which approach worked better.

Technical Explanation

The study compared the performance of three deep learning architectures for MRI-based PD classification:

1. Convolutional Neural Networks (CNNs): A widely used deep learning model that processes spatial data like images through a series of convolutional and pooling layers.

2. Convolutional Kolmogorov-Arnold Networks (ConvKANs): A new approach that combines convolutional layers with adaptive, spline-based activation functions. This allows the model to learn more flexible feature representations.

3. Graph Convolutional Networks (GCNs): A type of deep learning model that represents MRI data as a graph, capturing structural relationships that may be overlooked by traditional CNN architectures.

The researchers used three open-source PD datasets, totaling 142 participants (75 with PD, 67 healthy controls). For the 2D analysis, they extracted 100 axial slices centered on the midbrain from each T1-weighted MRI scan. For the 3D analysis, they used the entire volumetric scans.

The results showed that ConvKANs demonstrated high performance across datasets and dimensionalities, achieving the highest 2D AUROC (0.98) in one dataset and matching CNN peak 3D performance (1.00). CNN models also performed well, while GCN models improved in the 3D analyses, reaching up to 0.97 AUROC.

Importantly, the 3D implementations yielded higher AUROC values compared to their 2D counterparts across all models, highlighting the value of volumetric data in capturing subtle PD-related changes.

Critical Analysis

While the results are promising, the study has a few limitations. The sample size, though reasonable for a medical imaging study, is still relatively small. Larger and more diverse datasets would be needed to truly assess the generalizability of these models.

Additionally, the study does not provide much insight into the interpretability of the different models. While they show some visualizations of ConvKAN spline activations and GCN graph embeddings, more detailed interpretability analyses could help understand the key features and relationships the models are learning.

Further research is also needed to explore the potential of these deep learning approaches as part of a multimodal diagnostic approach for PD, integrating MRI data with other biomarkers or clinical assessments.

Conclusion

This study demonstrates the potential of advanced deep learning architectures, particularly ConvKANs and 3D CNNs, for improving the classification of Parkinson's Disease using MRI data. The findings suggest that these approaches could contribute to more accurate and earlier diagnosis of PD, which is crucial for providing timely treatment and improving patient outcomes. While further research is needed, this work represents an important step forward in leveraging AI and medical imaging for neurological disease detection.