Abstract
Background
Convolutional neural networks (CNNs) have achieved remarkable success in medical image analysis, including Alzheimer’s disease (AD) classification. However, conventional convolution operations rely on fixed sampling patterns, and most existing attention mechanisms primarily focus on feature responses while neglecting spatial sampling geometry, limiting their ability to capture structural variations in brain images.
Methods
To address these limitations, this paper proposes a Geometry-Guided Deformable Attention Network (GGDA-Net) for medical image classification. The proposed framework integrates Linear Deformable Convolution (LDConv) with a Geometry-Aware (GA) Attention mechanism to jointly model feature semantics and spatial geometry. Specifically, LDConv introduces adaptive spatial sampling through learnable offsets, enabling flexible modeling of geometric deformations in brain structures, while the GA attention exploits the resulting geometric cues to guide the network toward more informative anatomical regions.
Results
The experimental results show that the accuracy rates on the two datasets reached 99.38 and 99.16% respectively, which are superior to the existing most advanced algorithms. At the same time, the model maintains a compact size and has a relatively low computational complexity. These results highlight the effectiveness of feature learning based on geometric perception in medical image analysis and Alzheimer’s disease diagnosis.
Keywords: Alzheimer’s disease, geometry-aware attention, GGDA-net, lightweight neural network, linear deformable convolution
1. Introduction
AD is the most common cause of dementia in the world and is a progressive irreversible neurodegenerative disease (Targa Dias Anastacio et al., 2022). The prevalence of all-cause dementia is projected to increase from 50 million people in 2010 to 113 million globally by 2050 (Knopman et al., 2021). Magnetic Resonance Imaging (MRI) (Leocadi et al., 2023) is widely used for the clinical assessment of AD, where characteristic structural changes such as progressive brain atrophy can be observed across cortical regions (Pini et al., 2016; Rao et al., 2023). In recent years, machine learning (Kavitha et al., 2022) and deep learning (Ebrahimighahnavieh et al., 2020) methods have significantly improved AD diagnosis by learning discriminative patterns from medical imaging data.
Convolutional neural networks (CNNs) have been widely adopted in medical image analysis, particularly for brain MRI–based Alzheimer’s disease diagnosis. Representative architectures such as AlexNet (Krizhevsky et al., 2017), VGG16 (Simonyan and Zisserman, 2015), ResNet50 (Helaly et al., 2022), and DenseNet (Huang et al., 2017), have achieved excellent performance in several visual tasks such as image classification, target detection (Zhang et al., 2022), and semantic segmentation (Neupane et al., 2021), and have driven significant advances in related technologies. However, despite their widespread success, CNNs still have limitations, most notably their constraints on the shape of the convolutional kernel and the sampling method.
To further enhance feature representation, attention mechanisms have been widely incorporated into convolutional networks. Studies such as Squeeze-and-Excitation (SENetV2) (Narayanan, 2023) focus on channel-wise feature recalibration, while the Bottleneck Attention Module (BAM) (Park et al., 2018) jointly models spatial and channel dependencies. Lightweight attention mechanisms such as ECA-Net (Wang et al., 2020) improve efficiency by avoiding dimensionality reduction, and recent efficient or linear attention methods (Zhuoran et al., 2021) aim to reduce computational complexity and improve scalability. However, most existing attention mechanisms rely solely on feature responses and lack explicit modeling of spatial sampling geometry.
To address these challenges, this paper integrates LDConv-based (Zhang et al., 2024) adaptive sampling with a GA mechanism to jointly model feature semantics and structural information, thereby enhancing representation capability in complex anatomical structures. The main contributions of this study are summarized as follows:
Linear deformable convolution: Unlike standard convolution, LDConv introduces learnable spatial offsets, enabling the network to adaptively sample and model irregular anatomical structures while maintaining linear parameter growth.
Geometry-aware attention: GA is proposed to model the interaction between feature responses and spatial sampling cues. By exploiting the geometric offsets generated during deformable sampling, the module guides the network to focus on more informative structural regions.
Efficient integration: We develop a GGDA-Net by integrating LDConv with the proposed GA module. LDConv enables adaptive spatial sampling through learnable offsets, while GA attention exploits the resulting geometric cues to perform geometry-aware feature reweighting.
Extensive experimental validation: The proposed method is evaluated on the ADNI dataset for Alzheimer’s disease classification. Experimental results demonstrate improved performance over representative deep learning models. These results validate that incorporating geometry-aware adaptive sampling is beneficial for capturing subtle structural variations in brain images.
The remainder of this paper is organized as follows. Section 2 reviews the relevant work on classifying Alzheimer’s disease using deep learning, with a focus on the shortcomings of convolutional neural networks (CNNs), architectures based on transformers, and attention mechanisms. Section 3 presents the proposed GGDA-Net methodology, including the LDConv module and GA attention mechanism. Section 4 describes the experimental setup, including datasets, implementation details, and evaluation metrics. Section 5 reports and discusses the experimental results, including comparisons with state-of-the-art methods and ablation studies. Finally, Section 6 concludes the paper and outlines future work.
2. Related work
Machine learning techniques have been widely explored for Alzheimer’s disease (AD) diagnosis using neuroimaging data. For instance, Ahmed et al. (2022) combined Boruta feature selection with gradient boosting trees (GBT) to achieve high-precision early prediction of AD, although the method still suffers from relatively high computational complexity. Other studies have explored various machine learning techniques, including support vector machines (Sharma et al., 2021), random forests (Aria et al., 2021), and K-means clustering (Ikotun et al., 2023), to analyze neuroimaging features. However, a comprehensive review (Mirzaei and Adeli, 2022) suggests that no single traditional machine learning approach consistently outperforms others across different datasets and tasks. Moreover, these methods generally rely on handcrafted feature extraction, which limits their ability to capture complex patterns in high-dimensional medical images.
With the rapid development of deep learning (DL), convolutional neural networks (CNNs) have become a dominant approach for medical image analysis (Goenka and Tiwari, 2021). Several studies have proposed CNN-based frameworks to improve diagnostic accuracy. For example, AbdulAzeem et al. (2021) introduced an end-to-end CNN framework for AD classification, while Helaly et al. (2022) designed a lightweight CNN architecture to reduce computational complexity and mitigate overfitting. El-Assy et al. (2024) proposed a dual-CNN structure to enhance the classification of AD subtypes and stages, and Hemanth et al. (2019) explored an efficient weight assignment strategy for fully connected layers to reduce computational cost. Neetha et al., based on the DEMNET framework, successively proposed the Borderline-DEMNET (Umalakshmi et al., 2024) and BD2EMNET (Neetha et al., 2024) models. Both of these models significantly improved the accuracy of the four-class and five-class classification tasks for Alzheimer’s disease by introducing the Borderline-SMOTE class balancing technique. Özkaraca et al. (2023) further designed a hierarchically dense CNN structure to accelerate model processing. Ahila et al. (2022) proposed a novel enhanced CAD system based on CNN to distinguish normal controls from AD patients. In addition to CNN-based approaches, transformer-based architectures have also attracted increasing attention in medical image analysis due to their capability to model long-range dependencies. For instance, Mahim et al. (2024) proposed a hybrid ViT–GRU model combined with explainable AI techniques, while Hu et al. (2023) developed the VGG-TSwinformer model for short-term longitudinal MCI analysis. These deep learning approaches have significantly improved feature learning capability and classification performance in AD-related image analysis.
Despite these advances, most existing models rely on fixed convolutional sampling patterns, which limit their ability to capture geometric deformations in complex brain structures. Furthermore, brain MRI images often exhibit intricate anatomical structures, irregular spatial patterns. Existing attention mechanisms primarily focus on feature responses while neglecting spatial sampling geometry, thereby restricting their capacity to model structural information in medical images. To address these limitations, this paper proposes a lightweight classification framework. Specifically, LDConv enables adaptive spatial sampling through learnable offsets, effectively capturing structural variations across different classes, while the GA attention mechanism leverages geometric offsets to guide the network toward discriminative regions.
3. Methodology
In this section, the proposed model is explained in detail and then the basic components used are described.
3.1. The proposed model
The overall framework of the proposed model follows a sequential pipeline that integrates backbone feature extraction, adaptive kernel convolution, geometry-aware attention, and classification, as shown Figure 1. Given an input medical image, the backbone network first extracts hierarchical feature representations, which encode both low-level texture and high-level semantic information. These features are then fed into the LDConv module, which serves as the core component for adaptive spatial modeling. Unlike conventional convolution with fixed sampling locations, LDConv introduces learnable offsets to dynamically adjust the sampling positions, enabling the convolution operation to better align with irregular anatomical structures. Specifically, the convolution process is formulated as y (p) = ∑Wk·x (p + k + Δpk), where Δpk denotes the learnable offset for the k-th sampling point. Within this process, LDConv generates two key outputs: the adaptive feature representation denoted as Fa, and the corresponding spatial offsets that describe the sampling geometry. The feature Fa represents the aggregated feature embedding after adaptive sampling, capturing spatially aligned structural information. Meanwhile, the offset term Δpk encodes the geometric deformation of sampling positions, which is further utilized to model spatial relationships. To enhance the representation capability, The GA Attention module takes both the adaptive feature Fa and the learned offsets Δp as inputs. Specifically, the offsets are further decomposed to extract deformation magnitude, enabling the attention mechanism to capture both spatial geometry and deformation intensity. The attention mechanism produces a geometry-aware weighting map A, which highlights informative regions while suppressing irrelevant responses. This process effectively integrates feature representation and geometric information into a unified representation. Subsequently, a global average pooling layer is applied to aggregate spatial information, followed by a classification head to generate the final prediction y. Through this pipeline, the proposed method achieves the research objective of improving spatial feature representation by combining adaptive sampling and geometry-aware attention, thereby enhancing the model’s ability to capture complex structural patterns in medical images.
Figure 1.
The overall architecture of the proposed GGDA-Net.
Given an input image , the backbone network first extracts feature representations. The extracted features are then refined by LDConv to enable adaptive spatial sampling. Based on the learned sampling geometry, GA attention module performs structure-conditioned feature reweighting. Finally, the enhanced features are fed into a classifier for prediction. The entire process can be represented by Equation 1:
| (1) |
3.2. Backbone feature extraction
The backbone serves as the primary feature extraction network, responsible for transforming the input image into discriminative feature representations. It extracts hierarchical spatial and semantic information that provides the foundation for the subsequent adaptive convolution and attention modules. Let the input image be denoted as , where , and denote the number of channels, height, and width of the input image, respectively.
Through a sequence of convolutional operations, the backbone network aggregates local spatial information and generates a deep feature representation. The convolution operation at spatial location p can be expressed as Equation 2:
| (2) |
where denotes the spatial position in the feature map, represents the predefined sampling offset within the convolution kernel, and denotes the corresponding convolution weight.
The resulting feature map encodes rich spatial and semantic information, which serves as the input to the proposed LDConv module for further geometry-guided feature learning.
3.3. Linear deformable convolution
Compared with standard convolution, LDConv allows convolution kernels with arbitrary numbers of parameters and flexible sampling shapes, providing a richer trade-off between model complexity and representation capability. LDConv modifies the parameter growth trend from quadratic to linear with respect to kernel size. When the number of parameters is set to , LDConv degenerates to the standard deformable convolution.
As illustrated in Figure 2, the LDConv module first predicts sampling offsets using a 3 × 3 convolution layer, generating the offset vector Δp. The offsets are then added to a predefined base sampling grid to obtain the deformed sampling positions. Since the resulting sampling coordinates are generally fractional, bilinear interpolation is applied to extract the corresponding feature values from the input feature map. The sampled features are subsequently rearranged to match the convolutional aggregation structure of LDConv. Finally, an N × 1 convolution is applied to aggregate the sampled features and produce the output feature map Y . This mechanism enables adaptive spatial sampling and flexible kernel shapes, allowing the network to better capture geometric variations in complex structures.
Figure 2.
Illustration of the LDConv module with adaptive sampling and feature aggregation.
Suppose the input feature map is , where denotes the number of channels, and H and W are the height and width of the feature map, respectively. For each spatial location , the convolution operation aggregates information from N sampling points. Instead of using a fixed k × k sampling grid, LDConv learns spatial offsets to generate adaptive sampling coordinates. The initial sampling set is defined by Equation 3:
| (3) |
The offset field is predicted from the input feature map, as shown in Equation 4:
| (4) |
Thus, the adaptive sampling coordinates become (as shown in Equation 5):
| (5) |
where, p: center position; pn: initial sampling location; Δpn: learned offset.
Since the updated coordinates may lie on fractional positions, bilinear interpolation is used to sample feature values (as shown in Equation 6):
| (6) |
where denotes the sampled feature at the n-th adaptive location.
The sampled features are aggregated using convolution weights (as shown in Equation 7)
| (7) |
where, represents the weight of the n-th sampling point.
Through this adaptive sampling mechanism, LDConv dynamically adjusts the spatial receptive field and improves the ability to capture irregular geometric patterns.
3.4. Geometry-aware attention
In this paper, the proposed GA module enhances spatial feature representation by explicitly incorporating geometric deformation cues derived from LDConv. As illustrated in Figure 3, unlike conventional attention mechanisms that estimate feature importance solely from feature activations, the GA module jointly models feature responses and spatial sampling geometry. Specifically, the module integrates three complementary components: feature embedding, geometry embedding derived from offset fields, and deformation magnitude embedding, which together capture both semantic information and structural deformation patterns.
Figure 3.
Architecture of the proposed geometry-aware attention (GA) module.
The proposed GA module enhances spatial feature representation by explicitly incorporating geometric deformation cues derived from the LDConv module. As shown in Figure 3, unlike conventional attention mechanisms that estimate feature importance solely from feature activations, the GA module jointly models feature responses and spatial sampling geometry. Specifically, the module integrates three complementary components: feature embedding, geometry embedding derived from offset fields, and deformation magnitude embedding, which together capture both semantic information and structural deformation patterns. The feature embedding captures the essential feature characteristics, while the geometry embedding comes from the offsets, which guide the sampling process to adapt to spatial variations. The deformation magnitude embedding represents the strength of deformation, which is crucial for capturing varying patterns of structural transformation.
The embedded representations are fused through a lightweight attention generation unit composed of a 1 × 1 convolution followed by a sigmoid activation to produce geometry-guided attention weights. By explicitly leveraging the geometric offsets generated during deformable sampling, the GA module enables the network to focus on structurally informative regions and better capture irregular spatial patterns. This design introduces structural inductive bias into the attention mechanism, allowing the model to more effectively handle anatomical variability and complex deformation patterns compared with conventional channel or spatial attention mechanisms.
The GA attention is introduced following the LDConv layer. Given the output feature map Y and the sampling offsets Δ produced by LDConv, the attention weights A are computed using Equation 8:
| (8) |
where , , are lightweight transformations for feature, geometry, and deformation magnitude, respectively, and is the sigmoid function.
The refined feature map is then calculated by Equation (9) as follows:
| (9) |
4. Experiment analysis
4.1. Data description
All data used in this study were obtained from the Alzheimer’s Disease Neuroimaging Initiative (ADNI) database. Two independent datasets were constructed from different ADNI phases to evaluate the generalization ability of the proposed method. The demographic information is summarized in Table 1.
Table 1.
Demographic details of different diagnostic groups in ADNI datasets.
| ADNI | ADNI2 | |||||||
|---|---|---|---|---|---|---|---|---|
| Class | Subject | Trian images | Test images | Images | Subject | Trian images | Test images | Images |
| AD | 89 | 1747 | 437 | 2,184 | 87 | 1,184 | 297 | 1,481 |
| CN | 116 | 2,426 | 607 | 3,033 | 87 | 1,388 | 347 | 1735 |
| MCI | 166 | 3,612 | 903 | 4,515 | 99 | 1,601 | 401 | 2002 |
| SMC | — | — | — | — | 64 | 1,060 | 266 | 1,326 |
| Total | 371 | 7,785 | 1947 | 9,732 | 336 | 5,233 | 1,311 | 6,544 |
4.1.1. Three-class dataset (ADNI)
The first dataset was derived from the complete one-year 1.5 T MRI data available in the ADNI-1 database. Subjects with complete baseline scans and clearly defined diagnostic labels were selected. The final dataset comprised 371 subjects across three diagnostic categories: Alzheimer’s disease (AD, n = 89), cognitively normal (CN, n = 116), and mild cognitive impairment (MCI, n = 166). The original data were stored in 3D NIfTI format. Each 3D volume was sliced along the coronal plane to generate a series of 2D images. Subsequently, each slice underwent window width and level adjustment to optimize contrast, followed by intensity normalization. Finally, all images were uniformly resized to 256 × 256 pixels and saved as PNG format.
4.1.2. Four-class dataset (ADNI2)
To construct a more challenging task that better aligns with fine-grained clinical diagnostic scenarios, a second dataset was built from the ADNI2 database. This dataset introduces the subjective memory complaint (SMC) category, an early-stage class, in addition to AD, CN, and MCI, thereby forming a four-class classification task (AD, n = 87; CN, n = 87; MCI, n = 99; SMC, n = 64). The original DICOM files were linearly normalized to a grayscale range of [0, 255]. Blank slices lacking brain parenchyma were excluded. All images were resized to 256 × 256 pixels.
Both datasets were split at the subject level into training and testing sets with a ratio of 80/20. Notably, as the two datasets originate from different ADNI phases with distinct acquisition protocols and subject cohorts, they naturally constitute a cross-dataset validation scenario. To further mitigate the risk of overfitting and enhance model generalization, data augmentation strategies including random rotation and random horizontal flipping were applied during training. As shown in Figure 4, representative MRI image examples from the ADNI dataset are presented. Top row (ADNI1): (a) AD, (b) CN, (c) MCI. Bottom row (ADNI2): (a) AD, (b) CN, (c) MCI, (d) SMC. All images are provided in PNG format to simplify subsequent processing.
Figure 4.
Different types of images on the ADNI datasets.
4.2. Implementation details
All experiments were conducted on a workstation running Windows 10 with 32 GB RAM and an NVIDIA RTX 4060 Ti GPU (16 GB memory). The proposed model was implemented using the PyTorch deep learning framework. The employed parameters for model training are shown in Table 2.
Table 2.
Parameters used for model training.
| Parameters | Settings |
|---|---|
| Hardware | GPU, CPU |
| Framework | PyTorch |
| Optimizer | Adam |
| Learning rate | 1e-4 |
| Batch size | 8 |
| Loss function | CrossEntropyLoss |
| Epochs | 150 |
During training, the Adam optimizer was employed with an initial learning rate of . The learning rate was dynamically adjusted using a decay strategy during the training process. The batch size was set to 8, and the model was trained for a maximum of 150 epochs. To improve generalization and prevent overfitting, an early stopping strategy based on validation accuracy was adopted. The cross-entropy loss function was used as the training objective to minimize the classification loss between the predicted outputs and the ground truth labels.
4.3. Evaluation metrics
This section firstly focuses on describing the influences on performance in the AD stage that are used to predict the classification of outcomes. The underlying outcomes for assessing the validity of classification systems cover four main categories: true positives (TP), true negatives (TN), false positives (FP), and false negatives (FN). The model performance was evaluated using several key metrics, including Precision (Pre), Recall (Rec), and F1-score (F1). The mathematical expressions for these metrics are shown below in Equations (10)–(13):
| (10) |
| (11) |
| (12) |
| (13) |
4.4. Comparison with classical networks
We compare the proposed model with several widely used deep learning architectures, including AlexNet, VGG16, ResNet50, DenseNet121, and Swin-Transformer, as shown in the Table 3.
Table 3.
Performance comparison with different deep learning architectures.
| Models | Pre (%) | Rec (%) | F1 (%) | Acc (%) | Para (m) | FLOPs (G) |
|---|---|---|---|---|---|---|
| AlexNet | 96.89 | 97.04 | 96.96 | 96.61 | 14.59 | 0.309 |
| VGG16 | 98.68 | 98.66 | 98.67 | 98.51 | 134.27 | 15.470 |
| ResNet50 | 99.32 | 99.21 | 99.26 | 99.18 | 23.51 | 4.132 |
| DenseNet121 | 99.21 | 99.14 | 99.17 | 99.08 | 6.957 | 2.896 |
| Swintransformer | 99.30 | 99.32 | 99.31 | 99.23 | 27.520 | 4.371 |
| Ours | 99.45 | 99.45 | 99.45 | 99.38 | 3.697 | 0.988 |
Bold values indicate the best performance among all compared methods.
Traditional networks such as AlexNet and VGG16 achieve relatively lower performance and require significantly larger model parameters. Modern architectures including ResNet50, DenseNet121, and Swin-Transformer achieve competitive results with accuracies exceeding 99%. However, the proposed model achieves the best performance, reaching 99.38% accuracy and 99.45% F1-score, while maintaining significantly lower model complexity (3.697 M parameters and 0.988G FLOPs).
Regarding the robustness of the model in the presence of class imbalance, further corroboration is provided by the confusion matrix in Figure 5. The matrix illustrates that a vast majority of instances across every category are correctly classified, with highly concentrated diagonal entries and near-zero off-diagonal misclassifications. This indicates that the model does not exhibit overfitting or bias toward any specific category. Even when a particular class has relatively fewer samples, the model can still accurately distinguish it based on highly discriminative features. Furthermore, from a quantitative perspective, the proposed model achieves perfectly equal precision (99.45%) and recall (99.45%), with the same F1-score of 99.45%. This perfect alignment between precision and recall implies that the model is insensitive to class distribution shifts: the high recall indicates that the model can identify the vast majority of positive samples, while the high precision indicates that the model rarely misclassifies other categories as the target category. Compared to AlexNet (Pre 96.89%/Rec 97.04%) and VGG16 (Pre 98.68%/Rec 98.66%), the proposed model achieves a better balance between precision and recall, further validating its superior robustness under class imbalance conditions.
Figure 5.
The test confusion matrix of different methods.
In summary, the confusion matrix in Figure 5 and the precision-recall consistency collectively demonstrate that the proposed method not only achieves the highest overall classification accuracy but also effectively enhances robustness to class imbalance while maintaining computational economy, thus realizing a favorable trade-off between high accuracy and strong robustness.
4.5. Ablation study on convolution architectures
To investigate the impact of different convolution strategies on model performance, several model variants were constructed by replacing the convolution operations while maintaining the same backbone architecture and attention modules (BAM and SENetV2). The evaluated variants include Model-Transpose, Model-Dynamic, Model-Group, and Model-LDConv, and the corresponding results are summarized in Table 4.
Table 4.
Performance comparison of different convolution architectures.
| Models | Convolution | Attention | Pre (%) | Rec (%) | F1 (%) | Acc (%) | Para (m) | FLOPs (G) |
|---|---|---|---|---|---|---|---|---|
| Model-Transpose | Transpose Conv | BAM + SENetv2 | 98.75 | 98.50 | 98.62 | 98.46 | 4.470 | 2.249 |
| Model-Dynamic | Dynamic Conv | BAM + SENetv2 | 98.40 | 98.23 | 98.32 | 98.15 | 7.960 | 0.170 |
| Model-Group | Group Conv | BAM + SENetv2 | 98.93 | 98.86 | 98.90 | 98.77 | 3.598 | 0.688 |
| Model-LDconv | LDConv | BAM + SENetv2 | 99.01 | 98.92 | 99.04 | 98.92 | 3.733 | 0.987 |
Model-Transpose achieves an accuracy of 98.46% but suffers from high computational complexity. Model-Dynamic substantially reduces FLOPs, yet introduces more parameters and yields inferior accuracy. Model-Group offers competitive performance with 98.77% accuracy while maintaining lower computational cost. By incorporating LDConv, which enables adaptive spatial sampling for better modeling of structural variations in medical images, the proposed Model-LDConv achieves the highest accuracy of 98.92% with reduced parameter complexity. Overall, these results highlight the importance of selecting appropriate convolution strategies.
4.6. Ablation study on attention modules
To further investigate the role of attention mechanisms in deformable feature modeling, we conduct an ablation study by progressively introducing LDConv (L), BAM (B), and SE (S), and the proposed GGDA-Net. The experimental results are presented in Table 5.
Table 5.
Ablation results of different attention mechanisms.
| Model | ||||||||
|---|---|---|---|---|---|---|---|---|
| LDConv (L) | BAM (B) | Se (S) | Pre (%) | Re (%) | F1 (%) | Acc (%) | Para (m) | FLOPs (G) |
| × | × | × | 98.29 | 98.22 | 98.25 | 98.05 | 4.421 | 2.222 |
| √ | × | × | 96.52 | 96.49 | 96.51 | 96.10 | 4.074 | 1.835 |
| √ | √ | × | 96.46 | 96.07 | 96.25 | 95.84 | 3.711 | 0.986 |
| √ | × | √ | 98.66 | 98.47 | 98.56 | 98.41 | 3.706 | 0.960 |
| × | √ | √ | 98.34 | 98.14 | 98.24 | 98.05 | 4.470 | 2.249 |
| √ | √ | √ | 99.01 | 98.92 | 99.04 | 98.92 | 3.733 | 0.987 |
The baseline model achieves 98.05% accuracy. When LDConv is introduced, the performance drops to 96.10%, indicating that although deformable convolution enhances geometric representation ability, the additional spatial flexibility may also introduce unstable feature responses without proper feature calibration.
To stabilize the deformable representations, conventional attention mechanisms are further explored. Incorporating SE attention together with LDConv improves the accuracy to 98.41%, demonstrating that channel-wise recalibration can partially regulate deformable features. In contrast, combining BAM with LDConv does not yield further improvement, suggesting that traditional spatial–channel attention is not fully compatible with geometry-aware representations. Moreover, stacking BAM and SE without LDConv shows limited benefit compared with the baseline, implying that conventional attention mechanisms are not sufficient to capture deformation-related information.
Motivated by these observations, we introduce the Geometry-guided Deformable Attention (GGDA) module, which explicitly incorporates geometric offsets and deformation magnitude into the attention computation. With the proposed design, the final model achieves the best performance with 99.38% accuracy and 99.45% F1-score, demonstrating the effectiveness of geometry-aware attention for deformable feature learning. The results of the ablation study are shown in Figure 6, where (a) compares validation accuracy (Val-Acc) across different models, and (b) compares validation loss (Val-Loss) across different models. These results suggest that attention mechanisms for deformable convolution should explicitly consider geometric deformation information, which motivates the design of the proposed GGDA module.
Figure 6.
Results description of accuracy-loss curves.
4.7. Ablation study on LDConv and GA attention modules
To further investigate the individual contribution of each component, ablation experiments were conducted on LDConv and GA attention, as shown in Table 6. The baseline model, which employs standard convolution, achieves an accuracy of 98.05%. When LDConv is introduced alone, the accuracy slightly decreases to 96.10%, which can be attributed to the additional difficulty of learning spatial offsets in deformable convolution. When GA attention is incorporated into the baseline model alone, the accuracy improves to 98.92%, demonstrating that GA attention effectively enhances feature representation. Notably, when LDConv and GA attention are jointly applied, the model achieves its best performance with 99.38% accuracy, while simultaneously reducing model complexity to 3.697 M parameters and 0.988G FLOPs.
Table 6.
Ablation results of different module combinations.
| Model | Pre (%) | Re (%) | F1 (%) | Acc (%) | Para (m) | FLOPs (G) |
|---|---|---|---|---|---|---|
| Baseline | 98.29 | 98.22 | 98.25 | 98.05 | 4.42 | 2.22 |
| LDconv | 96.52 | 96.49 | 96.51 | 96.10 | 4.074 | 1.835 |
| Baseline+GA | 98.98 | 99.04 | 99.01 | 98.92 | 4.435 | 2.261 |
| Ours | 99.45 | 99.45 | 99.45 | 99.38 | 3.697 | 0.988 |
These results indicate that GA attention effectively exploits the geometric information encoded in the offsets generated by LDConv, guiding the network to focus on more discriminative regions and thereby improving both model performance and efficiency. Specifically, LDConv provides geometric priors through adaptive sampling, while GA attention reweights these geometric cues, forming an effective synergistic mechanism.
4.8. Multi-class classification results
To further evaluate the robustness of the proposed model, experiments were conducted on both three-class and four-class classification tasks. Table 7 presents the performance of the proposed model on three-class and four-class classification tasks. Figure 7 presents the Grad-CAM visualization results for the proposed model. The highlighted regions indicate the areas that contribute most to the model’s prediction.
Table 7.
Performance comparison of the proposed model on three-class and four-class classification tasks.
| Task | Class | Pre (%) | Rec (%) | F1 (%) | Acc (%) |
|---|---|---|---|---|---|
| 3-class | AD | 100.00 | 100.00 | 100.00 | 99.38 |
| CN | 99.01 | 99.01 | 99.01 | ||
| MCI | 99.34 | 99.34 | 99.34 | ||
| 4-class | AD | 99.33 | 99.33 | 99.33 | 99.16 |
| CN | 99.42 | 99.14 | 99.28 | ||
| MCI | 98.52 | 99.50 | 99.01 | ||
| SMC | 99.62 | 98.50 | 99.05 |
Figure 7.
Grad-CAM visualization results of the proposed GGDANet model.
For the three-class task, the proposed model achieves excellent performance, with 100% precision, recall, and F1-score for the AD class. The CN and MCI classes also obtain high F1-scores of 99.01 and 99.34%, respectively, leading to an overall accuracy of 99.38%. For the more challenging four-class task, the model still maintains strong performance with 99.16% accuracy, achieving F1-scores of 99.33, 99.28, 99.01, and 99.05% for AD, CN, MCI, and SMC, respectively. The slight performance decrease in the four-class task is expected due to the increased classification difficulty.
It is worth noting that even in the four-class task where class sample sizes may be imbalanced (e.g., the SMC class often has fewer samples), the proposed model maintains highly consistent precision and recall across all classes. For instance, the SMC class achieves a precision of 99.62% and a recall of 98.50%, which are very close to each other. Similarly, the MCI class achieves a precision of 98.52% and a recall of 99.50%, also demonstrating a good balance. This consistency between precision and recall indicates that the model does not bias toward majority classes or neglect minority classes due to potential imbalanced data distributions. In other words, the model exhibits strong robustness to class distribution shifts, enabling fair and accurate discrimination across all classes even when sample sizes vary. These results collectively demonstrate that the proposed model not only excels in the three-class task but also exhibits high robustness and strong discriminative capability in the more complex four-class task. As shown in the Figure 7, the model mainly focuses on meaningful brain regions related to Alzheimer’s disease, demonstrating that the network can capture discriminative pathological features from MRI images. This visualization further confirms the interpretability and reliability of the proposed method.
Figure 8 shows the curves of training accuracy and test accuracy, where (a) presents the three-class classification results, and (b) presents the four-class classification results. In the three-class task, training accuracy increased from approximately 65 to 99%, while validation accuracy rose from 68 to 99%. The two curves remained highly synchronized throughout most epochs, with validation accuracy slightly higher than or comparable to training accuracy. In the four-class task, training accuracy improved from 38 to 99%, and validation accuracy from 46 to 99.2%, with only minor and transient fluctuations in a few epochs. The close alignment of the two curves indicates no significant overfitting, a stable learning process, and strong generalization capability.
Figure 8.
Training and validation accuracy curves for the three-class and four-class tasks.
5. Discussion
Compared with state-of-the-art models in Table 8, the proposed method achieves superior or competitive performance against existing approaches. The proposed GGDA framework improves the capability of the network to model complex structural features.
Table 8.
Comparative analysis with the state-of-the-art techniques.
| Number of classes | References | Methods | Acc (%) | Pre (%) | Rec (%) | F1 (%) |
|---|---|---|---|---|---|---|
| 4-class | Loddo et al. (2022) | Deep-Ensemble | 97.71 | — | — | — |
| Al-Adhaileh (2022) | AlexNet model | 94 | — | — | — | |
| Papanna Umalakshmi et al. (2024) | ADEMNET architecture | 95.45 | — | — | — | |
| Choudhury et al. (2024) | Coupled-GAN | 94.5 | 95 | — | 94 | |
| Hassan et al. (2025) | SCCAN | 99.22 | 99.21 | 99.21 | 99.22 | |
| Yilmaz and Kuş (2026) | GA-optimized | 96.43 | 96.50 | 96.44 | 96.45 | |
| Ours | GGDA-Net | 99.28 | 99.22 | 99.11 | 99.17 | |
| 3-class | Choi and Lee (2020) | DCNN ensemble | 93.84 | — | — | — |
| Chen et al. (2025) | MMDF | 97.65 | 96.98 | 96.40 | 96.69 | |
| Haq et al. (2025) | CNN-LSTM | 92.3 | 92.1 | — | 92.25 | |
| Ours | GGDA-Net | 99.38 | 99.45 | 99.45 | 99.45 |
Bold values indicate the best performance among all compared methods.
For the three-class task, the proposed model achieves 99.38% accuracy, along with 99.45% precision, recall, and F1-score, comprehensively outperforming all compared methods. Specifically, compared to Choi and Lee (2020) (93.84% accuracy with a DCNN ensemble), Chen et al. (2025) (97.65% accuracy, 96.98% precision, 96.40% recall with MMDF), and Haq et al. (2025) (92.3% accuracy, 92.1% precision with CNN-LSTM), the proposed model achieves substantial improvements of 5.54, 1.73, and 7.08%, respectively, in the Acc metric. These results fully demonstrate the superior capability of the proposed GGDA framework in modeling complex structural features.
For the four-class task, the proposed model achieves 99.28% accuracy, 99.22% precision, 99.11% recall, and 99.17% F1-score, ranking second only to the SCCAN method by Hassan et al. (99.22% accuracy). It is worth noting that the SCCAN method was published in 2025, and the proposed model achieves highly comparable performance with a marginal difference of only 0.06% in accuracy, while showing trade-offs in other metrics. More importantly, the proposed model significantly outperforms other compared methods: compared to Loddo et al. (2022) (97.71% with Deep-Ensemble), Al-Adhaileh et al. (2022) (94% with AlexNet), Papanna Umalakshmi et al. (2024) (95.45% with ADEMNET), Choudhury et al. (2024) (94.5% accuracy, 95% precision with Coupled-GAN), and Yilmaz and Kuş (2026) (96.43% accuracy with GA-optimized model), the proposed model achieves improvements of 1.57, 5.28, 3.83, 4.78, and 2.85%, respectively.
The proposed framework emphasizes three key contributions: (i) geometry-aware feature learning through LDConv and GA attention; (ii) lightweight and efficient architecture, facilitating deployment in clinical settings; and (iii) improved interpretability, with the ability to highlight anatomical regions relevant to AD progression.
6. Conclusion
This paper presents a novel deep learning architecture for Alzheimer’s disease classification based on MRI images. The proposed model integrates LDConv with a Geometry-guided deformable mechanism to enhance feature representation and spatial modeling capability. Specifically, LDConv introduces flexible sampling patterns through learnable offsets, enabling both regular and irregular convolution operations. When the kernel size is set to K2, the operation degenerates into standard deformable convolution, providing adaptable geometric feature extraction. Furthermore, the proposed GA attention effectively exploits the geometric information encoded in the offsets to guide the network toward more discriminative regions.
Future work will focus on improving the convolutional design by incorporating attention mechanisms to adaptively regulate feature extraction. Meanwhile, LDConv with more appropriate kernel sizes and sampling shapes will be explored. The proposed framework will also be extended to larger multi-center datasets to further evaluate its robustness and generalization capability.
Acknowledgments
We would like to thank ADNI for providing publicly available datasets.
Funding Statement
The author(s) declared that financial support was received for this work and/or its publication. This research is funded by the Major Science and Technology Projects of Henan Province (No. 221100210500), the Foundation of He′nan Educational Committee (No. 24A320004), the Medical and Health Research Project in Luoyang (No. 2001027A), and the Construction Project of Improving Medical Service Capacity of Provincial Medical Institutions in Henan Province (No. 2017–51).
Footnotes
Edited by: Fei Du, Harvard Medical School, United States
Reviewed by: Biswajit Brahma, McKesson, United States
P. U. Neetha, Indian Institute of Technology Kharagpur, India
Data availability statement
The original contributions presented in the study are included in the article/supplementary material, further inquiries can be directed to the corresponding authors.
Author contributions
DZ: Writing – review & editing, Methodology, Writing – original draft, Conceptualization, Software. JZ: Writing – original draft, Software, Conceptualization, Methodology. WC: Validation, Methodology, Visualization, Writing – original draft. GD: Validation, Investigation, Supervision, Writing – review & editing.
Conflict of interest
The author(s) declared that this work was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.
Generative AI statement
The author(s) declared that Generative AI was not used in the creation of this manuscript.
Any alternative text (alt text) provided alongside figures in this article has been generated by Frontiers with the support of artificial intelligence and reasonable efforts have been made to ensure accuracy, including review by the authors wherever possible. If you identify any issues, please contact us.
Publisher’s note
All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.
References
- AbdulAzeem Y., Bahgat W. M., Badawy M. (2021). A CNN based framework for classification of Alzheimer’s disease. Neural Comput. Applic 33, 10415–10428. doi: 10.1007/s00521-021-05799-w [DOI] [Google Scholar]
- Ahila A., Poongodi M., Hamdi M., Bourouis S., Rastislav K., Mohmed F. (2022). Evaluation of neuro images for the diagnosis of Alzheimer’s disease using deep learning neural network. Front. Public Health 10:834032. doi: 10.3389/fpubh.2022.834032, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Ahmed H., Soliman H., Elmogy M. (2022). Early detection of Alzheimer’s disease using single nucleotide polymorphisms analysis based on gradient boosting tree. Comput. Biol. Med. 146:105622. doi: 10.1016/j.compbiomed.2022.105622, [DOI] [PubMed] [Google Scholar]
- Al-Adhaileh M. H. (2022). Diagnosis and classification of Alzheimer’s disease by using a convolution neural network algorithm. Soft. Comput. 26, 7751–7762. doi: 10.1007/s00500-022-06762-0 [DOI] [Google Scholar]
- Aria M., Cuccurullo C., Gnasso A. (2021). A comparison among interpretative proposals for random forests. Mach. Learn. Appl. 6:100094. doi: 10.1016/j.mlwa.2021.100094 [DOI] [Google Scholar]
- Chen J., Wang Y., Zeb A., Suzauddola M. D., Wen Y. (2025). Multimodal mixing convolutional neural network and transformer for Alzheimer’s disease recognition. Expert Syst. Appl. 259:125321. doi: 10.1016/j.eswa.2024.125321 [DOI] [Google Scholar]
- Choi J. Y., Lee B. (2020). Combining of multiple deep networks via ensemble generalization loss, based on MRI images, for Alzheimer’s disease classification. IEEE Signal Process. Lett. 27, 206–210. doi: 10.1109/LSP.2020.2964161 [DOI] [Google Scholar]
- Choudhury C., Goel T., Tanveer M. (2024). A coupled-GAN architecture to fuse MRI and PET image features for multi-stage classification of Alzheimer’s disease. Inf. Fusion 109:102415. doi: 10.1016/j.inffus.2024.102415 [DOI] [Google Scholar]
- Ebrahimighahnavieh M. A., Luo S., Chiong R. (2020). Deep learning to detect Alzheimer’s disease from neuroimaging: A systematic literature review. Comput. Methods Prog. Biomed. 187:105242. doi: 10.1016/j.cmpb.2019.105242, [DOI] [PubMed] [Google Scholar]
- El-Assy A. M., Amer H. M., Ibrahim H. M., Mohamed M. A. (2024). A novel CNN architecture for accurate early detection and classification of Alzheimer’s disease using MRI data. Sci. Rep. 14:3463. doi: 10.1038/s41598-024-53733-6, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Goenka N., Tiwari S. (2021). Deep learning for Alzheimer prediction using brain biomarkers. Artif. Intell. Rev. 54, 4827–4871. doi: 10.1007/s10462-021-10016-0 [DOI] [Google Scholar]
- Haq E. U., Yong Q., Yuan Z., Huarong X., Haq R. U. (2025). Multimodal fusion diagnosis of the Alzheimer’s disease via lightweight CNN-LSTM model using magnetic resonance imaging (MRI). Biomed. Signal Process. Control 104:107545. doi: 10.1016/j.bspc.2025.107545 [DOI] [Google Scholar]
- Hassan N., Miah A. S. M., Suzuki K., Okuyama Y., Shin J. (2025). Stacked CNN-based multichannel attention networks for Alzheimer disease detection. Sci. Rep. 15:5815. doi: 10.1038/s41598-025-85703-x, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Helaly H. A., Badawy M., Haikal A. Y. (2022). Deep learning approach for early detection of Alzheimer’s disease. Cogn. Comput. 14, 1711–1727. doi: 10.1007/s12559-021-09946-2, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Hemanth D. J., Anitha J., Naaji A., Geman O., Popescu D. E., Hoang Son L. (2019). A modified deep convolutional neural network for abnormal brain image classification. IEEE Access 7, 4275–4283. doi: 10.1109/ACCESS.2018.2885639 [DOI] [Google Scholar]
- Hu Z., Wang Z., Jin Y., Hou W. (2023). VGG-TSwinformer: transformer-based deep learning model for early Alzheimer’s disease prediction. Comput. Methods Prog. Biomed. 229:107291. doi: 10.1016/j.cmpb.2022.107291, [DOI] [PubMed] [Google Scholar]
- Huang G., Liu Z., Van Der Maaten L., Weinberger K. Q. (2017). Densely Connected Convolutional Networks. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Honolulu: IEEE. [Google Scholar]
- Ikotun A. M., Ezugwu A. E., Abualigah L., Abuhaija B., Heming J. (2023). K-means clustering algorithms: a comprehensive review, variants analysis, and advances in the era of big data. Inf. Sci. 622, 178–210. doi: 10.1016/j.ins.2022.11.139 [DOI] [Google Scholar]
- Kavitha C., Mani V., Srividhya S. R., Khalaf O. I., Tavera Romero C. A. (2022). Early-stage Alzheimer’s disease prediction using machine learning models. Front. Public Health 10:853294. doi: 10.3389/fpubh.2022.853294, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Knopman D. S., Amieva H., Petersen R. C., Chételat G., Holtzman D. M., Hyman B. T., et al. (2021). Alzheimer disease. Nat. Rev. Dis. Primers 7:33. doi: 10.1038/s41572-021-00269-y, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Krizhevsky A., Sutskever I., Hinton G. E. (2017). ImageNet classification with deep convolutional neural networks. Commun. ACM 60, 84–90. doi: 10.1145/3065386 [DOI] [Google Scholar]
- Leocadi M., Canu E., Paldino A., Agosta F., Filippi M. (2023). Awareness impairment in Alzheimer’s disease and frontotemporal dementia: a systematic MRI review. J. Neurol. 270, 1880–1907. doi: 10.1007/s00415-022-11518-9 [DOI] [PubMed] [Google Scholar]
- Loddo A., Buttau S., Di Ruberto C. (2022). Deep learning based pipelines for Alzheimer’s disease diagnosis: A comparative study and a novel deep-ensemble method. Comput. Biol. Med. 141:105032. doi: 10.1016/j.compbiomed.2021.105032, [DOI] [PubMed] [Google Scholar]
- Mahim S. M., Ali M. S., Hasan M. O., Nafi A. A. N., Sadat A., Hasan S. A., et al. (2024). Unlocking the potential of XAI for improved Alzheimer’s disease detection and classification using a ViT-GRU model. IEEE Access 12, 8390–8412. doi: 10.1109/ACCESS.2024.3351809 [DOI] [Google Scholar]
- Mirzaei G., Adeli H. (2022). Machine learning techniques for diagnosis of Alzheimer disease, mild cognitive disorder, and other types of dementia. Biomed. Signal Process. Control 72:103293. doi: 10.1016/j.bspc.2021.103293 [DOI] [Google Scholar]
- Narayanan M. (2023). SENetV2: aggregated dense layer for channelwise and global representations. arXiv:10807. doi: 10.48550/arXiv.2311.10807 [DOI] [Google Scholar]
- Neetha P. U., Simran S., Pushpa C. N., Thriveni J., Venugopal K. R. (2024). Dealing with class imbalance and multi-class classification in Alzheimer’s disease and dementia: an innovative methodology. Int. J. Intell. Syst. Appl. Eng. 12:7028. doi: 10.17762/ijisae.v12i4.7028 [DOI] [Google Scholar]
- Neupane B., Horanont T., Aryal J. (2021). Deep learning-based semantic segmentation of urban features in satellite images: a review and meta-analysis. Remote Sens. 13:808. doi: 10.3390/rs13040808 [DOI] [Google Scholar]
- Özkaraca O., Bağrıaçık O. İ., Gürüler H., Khan F., Hussain J., Khan J., et al. (2023). Multiple brain tumor classification with dense CNN architecture using brain MRI images. Life 13:349. doi: 10.3390/life13020349, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Papanna Umalakshmi N., Sathyanarayana S., Chicktotlikere Nagappa P., Javarappa T., Kuppanna Rajuk V. (2024). ADEMNET architecture: an innovative solution for adaptive multi-class balancing problem in image classification. IJEECS 33:1252. doi: 10.11591/ijeecs.v33.i2.pp1252-1260 [DOI] [Google Scholar]
- Park J., Woo S., Lee J.-Y., Kweon I. S. (2018). BAM: bottleneck attention module. arXiv:e06514. doi: 10.48550/arXiv.1807.06514 [DOI] [Google Scholar]
- Pini L., Pievani M., Bocchetta M., Altomare D., Bosco P., Cavedo E., et al. (2016). Brain atrophy in Alzheimer’s disease and aging. Ageing Res. Rev. 30, 25–48. doi: 10.1016/j.arr.2016.01.002, [DOI] [PubMed] [Google Scholar]
- Rao G., Gao H., Wang X., Zhang J., Ye M., Rao L. (2023). MRI measurements of brain hippocampus volume in relation to mild cognitive impairment and Alzheimer disease: A systematic review and meta-analysis. Medicine 102:e34997. doi: 10.1097/MD.0000000000034997, [DOI] [PMC free article] [PubMed] [Google Scholar]
- Sharma A., Kaur S., Memon N., Jainul Fathima A., Ray S., Bhatt M. W. (2021). Alzheimer’s patients detection using support vector machine (SVM) with quantitative analysis. Neurosci. Inform. 1:100012. doi: 10.1016/j.neuri.2021.100012 [DOI] [Google Scholar]
- Simonyan K., Zisserman A. (2015). Very deep convolutional networks for large-scale image recognition. arXiv:1556. doi: 10.48550/arXiv.1409.1556 [DOI] [Google Scholar]
- Targa Dias Anastacio H., Matosin N., Ooi L. (2022). Neuronal hyperexcitability in Alzheimer’s disease: what are the drivers behind this aberrant phenotype? Transl. Psychiatry 12:257. doi: 10.1038/s41398-022-02024-7 [DOI] [PMC free article] [PubMed] [Google Scholar]
- Umalakshmi N. P., Sathyanarayana S., Chicktotlikere Nagappa P., Javarappa T., Kuppanna Rajuk V. (2024). Borderline-DEMNET: a workflow for detecting Alzheimer’s and dementia stage by solving class imbalance problem. JST 32, 1629–1650. doi: 10.47836/pjst.32.4.10 [DOI] [Google Scholar]
- Wang Q, Wu B, Zhu P, Li P, Zuo W, Hu Q. ECA-net: efficient Channel attention for deep convolutional neural networks. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle: IEEE; (2020). [Google Scholar]
- Yilmaz A., Kuş İ. (2026). General CNN model for biomedical image classification via genetic algorithm-based hyperparameter optimization. Ain Shams Eng. J. 17:103891. doi: 10.1016/j.asej.2025.103891 [DOI] [Google Scholar]
- Zhang X., Song Y., Song T., Yang D., Ye Y., Zhou J., et al. (2024). LDConv: linear deformable convolution for improving convolutional neural networks. Image Vis. Comput. 149:105190. doi: 10.1016/j.imavis.2024.105190 [DOI] [Google Scholar]
- Zhang P., Tang J., Zhong H., Ning M., Liu D., Wu K. (2022). Self-trained target detection of radar and sonar images using automatic deep learning. IEEE Trans Geosci Remote Sensing 60, 1–14. doi: 10.1109/TGRS.2021.3096011, 25079929 [DOI] [Google Scholar]
- Zhuoran S, Mingyuan Z, Haiyu Z, Shuai Y, Hongsheng L. Efficient attention: attention with linear complexities. (2021) IEEE Winter Conference on Applications of Computer Vision (WACV), Waikoloa: IEEE. [Google Scholar]
Associated Data
This section collects any data citations, data availability statements, or supplementary materials included in this article.
Data Availability Statement
The original contributions presented in the study are included in the article/supplementary material, further inquiries can be directed to the corresponding authors.








