Author
Listed:
- Simran Gupta
(Alliance Academy of Innovation, Cumming, GA 30040, USA)
- Md. Rahad Islam Bhuiyan
(Health Informatics Research Lab, Department of Computer Science and Engineering, Daffodil International University, Dhaka 1216, Bangladesh)
- Sadia Sultana Chowa
(Health Informatics Research Lab, Department of Computer Science and Engineering, Daffodil International University, Dhaka 1216, Bangladesh)
- Sidratul Montaha
(Department of Computer Science, University of Calgary, Calgary, AB T2N 1N4, Canada)
- Rashik Rahman
(Department of Computer Science, University of Calgary, Calgary, AB T2N 1N4, Canada)
- Sk. Tanzir Mehedi
(School of Computer Science, Queensland University of Technology, Brisbane, QLD 4000, Australia)
- Ziaur Rahman
(School of Computer Science, Queensland University of Technology, Brisbane, QLD 4000, Australia)
Abstract
Autism spectrum disorder (ASD) is a complex neurodevelopmental condition that presents significant diagnostic challenges due to its varied symptoms and nature. This study aims to improve ASD classification using advanced deep learning techniques applied to neuroimaging data. We developed an automated system leveraging the ABIDE-1 dataset and a novel lightweight quantized one-dimensional (1D) Convolutional Neural Network (Q-CNN) model to analyze fMRI data. Our approach employs the NIAK pipeline with multiple brain atlases and filtering methods. Initially, the Regions of Interest (ROIs) are converted into feature vectors using tangent space embedding to feed into the Q-CNN model. The proposed 1D-CNN is quantized through Quantize Aware Training (QAT). As the quantization method, int8 quantization is utilized, which makes it both robust and lightweight. We propose a federated learning (FL) framework to ensure data privacy, which allows decentralized training across different data centers without compromising local data security. Our findings indicate that the CC200 brain atlas, within the NIAK pipeline’s filt-global filtering methods, provides the best results for ASD classification. Notably, the ASD classification outcomes have achieved a significant test accuracy of 98% using the CC200 and filt-global filtering techniques. To the best of our knowledge, this performance surpasses previous studies in the field, highlighting a notable enhancement in ASD detection from fMRI data. Furthermore, the FL-based Q-CNN model demonstrated robust performance and high efficiency on a Raspberry Pi 4, underscoring its potential for real-world applications. We exhibit the efficacy of the Q-CNN model by comparing its inference time, power consumption, and storage requirements with those of the 1D-CNN, quantized CNN, and the proposed int8 Q-CNN models. This research has made several key contributions, including the development of a lightweight int8 Q-CNN model, the application of FL for data privacy, and the evaluation of the proposed model in real-world settings. By identifying optimal brain atlases and filtering methods, this study provides valuable insights for future research in the field of neurodevelopmental disorders.
Suggested Citation
Simran Gupta & Md. Rahad Islam Bhuiyan & Sadia Sultana Chowa & Sidratul Montaha & Rashik Rahman & Sk. Tanzir Mehedi & Ziaur Rahman, 2024.
"Enhancing Autism Spectrum Disorder Classification with Lightweight Quantized CNNs and Federated Learning on ABIDE-1 Dataset,"
Mathematics, MDPI, vol. 12(18), pages 1-28, September.
Handle:
RePEc:gam:jmathe:v:12:y:2024:i:18:p:2886-:d:1479066
Download full text from publisher
Corrections
All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:18:p:2886-:d:1479066. See general information about how to correct material in RePEc.
If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.
We have no bibliographic references for this item. You can help adding them by using this form .
If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.
For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .
Please note that corrections may take a couple of weeks to filter through
the various RePEc services.