cs.CVMar 17, 2026

Synergizing Deep Learning and Biological Heuristics for Extreme Long-Tail White Blood Cell Classification

Authors: Duc T. Nguyen, Hoang-Long Nguyen, Huy-Hieu Pham

Organizations: College of Engineering & Computer Science, VinUniversity, Hanoi, Vietnam · VinUni-Illinois Smart Health Center, VinUniversity, Hanoi, Vietnam · The Computer Vision and Medical AI Lab, VinUniversity, Hanoi, Vietnam

Abstract

Automated white blood cell (WBC) classification is essential for leukemia screening but remains challenged by extreme class imbalance, long-tail distributions, and domain shift, leading deep models to overfit dominant classes and fail on rare subtypes. We propose a hybrid framework for rare-class generalization that integrates a generative Pix2Pix-based restoration module for artifact removal, a Swin Transformer ensemble with MedSigLIP contrastive embeddings for robust representation learning, and a biologically-inspired refinement step using geometric spikiness and Mahalanobis-based morphological constraints to recover out-of-distribution predictions. Evaluated on the WBCBench 2026 challenge, our method achieves a Macro-F1 of 0.77139 on the private leaderboard, demonstrating strong performance under severe imbalance and highlighting the value of incorporating biological priors into deep learning for hematological image analysis. The code is available at https://github.com/trongduc-nguyen/WBCBench2026

Figures & tables

Explore similar work

CardsList
  1. WBCBench 2026: A Challenge for Robust White Blood Cell Classification Under Class Imbalance

    Apr 12, 2026Xin Tian, Xudong Ma, Tianqi Yang +4Medical Image ClassificationMedical Image Benchmarks

  2. A Hierarchical Ensemble Inference Pipeline for Robust White Blood Cell Classification Under Domain Shifts

    Apr 25, 2026Ruyi Dai, Tingkwong Ng, Hao ChenHierarchical ClassificationEnsemble Learning

  3. Robust White Blood Cell Classification with Stain-Normalized Decoupled Learning and Ensembling

    Mar 2, 2026Luu Le, Hoang-Loc Cao, Ha-Hieu Pham +2Medical Image ClassificationClass-Imbalanced Learning