Search papers, labs, and topics across Lattice.
This paper addresses the challenges of WBC classification arising from staining variations and class imbalance by proposing a stain-normalized, decoupled training framework. The framework learns transferable representations using instance-balanced sampling and rebalances the classifier with class-aware sampling and a hybrid loss. The method achieves state-of-the-art performance on the WBCBench 2026 challenge through ensembling various trained backbones with test-time augmentation.
Stain normalization and decoupled learning can dramatically improve the robustness of white blood cell classification, even in the face of significant staining variations and class imbalances.
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 stain-normalized, 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 inference stage, we further enhance robustness by ensembling various 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.
