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
|
Pdf
Evaluation of Transfer Learning Pipeline.pdf Download (47kB) | Preview |
|
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 |