mlsanityData to 14 Oct 2026

Multi-Attention Stacked Ensemble for Lung Cancer Detection in CT Scans

arXiv 2507.20221 ↗First posted 27 Jul 2025eess.IV cs.AI cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

In this work, we address the challenge of binary lung nodule classification (benign vs malignant) using CT images by proposing a multi-level attention stacked ensemble of deep neural networks. Three pretrained backbones -- EfficientNet V2 S, MobileViT XXS, and DenseNet201 -- are each adapted with a custom classification head tailored to 96 x 96 pixel inputs. A two-stage attention mechanism learns both model-wise and class-wise importance scores from concatenated logits, and a lightweight meta-learner refines the final prediction. To mitigate class imbalance and improve generalization, we employ dynamic focal loss with empirically calculated class weights, MixUp augmentation during training, and test-time augmentation at inference. Experiments on the LIDC-IDRI dataset demonstrate exceptional performance, achieving 98.09 accuracy and 0.9961 AUC, representing a 35 percent reduction in error rate compared to state-of-the-art methods. The model exhibits balanced performance across sensitivity (98.73) and specificity (98.96), with particularly strong results on challenging cases where radiologist disagreement was high. Statistical significance testing confirms the robustness of these improvements across multiple experimental runs. Our approach can serve as a robust, automated aid for radiologists in lung cancer screening.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 38 references that are in the map. First posted on arXiv 27 Jul 2025.

Builds on

10 of 38 references.

References · 10

  1. 2016Densely Connected Convolutional Networkscited by 6,720
  2. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
  3. 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,186
  4. 2021EfficientNetV2: Smaller Models and Faster Trainingcited by 1,135
  5. 2017Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networkscited by 939
  6. 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 642
  7. 2019Selective Kernel Networkscited by 403
  8. 2019Transfusion: Understanding Transfer Learning for Medical Imagingcited by 368
  9. 2017Evaluate the Malignancy of Pulmonary Nodules Using the 3D Deep Leaky Noisy-or Networkcited by 53
  10. 2018Attention-Gated Networks for Improving Ultrasound Scan Plane Detectioncited by 22

Built on by

Not cited yet.