Cargando…

Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification

Three-dimensional convolutional neural networks (3D CNNs) have been widely applied to analyze Alzheimer’s disease (AD) brain images for a better understanding of the disease progress or predicting the conversion from cognitively impaired (CU) or mild cognitive impairment status. It is well-known tha...

Descripción completa

Detalles Bibliográficos
Autores principales: Xing, Xin, Rafique, Muhammad Usman, Liang, Gongbo, Blanton, Hunter, Zhang, Yu, Wang, Chris, Jacobs, Nathan, Lin, Ai-Ling
Formato: Online Artículo Texto
Lenguaje:English
Publicado: 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9910214/
https://www.ncbi.nlm.nih.gov/pubmed/36778519
http://dx.doi.org/10.3390/electronics12020467
_version_ 1784884737784938496
author Xing, Xin
Rafique, Muhammad Usman
Liang, Gongbo
Blanton, Hunter
Zhang, Yu
Wang, Chris
Jacobs, Nathan
Lin, Ai-Ling
author_facet Xing, Xin
Rafique, Muhammad Usman
Liang, Gongbo
Blanton, Hunter
Zhang, Yu
Wang, Chris
Jacobs, Nathan
Lin, Ai-Ling
author_sort Xing, Xin
collection PubMed
description Three-dimensional convolutional neural networks (3D CNNs) have been widely applied to analyze Alzheimer’s disease (AD) brain images for a better understanding of the disease progress or predicting the conversion from cognitively impaired (CU) or mild cognitive impairment status. It is well-known that training 3D-CNN is computationally expensive and with the potential of overfitting due to the small sample size available in the medical imaging field. Here we proposed a novel 3D-2D approach by converting a 3D brain image to a 2D fused image using a Learnable Weighted Pooling (LWP) method to improve efficient training and maintain comparable model performance. By the 3D-to-2D conversion, the proposed model can easily forward the fused 2D image through a pre-trained 2D model while achieving better performance over different 3D and 2D baselines. In the implementation, we chose to use ResNet34 for feature extraction as it outperformed other 2D CNN backbones. We further showed that the weights of the slices are location-dependent and the model performance relies on the 3D-to-2D fusion view, with the best outcomes from the coronal view. With the new approach, we were able to reduce 75% of the training time and increase the accuracy to 0.88, compared with conventional 3D CNNs, for classifying amyloid-beta PET imaging from the AD patients from the CU participants using the publicly available Alzheimer’s Disease Neuroimaging Initiative dataset. The novel 3D-2D model may have profound implications for timely AD diagnosis in clinical settings in the future.
format Online
Article
Text
id pubmed-9910214
institution National Center for Biotechnology Information
language English
publishDate 2023
record_format MEDLINE/PubMed
spelling pubmed-99102142023-02-09 Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification Xing, Xin Rafique, Muhammad Usman Liang, Gongbo Blanton, Hunter Zhang, Yu Wang, Chris Jacobs, Nathan Lin, Ai-Ling Electronics (Basel) Article Three-dimensional convolutional neural networks (3D CNNs) have been widely applied to analyze Alzheimer’s disease (AD) brain images for a better understanding of the disease progress or predicting the conversion from cognitively impaired (CU) or mild cognitive impairment status. It is well-known that training 3D-CNN is computationally expensive and with the potential of overfitting due to the small sample size available in the medical imaging field. Here we proposed a novel 3D-2D approach by converting a 3D brain image to a 2D fused image using a Learnable Weighted Pooling (LWP) method to improve efficient training and maintain comparable model performance. By the 3D-to-2D conversion, the proposed model can easily forward the fused 2D image through a pre-trained 2D model while achieving better performance over different 3D and 2D baselines. In the implementation, we chose to use ResNet34 for feature extraction as it outperformed other 2D CNN backbones. We further showed that the weights of the slices are location-dependent and the model performance relies on the 3D-to-2D fusion view, with the best outcomes from the coronal view. With the new approach, we were able to reduce 75% of the training time and increase the accuracy to 0.88, compared with conventional 3D CNNs, for classifying amyloid-beta PET imaging from the AD patients from the CU participants using the publicly available Alzheimer’s Disease Neuroimaging Initiative dataset. The novel 3D-2D model may have profound implications for timely AD diagnosis in clinical settings in the future. 2023-01-02 2023-01-16 /pmc/articles/PMC9910214/ /pubmed/36778519 http://dx.doi.org/10.3390/electronics12020467 Text en https://creativecommons.org/licenses/by/4.0/This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Xing, Xin
Rafique, Muhammad Usman
Liang, Gongbo
Blanton, Hunter
Zhang, Yu
Wang, Chris
Jacobs, Nathan
Lin, Ai-Ling
Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title_full Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title_fullStr Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title_full_unstemmed Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title_short Efficient Training on Alzheimer’s Disease Diagnosis with Learnable Weighted Pooling for 3D PET Brain Image Classification
title_sort efficient training on alzheimer’s disease diagnosis with learnable weighted pooling for 3d pet brain image classification
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9910214/
https://www.ncbi.nlm.nih.gov/pubmed/36778519
http://dx.doi.org/10.3390/electronics12020467
work_keys_str_mv AT xingxin efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT rafiquemuhammadusman efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT lianggongbo efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT blantonhunter efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT zhangyu efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT wangchris efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT jacobsnathan efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification
AT linailing efficienttrainingonalzheimersdiseasediagnosiswithlearnableweightedpoolingfor3dpetbrainimageclassification