ABFR-KAN: Kolmogorov-Arnold Networks for Functional Brain Analysis
About
Functional connectivity (FC) analysis, a valuable tool for computer-aided brain disorder diagnosis, traditionally relies on atlas-based parcellation. However, issues relating to selection bias and a lack of regard for subject specificity can arise as a result of such parcellations. Addressing this, we propose ABFR-KAN, a transformer-based classification network that incorporates novel advanced brain function representation components with the power of Kolmogorov-Arnold Networks (KANs) to mitigate structural bias, improve anatomical conformity, and enhance the reliability of FC estimation. Extensive experiments on the ABIDE I dataset, including cross-site evaluation and ablation studies across varying model backbones and KAN configurations, demonstrate that ABFR-KAN consistently outperforms state-of-the-art baselines for autism spectrum distorder (ASD) classification. Our code is available at https://github.com/tbwa233/ABFR-KAN.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| ASD Classification | ABIDE I UM NYU (train test) | Accuracy57.07 | 9 | |
| Autism Spectrum Disorder Classification | ABIDE I pre-processed (NYU site) | Accuracy74.27 | 9 | |
| Autism Spectrum Disorder Classification | ABIDE I pre-processed (UM site) | Accuracy77.27 | 9 | |
| ASD Classification | ABIDE I (train: NYU, test: UM) | Accuracy55.17 | 9 |