MalariAI: A Label-Resilient Decoupled Framework for Annotation-Agnostic Cell Segmentation and Explainable Stage Classification in Dense Malaria Blood Smears
Not on the map. Why?
Abstract (arXiv)
Automated malaria diagnosis from blood smear microscopy is a critical global health AI challenge; expert scarcity remains the primary diagnostic bottleneck. Existing deep learning systems face three compounding failures: end-to-end detectors treat unannotated cells as background, skewing recall by annotation completeness rather than true cell recovery; Non-Maximum Suppression suppresses valid detections in dense smears; and pipelines lack per-cell spatial evidence for clinical audit. We present MalariAI, a two-stage decoupled framework addressing all three. Stage 1 applies an annotation-agnostic watershed algorithm to isolate every cell in a full 1600x1200 image, recovering 75.95% of ground-truth cells without any ground-truth input. End-to-end, the pipeline reaches a binary parasitized [email protected] of 29.10% - the clinically relevant metric for flagging any infected cell - while the stricter multi-class [email protected] of 8.67% mainly reflects watershed's organic region boundaries being penalized against axis-aligned ground-truth boxes, not a localisation failure. Stage 2 fine-tunes EfficientNet-B0 with Focal Loss on ground-truth crops, achieving 98.36% classification accuracy - an oracle upper bound once a cell is correctly localised - with 87.5% and 75.0% accuracy on the rare schizont and gametocyte stages, versus 38.45% and 57.27% AP for a modern YOLOv8s detector evaluated end-to-end on the same classes. Grad-CAM++ heatmaps generated per detected cell provide instance-level spatial evidence for clinical audit; a quantitative energy-in-box analysis confirms this activation is concentrated on the annotated cell body significantly above a geometric chance baseline (+0.0485, paired p = 1.4 x 10^-33), letting microscopists verify predictions at the individual parasite level without sacrificing classification performance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 54 references that are in the map. First posted on arXiv 1 Jul 2026.
Builds on
21 of 54 references.
References · 21
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,639
- 2023Segment Anythingcited by 8,234
- 2017Mask R-CNNcited by 7,277
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,733
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2015SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentationcited by 1,487
- 2017Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networkscited by 939
9 more
- 2022DN-DETR: Accelerate DETR Training by Introducing Query DeNoisingcited by 488
- 2021Domain Adaptation for Medical Image Analysis: A Surveycited by 352
- 2021Explainable artificial intelligence (XAI) in deep learning-based medical image analysiscited by 147
- 2023CellViT: Vision Transformers for Precise Cell Segmentation and Classificationcited by 108
- 2021Unified Focal loss: Generalising Dice and cross entropy-based losses to handle class imbalanced medical image segmentationcited by 88
- 2020Black-box Explanation of Object Detectors via Saliency Mapscited by 76
- 2018Applying Faster R-CNN for Object Detection on Malaria Imagescited by 15
- 2026Disco: Densely-overlapping Cell Instance Segmentation via Adjacency-aware Collaborative Coloringcited by 2
- 2023Tackling the Incomplete Annotation Issue in Universal Lesion Detection Task By Exploratory Trainingcited by 1
Built on by
Not cited yet.