Robust White Blood Cell Classification With Stain-Normalized Decoupled Learning and Ensembling

View at publisher
Authors: Luu Van Le, Hoang-Loc Cao, Ha-Hieu Pham, T U O N G Nguyen, Ulas Bagci
Cited by: 1
References: 11

Abstract

White blood cell (WBC) classification is fundamental for hematology applications such as infection assessment, leukemia screening, and treatment monitoring. However, real-world WBC datasets present substantial appearance variations caused by staining and scanning conditions, as well as severe class imbalance in which common cell types dominate while rare but clinically important categories are underrepresented. To address these challenges, we propose a stainnormalized, decoupled training framework that first learns transferable representations using instance-balanced sampling, and then rebalances the classifier with class-aware sampling and a hybrid loss combining effective-number weighting and focal modulation. In the inference stage, we further enhance robustness by ensembling multiple trained backbones with test-time augmentation. Our approach achieved the top rank on the leaderboard of the WBCBench 2026: Robust White Blood Cell Classification Challenge at ISBI 2026. Our code implementation is publicly available at https://github.com/luulecmg/WBCBench2026

Fields of Study & Subjects

Digital Imaging for Blood Diseases Machine Learning and ELM AI in cancer detection

Publication details

Journal / source
2026 IEEE 23rd International Symposium on Biomedical Imaging (ISBI)
Publisher
IEEE
Pages
1–4
Publication date
8 April 2026