Evaluation of Transfer Learning Pipeline for ADHD Classification via fMRI Images

Nur Atiqah, Kamal and Ahmad Fakhri, Ab Nasir and Abdul Majeed, Anwar P. P. and Muhammad Zulfahmi, Toh Abdullah@ Toh Chin Lai and Ismail, Mohd Khairuddin (2024) Evaluation of Transfer Learning Pipeline for ADHD Classification via fMRI Images. In: Intelligent Manufacturing and Mechatronics, Lecture Notes in Networks and Systems. 4th International conference on Innovative Manufacturing, Mechatronics and Materials Forum, iM3F2023 , 07 – 08 August 2023 , Pekan, Malaysia. pp. 251-262., 850. ISSN 2367-3389 ISBN 978-981-99-8819-8

[img]
Preview
Pdf
Evaluation of Transfer Learning Pipeline.pdf

Download (47kB) | Preview
[img] Pdf
Evaluation of Transfer Learning Pipeline for ADHD Classification via fMRI Images.pdf
Restricted to Repository staff only

Download (290kB) | Request a copy

Abstract

In recent times, diverse machine learning models have been employed in this field of technology. Nevertheless, the implementation of learning models for image classification remains uncertain and has proven to be challenging. The utilization of transfer learning (TL) has been showcased as a potent technique for extracting crucial features and can significantly reduce training time. Moreover, the feature extractor model has demonstrated excellent performance in the TL method across numerous applications. As of now, there has been no evaluation of using these methods for ADHD classification through functional magnetic resonance imaging (fMRI) applications. The objective of this study is to identify an appropriate pipeline consisting of transfer learning and conventional classifiers for effectively discriminating between individuals with ADHD and those without. For feature extraction, InceptionV3, VGG16, and VGG19 models were employed, which were subsequently combined with either k-nearest neighbor (k-NN) or support vector machine (SVM) classifiers. A dataset consisting of 556 images was collected from the ADHD-200 competition dataset. The data were divided into an 80:20 ratio, with 80% used for training and 20% for testing. The hyperparameters of both k-NN and SVM were optimized using the grid search method. The experimental results revealed that the optimal pipelines were achieved using InceptionV3 coupled with k-NN classifier, where the best parameters were determined as the Minkowski distance metric and a k-value of 1. The pipeline demonstrated a macro-average classification accuracy of 1.00 for the training set and 0.95 for the test set. In summary, the results demonstrate that TL models have successfully exhibited the capability to differentiate fMRI images for ADHD classification.

Item Type: Conference or Workshop Item (Paper)
Additional Information: Indexed by Scopus
Uncontrolled Keywords: ADHD; fMRI images; Machine learning; Transfer learning
Subjects: Q Science > QA Mathematics > QA75 Electronic computers. Computer science
T Technology > TS Manufactures
Faculty/Division: Institute of Postgraduate Studies
Faculty of Computing
Faculty of Manufacturing and Mechatronic Engineering Technology
Depositing User: Miss Amelia Binti Hasan
Date Deposited: 24 May 2024 02:44
Last Modified: 24 May 2024 02:44
URI: http://umpir.ump.edu.my/id/eprint/41383
Download Statistic: View Download Statistics

Actions (login required)

View Item View Item