Organizations: University of Technology, Ho Chi Minh City, Vietnam · Vietnam National University, Ho Chi Minh City, Vietnam · University of Science, Ho Chi Minh City, Vietnam · Carnegie Mellon University, Pittsburgh, PA, USA · Northwestern University, Chicago, IL, USA
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.
Figures & tables
Figure 1: Overview of our proposed framework for WBC classification
Method
MF1 ↑
BAcc ↑
MP ↑
MSpec ↑
Supervised (R50)
65.8±0.6
63.2±0.9
71.2±0.8
99.3±0.0
Supervised (R152)
68.7±2.2
66.1±1.1
75.5±5.4
99.4±0.0
Supervised (Swin)
71.2±1.4
69.0±1.3
76.4±3.8
99.4±0.0
Decoupled (R50)
70.4±2.6
72.0±2.3
70.0±3.4
99.3±0.0
Decoupled (R152)
70.6±2.1
70.9±1.4
71.9±4.0
99.4±0.0
Decoupled (Swin)
66.0±2.2
80.9±1.2
61.9±2.1
99.2±0.1
Table 1: Main comparison on the WBCBench 2026 Challenge benchmark. Best results are in bold ; second-best are underlined .
Figure 2: Confusion matrix of our best model on WBCBench 2026 Challenge Benchmark.
Department of Computer Science and Engineering, The Hong Kong University of Science and Technology, Hong Kong, China · Department of Chemical and Biological Engineering, The Hong Kong University of Science and Technology, Hong Kong, China · Division of Life Science, Hong Kong University of Science and Technology, Hong Kong, China +2
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