Skip to content
This repository was archived by the owner on Apr 25, 2026. It is now read-only.

FahimFBA/Alz-Predict-ROIvsNonROI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

46 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Important

This repository is archived and is no longer actively maintained. No further updates, bug fixes, or new experiments will be made. The code and results are preserved here for reference and reproducibility of the published paper. Please do not open issues expecting resolution; the repository is kept public solely as a research artifact.

Efficient Prediction for Alzheimer's Disease

Paper

This repository is based on the following peer-reviewed publication:

Paper Title: Enhanced ROI guided deep learning model for Alzheimer's detection using 3D MRI images

Journal: Informatics in Medicine Unlocked, Volume 56, 2025, Page 101650

DOI: https://doi.org/10.1016/j.imu.2025.101650

Read on:

Citation

@article{KHAN2025101650,
title = {Enhanced ROI guided deep learning model for Alzheimer's detection using 3D MRI images},
journal = {Informatics in Medicine Unlocked},
volume = {56},
pages = {101650},
year = {2025},
issn = {2352-9148},
doi = {https://doi.org/10.1016/j.imu.2025.101650},
url = {https://www.sciencedirect.com/science/article/pii/S2352914825000383},
author = {Israt Jahan Khan and Md. Fahim Bin Amin and Md. Delwar Shahadat Deepu and Hazera Khatun Hira and Asif Mahmud and Anas Mashad Chowdhury and Salekul Islam and Md. Saddam Hossain Mukta and Swakkhar Shatabda},
keywords = {Alzheimer's disease, Regions of Interest (ROIs), 3D MRI, Transfer learning, ResNet 3D, Efficient computation},
abstract = {Alzheimer's disease is an incurable condition that predominantly affects the human brain, leading to the shrinkage of various brain regions and the disruption of neuronal connections. Current state-of-the-art methods for detecting Alzheimer's disease using 3D MRI images are resource-intensive and time-consuming. In this paper, we propose a Regions of Interest (ROI)-guided detection paradigm to address these challenges. We employ a 3D ResNet integrated with a Convolutional Block Attention Module (CBAM), demonstrating that emphasising ROIs in brain imaging can substantially reduce both computational expenditure and training time. Our model exhibits robust performance in discriminating Alzheimer's disease from mild cognitive impairment, achieving an accuracy of 88% across the entire brain and 92% within targeted ROIs on the ADNI dataset. The accuracy on the OASIS dataset is even higher, reaching 98% for all regions and 98.33% for the ROIs. When distinguishing Alzheimer's disease from cognitively normal individuals, the accuracy improves further, achieving 93.33% for the ROIs on the ADNI dataset and 97.8% on the OASIS dataset. In differentiating cognitively normal individuals from those with mild cognitive impairment, the model attains an accuracy of 88.2% for the ROIs on the ADNI dataset and 98.6% on the OASIS dataset. These findings highlight a notable enhancement in detection accuracy through the utilisation of fewer, yet more salient brain regions, underscoring the efficacy of our ROI-guided approach.}
}

Short Demonstration

This project leverages advanced Convolutional Neural Networks (CNNs) for automated segmentation and analysis of 3D MRI brain images to detect Alzheimer's Disease. The core model architecture is a 3D ResNet enhanced with a Convolutional Block Attention Module (CBAM), significantly improving both model accuracy and interpretability.

Our workflow includes:

  • Preprocessing (bias correction, skull stripping, CLAHE)
  • Image registration and standardization
  • FastSurfer-based segmentation
  • ROI extraction
  • Extensive data augmentation

Datasets were obtained from ADNI and OASIS, and experiments focused on classifying subjects into:

  • Alzheimer's Disease (AD)
  • Mild Cognitive Impairment (MCI)
  • Cognitively Normal (CN)

This method is particularly effective for early-stage detection, guiding clinical decision-making through high-accuracy, ROI-focused models.

Overall Pipeline

Overall pipeline showing data collection, segmentation, preprocessing, augmentation, model training, and prediction

Model Architecture

3D ResNet with CBAM model architecture

Abstract

Alzheimer's disease is an incurable condition that predominantly affects the human brain, leading to the shrinkage of various brain regions and the disruption of neuronal connections. Current state-of-the-art methods for detecting Alzheimer's disease using 3D MRI images are resource-intensive and time-consuming. In this paper, we propose a Regions of Interest (ROI)-guided detection paradigm to address these challenges. We employ a 3D ResNet integrated with a Convolutional Block Attention Module (CBAM), demonstrating that emphasising ROIs in brain imaging can substantially reduce both computational expenditure and training time. Our model exhibits robust performance in discriminating Alzheimer's disease from mild cognitive impairment, achieving an accuracy of 88% across the entire brain and 92% within targeted ROIs on the ADNI dataset. The accuracy on the OASIS dataset is even higher, reaching 98% for all regions and 98.33% for the ROIs. When distinguishing Alzheimer's disease from cognitively normal individuals, the accuracy improves further, achieving 93.33% for the ROIs on the ADNI dataset and 97.8% on the OASIS dataset. In differentiating cognitively normal individuals from those with mild cognitive impairment, the model attains an accuracy of 88.2% for the ROIs on the ADNI dataset and 98.6% on the OASIS dataset. These findings highlight a notable enhancement in detection accuracy through the utilisation of fewer, yet more salient brain regions, underscoring the efficacy of our ROI-guided approach.

Datasets

Warning

Dataset availability notice: The preprocessed ADNI and OASIS datasets that were previously hosted publicly on Kaggle have been permanently deleted. Both ADNI and OASIS contacted us directly and requested the immediate removal of the data from all public platforms, and we complied accordingly. Additionally, the CSV files containing subject IDs were also removed following a separate request received via email from ADNI. We do not retain copies of these datasets in any form. To access the original data, researchers must apply directly through the official ADNI and OASIS portals.

In the first phase, we gathered 3D MRI datasets from ADNI and OASIS, renowned initiatives for advancing the understanding of Alzheimer's disease. We categorized the data into Alzheimer's Disease (AD), Mild Cognitive Impairment (MCI), and Cognitively Normal (CN) classes. Additionally, we obtained CSV files containing detailed patient information for labeling. Strict privacy rules were followed to ensure the security of sensitive medical data.

Dataset Organization

For our research, we categorized the dataset into three distinct groups: AD (Alzheimer's Disease), CN (Cognitively Normal), and MCI (Mild Cognitive Impairment).

Dataset collection and organization flow for ADNI and OASIS

ADNI

We utilized the ADNI Screening 1.5T dataset, which includes subjects who underwent a screening scan. We collected Neuroimaging Informatics Technology Initiative (NIfTI) images from 2294 subjects.

OASIS

The dataset includes 416 patients aged 18 to 96. Each person undergoes three or four T1-weighted MRI scans in a single session. Among the participants over sixty, 100 have been diagnosed with Alzheimer's disease. Additionally, there is a reliability dataset with 20 nondemented participants.

Dataset Preprocessing

Full preprocessing pipeline including FastSurfer segmentation, ROI classification, augmentation, and model training

FastSurfer Segmentation

FastSurfer CNN architecture showing axial, coronal, and sagittal processing streams

Experiments

Warning

Missing notebooks — experiments 3, 4, 9, and 10: The notebooks for these four experiments were accidentally lost and are no longer recoverable. Because the experiments were extremely computationally expensive (large-scale 3D MRI training runs), we did not rerun them after the loss. The reported metrics were recorded at the time of the original runs and remain accurate as published in the paper.

Experiment Sl. Experiment Name Accuracy Val. Accuracy Notebook Model
1 ADNI - AD vs MCI 88% 88% View -- Notebook Resnet18 with CBAM
2 ADNI (6 ROIs) - AD vs MCI 92% 98.33% View -- Notebook Resnet18 with CBAM
3 OASIS - AD vs MCI 98% 98.3% Notebook accidentally lost — not rerun due to high compute cost Resnet18 with CBAM
4 OASIS (6 ROIs) - AD vs MCI 98.33% 97.78% Notebook accidentally lost — not rerun due to high compute cost Resnet18 with CBAM
5 ADNI - AD vs CN 88.89% 97.5% View -- Notebook Resnet18 with CBAM
6 ADNI (6 ROIs) - AD vs CN 93.33% 95.63% View -- Notebook Resnet18 with CBAM
7 OASIS - AD vs CN 100% 100% View -- Notebook Resnet18 with CBAM
8 OASIS (6 ROIs) - AD vs CN 97.8% 99.5% View -- Notebook Resnet18 with CBAM
9 ADNI - CN vs MCI 87.27% 89.5% Notebook accidentally lost — not rerun due to high compute cost Resnet18 with CBAM
10 ADNI (6 ROIs) - CN vs MCI 88.2% 91.03% Notebook accidentally lost — not rerun due to high compute cost Resnet18 with CBAM
11 OASIS - CN vs MCI 98.05% 98.89% View -- Notebook Resnet18 with CBAM
12 OASIS (6 ROIs) - CN vs MCI 98.6% 99.6% View -- Notebook Resnet18 with CBAM

Recreating the Conda Environment for this Project

This repository includes files to help you recreate the development environment used in this project. Follow the instructions below to set up the environment on your local machine.

Files Included

  • environment.yml: Captures the full Conda environment, including both Conda and pip-installed packages.
  • requirements.txt: Lists only pip-installed packages, useful for users who prefer pip-based installations.

1. Recreate the Conda Environment

Prerequisites

Steps

  1. Open a terminal and navigate to the project directory.
  2. Run the following command to create the environment:
    conda env create -f environment.yml
  3. Activate the newly created environment:
    conda activate fydp-old
  4. Verify the environment:
    conda list

2. Using the requirements.txt File (Pip-Only Option)

Prerequisites

  • Ensure Python and pip are installed on your system.

Steps

  1. Create a virtual environment:
    python -m venv venv
  2. Activate the virtual environment:
    • On Linux/macOS:
      source venv/bin/activate
    • On Windows:
      venv\Scripts\activate
  3. Install dependencies:
    pip install -r requirements.txt
  4. Verify the installation:
    pip list

Notes

  • Preferred Method: The environment.yml file is recommended as it fully captures both Conda and pip-installed packages, ensuring 100% reproducibility.
  • Pip Limitation: The requirements.txt file only includes pip-installed packages and may not recreate Conda-specific dependencies or system-level requirements.

Contact

This repository is archived and no further maintenance is planned. If you have any questions regarding the research, the methodology, or the published paper, please contact the team lead:

Md. Fahim Bin Amin

Email: fahimbinamin@gmail.com

About

Deep learning pipeline for efficiently predict Alzheimer's Disease from ADNI & OASIS

Topics

Resources

License

Stars

Watchers

Forks

Contributors