A Neural Architecture Search CNN for Alzheimer’s Disease Classification
Main Article Content
Abstract
The evolution of automated machine learning (AutoML) is gradually reengineering the design of deep learning architectures for various imaging tasks. AutoML effectively develops model architectures and tunes hyperparameters through neural architecture search (NAS). Deep learning model architecture design is generally considered a tedious and time-consuming task that requires mastery skills to develop robust and better-performing models for imaging tasks. Again, the model's hyperparameters must be well-tuned to ensure optimal performances, which can be tedious and time-consuming if the hyperparameters are manually selected; using existing hyperparameter optimization algorithms can be expensive regarding resources. This study addresses these challenges in developing an optimal convolutional neural network (CNN) for classifying Alzheimer's (AD). The study, therefore, adopted a NAS approach to generate a CNN model architecture using a customized search space comprising only CNN patterns implemented with a NAS framework. The search was done for ten (10) trials, yielding a CNN architecture with an accuracy of 95.85% and a loss of 0.22. Training the model with a 10-fold cross-validation approach using a 0.0009 learning rate for 150 epochs improved the model's performance. The model recorded 97.17% accuracy, 97.21% precision, 97.14% recall, and a 0.99 area under the curve (AUC) in classifying AD as one of AD, mild cognitive impairment (MCI), and normal control (NC). The model obtained 98.06%, 98.66%, and 98.62% accuracy on binary classes of AD/NC, AD/MCI, and NC/MCI, respectively. The model generally showed robustness and better performance than existing CNN architectures.
Article Details
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
References
Y. Li et al., “Hierarchical Reinforcement Learning for Sequencing Behaviors,” Wiley Interdiscip. Rev. Data Min. Knowl. Discov., vol. 2733, no. 6, pp. 2709–2733, 2018.
Y. Li, H. Zhang, X. Xue, Y. Jiang, and Q. Shen, “Deep learning for remote sensing image classification: A survey,” WIREs Data Mining and Knowledge Discovery, vol. 8, no. 6, pp. 1–17, 2018.
J. Song, S. Gao, Y. Zhu, and C. Ma, “A survey of remote sensing image classification based on CNNs,” Big Earth Data, vol. 3, no. 3, pp. 232–254, 2019.
G. Litjens et al., “A survey on deep learning in medical image analysis,” Medical Image Analysis, vol. 42, no. 1995, pp. 60–88, 2017.
H. Aerts, “Artificial Intelligence in Lung Cancer: Ready to Implement?,” pp. 1–16.
A. Srinivasulu et al., “RETRACTED ARTICLE: Advanced lung cancer prediction based on blockchain material using extended CNN,” Applied Nanoscience, vol. 13, no. 2, p. 985, 2023.
F. Sultana, A. Sufian, and P. Dutta, “A review of object detection models based on convolutional neural network,” Intelligent Computing: Image Processing Based Applications, vol. 1157, pp. 1–16, 2020.
C. B. Murthy, M. F. Hashmi, N. D. Bokde, and Z. W. Geem, “Investigations of object detection in images/videos using various deep learning techniques and embedded platforms comprehensive review,”Applied Sciences, vol. 10, no. 9, 2020.
H. Zhang and X. Hong, Recent progress on object detection: a brief review, vol. 78, no. 19. Multimedia Tools and Applications, 2019.
A. I¸sin, C. Direkoˇglu, and M. S¸ah, “Review of MRI-based Brain Tumor Image Segmentation Using Deep Learning Methods,” Procedia Computer Science, vol. 102, pp. 317–324, 2016.
S. Rehman, H. Ajmal, U. Farooq, Q. U. Ain, F. Riaz, and A. Hassan, “Convolutional neural network based image segmentation: a review,” no. April 2018, p. 26, 2018.
G. Du, X. Cao, J. Liang, X. Chen, and Y. Zhan, “Medical image segmentation based on U-Net: A review,” Journal of Imaging Science and Technology, vol. 64, no. 2, pp. 1–12, 2020.
T. Hoeser and C. Kuenzer, “Object detection and image segmentation with deep learning on Earth observation data: A review-part I: Evolution and recent trends,” Remote Sensing, vol. 12, no. 10, 2020.
S. S. Yadav and S. M. Jadhav, “Deep convolutional neural network based medical image classification for disease diagnosis,”Journal of Big Data , vol. 6, no. 113, 2019.
D. R. Sarvamangala and R. V. Kulkarni, “Convolutional neural networks in medical image understanding: a survey,” Evolutionary Intelligence , vol. 15, no. 1, pp. 1–22, 2022.
M. Bharath Simha Reddy and P. Rana, “Biomedical image classification using deep convolutional neural networks - Overview,” IOP Conference Series: Materials Science and Engineering, vol. 1022, no. 1, 2021.
J. Lu, L. Tan, and H. Jiang, “Review on convolutional neural network (CNN) applied to plant leaf disease classification,” Agriculture, vol. 11, no. 8, pp. 1–18, 2021.
K. L. Narayanan et al., “Banana Plant Disease Classification Using Hybrid Convolutional Neural Network,” Computational Intelligence and Neuroscience, vol. 2022, p. 9153699, 2022.
L. Falaschetti, L. Manoni, D. Di Leo, D. Pau, V. Tomaselli, and C. Turchetti, “A CNN-based image detector for plant leaf diseases classification,” HardwareX, vol. 12, p. e00363, 2022.
S. Cebollada, L. Pay´a, X. Jiang, and O. Reinoso, “Development and use of a convolutional neural network for hierarchical appearance-based localization,” Artificial Intelligence Review, vol. 55, no. 4, pp. 2847–2874, 2022.
Q. Zhang, J. Zhu, X. Sun, and M. Liu, “HTCGrasp: A Hybrid Transformer-CNN Architecture for Robotic Grasp Detection,” Electron, vol. 12, no. 6, 2023.
E. Watanyulertsakul, P. Khanthong, and B. Deeka, “Dementia U-Care: Comprehensive Cognitive Screening Application for Seniors,” ECTI-CIT Transactions, vol. 18, no. 2, pp. 207–221, Apr. 2024.
F. Angelucci, A. R. Ai, L. Piendel, J. Cerman, and J. Hort, “Integrating AI in fighting advancing Alzheimer: diagnosis, prevention, treatment, monitoring, mechanisms, and clinical trials,” Current Opinion in Structural Biology, vol. 87, p. 102857, Aug. 2024.
K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” Proc. IEEE Comput. Soc. Conf. Comput. Vis. Pattern Recognit., vol. 2016, pp. 770–778, 2016.
A. G. Howard et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications,” 2017, [Online]. Available: http://arxiv.org/abs/1704.04861.
K. Simonyan and A. Zisserman, “Intense convolutional networks for large-scale image recognition,” 3rd Int. Conf. Learn. Represent. ICLR 2015 - Conf. Track Proc., pp. 1–14, 2015.
C. Shorten and T. M. Khoshgoftaar, “A survey on Image Data Augmentation for Deep Learning,” Journal of Big Data , vol. 6, no. 1, 2019.
R. Poojary, R. Raina, and A. K. Mondal, “Effect of data-augmentation on fine-tuned cnn model performance,” IAES International Journal of Artificial Intelligence (IJ-AI), vol. 10, no. 1, pp. 84–92, 2021.
M. Das, M. Panda, and S. Dash, “Enhancing the Power of CNN Using Data Augmentation Techniques for Odia Handwritten Character Recognition,” Advances in Multimedia, vol. 2022, 2022.
M. Zhao, P. Shi, X. Xu, X. Xu, W. Liu, and H. Yang, “Identification System Using Different,” 2022.
D. A. Pitaloka, A. Wulandari, T. Basaruddin, and D. Y. Liliana, “Enhancing CNN with Preprocessing Stage in Automatic Emotion Recognition,” Procedia Computer Science, vol. 116, pp. 523–529, 2017.
S. H. S. Basha, V. Pulabaigari, and S. Mukherjee, “An information-rich sampling technique over spatiotemporal CNN for classification of human actions in videos,” Multimedia Tools and Applications, vol. 81, no. 28, pp. 40431–40449, 2022.
M. Gupta, K. Rajnish, and V. Bhattacharjee, “Impact of Parameter Tuning for Optimizing Deep Neural Network Models for Software Faults,” Scientific Programming, vol. 2021, p. 6662932, 2021.
L. Liao, H. Li, W. Shang, and L. Ma, “An Empirical Study of the Impact of Hyperparameter Tuning and Model Optimization on the Performance Properties of Deep Neural Networks,” ACM Transactions on Software Engineering and Methodology, vol. 31, no. 3, pp. 1-40, Apr. 2022.
T. Elsken, J. H. Metzen, and F. Hutter, “Survey on neural architecture search,” Journal of Machine Learning Research, vol. 26, no. 2, pp. 1–21, 2019.
A. Kwasigroch, M. Grochowski, and A. Mikolajczyk, “Neural architecture search for skin lesion classification,” IEEE Access, vol. 8, pp. 9061–9071, 2020.
M. Ahmad, M. Abdullah, H. Moon, S. J. Yoo, and D. Han, “Image classification based on automatic neural architecture search using binary crow search algorithm,” IEEE Access, vol. 8, pp. 189891–189912, 2020.
Y. Weng, T. Zhou, L. Liu, and C. Xia, “Automatic Convolutional Neural Architecture Search for Image Classification under Different Scenes,” IEEE Access, vol. 7, pp. 38495–38506, 2019.
Y. Weng, T. Zhou, Y. Li, and X. Qiu, “NAS-Unet: Neural architecture search for medical image segmentation,” IEEE Access, vol. 7, pp. 44247–44257, 2019.
M. Chatzianastasis, L. Ilias, D. Askounis, and M. Vazirgiannis, “Neural Architecture Search with Multimodal Fusion Methods for Diagnosing Dementia,” pp. 1–5, 2023.
J. Xin, A. Wang, R. Guo, W. Liu, and X. Tang, “CNN and swing-transformer based efficient model for Alzheimer’s disease diagnosis with sMRI,” Biomedical Signal Processing and Control, vol. 86, p. 105189, 2023.
P. Thangavel, Y. Natarajan, and K. R. Sri Preethaa, “EAD-DNN: Early Alzheimer’s disease prediction using deep neural networks,” Biomedical Signal Processing and Control, vol. 86, p. 105215, 2023.
A. V. Ambili, A. V. Senthil Kumar, and R. Latip, “CNN-Mobilenetv2-Deep Learning-Based Alzheimer’s Disease Prediction and Classification,” Journal of Theoretical and Applied Information Technology, vol. 101, no. 9, pp. 3590–3600, 2023.
R. Davuluri, V. Mallapragada, U. M. R. Mamillapalli, M. M, and S. Peeka, “Alzheimer’s Disease Diagnosis Using CNN Based Pre-trained Models,” International Journal on Recent and Innovation Trends in Computing and Communication, vol. 11, no. 4, pp. 315–323, 2023.
D. Agarwal, M. A. Berb´ıs, A. Luna, V. Lipari, J. B. Ballester, and I. de la Torre-D´ıez, “Automated Medical Diagnosis of Alzheimer’s Disease Using an Efficient Net Convolutional Neural Network,” Journal of Medical Systems, vol. 47, no. 1, 2023.
X. Long, L. Chen, C. Jiang, and L. Zhang, “Prediction and classification of Alzheimer disease based on quantification of MRI deformation,” PLoS One, vol. 12, no. 3, pp. 1–19, 2017.
E. N. Marzban, A. M. Eldeib, I. A. Yassine, and Y. M. Kadah, “Alzheimer’s disease diagnosis from diffusion tensor images using convolutional neural networks,” PLoS One, vol. 15, no. 3, pp. 1–16, 2020.
A. Punjabi, A. Martersteck, Y. Wang, T. B. Parrish, and A. K. Katsaggelos, “Neuroimaging modality fusion in Alzheimer’s classification using convolutional neural networks,” PLoS One, vol. 14, no. 12, pp. 1–14, 2019.
J. V. Shanmugam, B. Duraisamy, B. C. Simon, and P. Bhaskaran,“Alzheimer’s disease classification using pre-trained deep networks,” Biomedical Signal Processing and Control, vol. 71, p. 103217, 2022.
F. M. J. M. Shamrat, S. Akter, S. Azam, A. Karim, P. Ghosh, and Z. Tasnim, “AlzheimerNet: An Effective Deep Learning Based Proposition for Alzheimer’ s Disease Stages Classification From Functional Brain Changes in Magnetic Resonance Images,” IEEE Access, vol. 11, no. February, pp. 16376–16395, 2023.
V. Srinivasan, J. N. Dwivedi, S. Raja, and N. Rajeswari, “Prediction of Alzheimer’ s Disease Using DHO-Based Pretrained CNN Model Prediction of Alzheimer’ s Disease Using DHO-Based Pretrained,” Mathematical Problems in Engineering, Jun. 2023.
F. Hajamohideen, N. Shaffi, M. Mahmud, K. Subramanian, and A. Al Sariri, “Four-way classification of Alzheimer’s disease using a deep Siamese convolutional neural network with triplet-loss function,” Brain Informatics, vol. 10, no. 5, 2023.
A. Asgharzadeh-bona, H. Kalbkhani, and S. Azarfardian, “Healthcare Analytics An Alzheimer’s disease classification method using a fusion of features from brain Magnetic Resonance Image transforms and deep convolutional networks,” Healthcare Analytics, vol. 4, no. 100223, 2023.
M. Sethi and S. Ahuja, “Alzheimer disease classification using MRI images based on transfer learning,” vol. 040004, p. 040004, 2022.
S. V. Raghul and M. B. M. Kasim, “Dual tree complex wavelet transform based detection and classification of Alzheimer’s disease of MRI images using Artificial Neural Network (ANN) in comparison with Support Vector Machine (SVM),” International Conference on Environmental Engineering and Material Sciences: Iceems-2021, vol. 2655, no.1, p. 020098, 2023.
S. Samanta, I. Mazumder and C. Roy, “Deep Learning based Early Detection of Alzheimer’s Disease using Image Enhancement Filters,” 2023 Third International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies (ICAECT), Bhilai, India, pp. 1-5, 2023.
R. Joshi, P. Negi and T. Poongodi, “Multilabel Classifier Using DenseNet-169 for Alzheimer’s disease,” 2023 4th International Conference on Intelligent Engineering and Management (ICIEM), London, United Kingdom, pp. 1-7, 2023.
P. Lu, Y. Tan, Y. Xing, Q. Liang, X. Yan, and G. Zhang, “An Alzheimer’s disease classification method based on ConvNeXt,” 2023 3rd International Symposium on Computer Technology and Information Science (ISCTIS), Chengdu, China, pp. 884–888, 2023.
M. Kaur, C. Thacker, L. Goswami, T. TR, I. S. Abdulrahman, and A. S. Raj, “Alzheimer’s Disease Detection using Weighted KNN Classifier in Comparison with Medium KNN Classifier with Improved Accuracy,” 2023 3rd International Conference on Advance Computing and Innovative Technologies in Engineering (ICACITE), Greater Noida, India, pp. 715–718, 2023.
J. Wen, Y. Li, M. Fang, L. Zhu, D. D. Feng, and P. Li, “Fine-Grained and Multiple Classification for Alzheimer’s Disease with Wavelet Convolution Unit Network,” IEEE Transactions on Biomedical Engineering, vol. 70, no. 9, pp. 2592–2603, 2023.
B. Archana and K. Kalirajan, “Alzheimer’s Disease Classification using Convolutional Neural Networks,” Int. Conf. Innov. Data Commun. Technol. Appl. ICIDCA 2023 - Proc., pp. 1044–1048, 2023.
X. Liu et al., “BTSC-TNAS: A neural architecture search-based transformer for brain tumour segmentation and classification,” Computerized Medical Imaging and Graphics, vol. 110, no. 102307, Jun. 2023.
J. Ji and X. Wang, “Convolutional architecture search based on particle swarm algorithm for functional brain network classification[Formula presented],” Applied Soft Computing, vol. 149, no. 111049, Nov. 2023.
Y. Duan, Z. Wang, Y. Li, J. Tang, Y. K. Wang, and C. T. Lin, “Cross task neural architecture search for EEG signal recognition,” Neurocomputing, vol. 545, 2023.
J. Ma et al., “Distinguishing benign and malignant lesions on contrast-enhanced breast conebeam CT with deep learning neural architecture search,” European Journal of Radiology, vol. 142, no. 109878, Feb. 2021.
ADNI, “ADNI — Alzheimer’s Disease Neuroimaging Initiative,” https://adni.loni.usc.edu/ (accessed Jun. 21, 2023).
H. Jin, F. Chollet, Q. Song, and X. Hu, “AutoKeras: An AutoML Library for Deep Learning,” Journal of Machine Learning Research, vol. 24, pp. 1–6, 2023. [Online]. Available: http://jmlr.org/papers/v24/20-1355.html.