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

• URL: http://arxiv.org/abs/2603.01976v1
• Date: Mon, 02 Mar 2026 15:31:25 GMT
• Title: Robust White Blood Cell Classification with Stain-Normalized Decoupled Learning and Ensembling
• Abstract summary: White blood cell (WBC) classification is fundamental for hematology applications such as infection assessment, leukemia screening, and treatment monitoring.<n>We propose a stain-normalized, decoupled training framework that learns transferable representations using instance-balanced sampling.<n>In inference stage, we further enhance robustness by ensembling various trained backbones with test-time augmentation.
• Score: 6.750164263871006
• License: http://creativecommons.org/licenses/by/4.0/
• 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 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.

Related papers

• A Hierarchical Ensemble Inference Pipeline for Robust White Blood Cell Classification Under Domain Shifts [5.200916021795385]
  We propose a memory-augmented, hierarchical ensemble pipeline for WBC classification under domain shifts.<n>Our method ranks within the top ten by macro F1-score in the final testing phase.
  arXiv  Detail & Related papers  (2026-04-25T12:25:46Z)
• WBCBench 2026: A Challenge for Robust White Blood Cell Classification Under Class Imbalance [10.424924450197137]
  WBCBench 2026 is an I SBI challenge and benchmark for automated WBC classification.<n>This paper reviews the challenge and summarises the proposed solutions and final outcomes.<n>We specify a standardised submission schema, open-source evaluator, and macro-averaged F1 score as the primary ranking metric.
  arXiv  Detail & Related papers  (2026-04-12T20:04:26Z)
• Ensemble of Small Classifiers For Imbalanced White Blood Cell Classification [9.420673339712211]
  We propose a lightweight ensemble approach for classification of cells during Haematopoiesis.<n>We show that a simple ensemble of lightweight pretrained SwinV2-Tiny, DinoBloom-Small and ConvNeXT-V2-Tiny models achieves excellent performance on this challenging dataset.
  arXiv  Detail & Related papers  (2026-03-21T15:24:52Z)
• SG-CLDFF: A Novel Framework for Automated White Blood Cell Classification and Segmentation [0.0]
  Saliency-Guided Cross-Layer Deep Feature Fusion framework (SG-CLDFF)<n>A lightweight hybrid backbone (Swin-style) produces multi-resolution representations, which are fused by a ResNeXt-CCinspired cross-layer fusion module.<n>Interpretability is enforced through Grad-CAM visualizations and saliency consistency checks, allowing model decisions to be inspected at the regional level.
  arXiv  Detail & Related papers  (2025-10-20T08:07:39Z)
• ActiveSSF: An Active-Learning-Guided Self-Supervised Framework for Long-Tailed Megakaryocyte Classification [3.6535793744942318]
  We propose the ActiveSSF framework, which integrates active learning with self-supervised pretraining.<n> Experimental results on clinical megakaryocyte datasets demonstrate that ActiveSSF achieves state-of-the-art performance.
  arXiv  Detail & Related papers  (2025-02-12T08:24:36Z)
• Uncertainty-guided Boundary Learning for Imbalanced Social Event Detection [64.4350027428928]
  We propose a novel uncertainty-guided class imbalance learning framework for imbalanced social event detection tasks. Our model significantly improves social event representation and classification tasks in almost all classes, especially those uncertain ones.
  arXiv  Detail & Related papers  (2023-10-30T03:32:04Z)
• Class-Imbalanced Graph Learning without Class Rebalancing [62.1368829847041]
  Class imbalance is prevalent in real-world node classification tasks and poses great challenges for graph learning models. In this work, we approach the root cause of class-imbalance bias from an topological paradigm. We devise a lightweight topological augmentation framework BAT to mitigate the class-imbalance bias without class rebalancing.
  arXiv  Detail & Related papers  (2023-08-27T19:01:29Z)
• A Continual Learning Approach for Cross-Domain White Blood Cell Classification [36.482007703764154]
  We propose a rehearsal-based continual learning approach for class incremental and domain incremental scenarios in white blood cell classification. To choose representative samples from previous tasks, we employ set selection based on the model's predictions. We thoroughly evaluated our proposed approach on three white blood cell classification datasets that differ in color, resolution, and class composition.
  arXiv  Detail & Related papers  (2023-08-24T09:38:54Z)
• Balanced Classification: A Unified Framework for Long-Tailed Object Detection [74.94216414011326]
  Conventional detectors suffer from performance degradation when dealing with long-tailed data due to a classification bias towards the majority head categories. We introduce a unified framework called BAlanced CLassification (BACL), which enables adaptive rectification of inequalities caused by disparities in category distribution. BACL consistently achieves performance improvements across various datasets with different backbones and architectures.
  arXiv  Detail & Related papers  (2023-08-04T09:11:07Z)
• Imbalanced Domain Generalization for Robust Single Cell Classification in Hematological Cytomorphology [3.7007225479462402]
  We train a robust CNN for WBC classification by addressing cross-domain data imbalance and domain shifts. Our approach achieves the best F1 macro score compared to other existing methods.
  arXiv  Detail & Related papers  (2023-03-14T10:20:31Z)
• Density-Aware Personalized Training for Risk Prediction in Imbalanced Medical Data [89.79617468457393]
  Training models with imbalance rate (class density discrepancy) may lead to suboptimal prediction. We propose a framework for training models for this imbalance issue. We demonstrate our model's improved performance in real-world medical datasets.
  arXiv  Detail & Related papers  (2022-07-23T00:39:53Z)
• Learning Discriminative Representation via Metric Learning for Imbalanced Medical Image Classification [52.94051907952536]
  We propose embedding metric learning into the first stage of the two-stage framework specially to help the feature extractor learn to extract more discriminative feature representations. Experiments mainly on three medical image datasets show that the proposed approach consistently outperforms existing onestage and two-stage approaches.
  arXiv  Detail & Related papers  (2022-07-14T14:57:01Z)
• PCCT: Progressive Class-Center Triplet Loss for Imbalanced Medical Image Classification [55.703445291264]
  Imbalanced training data is a significant challenge for medical image classification. We propose a novel Progressive Class-Center Triplet (PCCT) framework to alleviate the class imbalance issue. The PCCT framework works effectively for medical image classification with imbalanced training images.
  arXiv  Detail & Related papers  (2022-07-11T11:43:51Z)
• Dynamic Bank Learning for Semi-supervised Federated Image Diagnosis with Class Imbalance [65.61909544178603]
  We study a practical yet challenging problem of class imbalanced semi-supervised FL (imFed-Semi) This imFed-Semi problem is addressed by a novel dynamic bank learning scheme, which improves client training by exploiting class proportion information. We evaluate our approach on two public real-world medical datasets, including the intracranial hemorrhage diagnosis with 25,000 CT slices and skin lesion diagnosis with 10,015 dermoscopy images.
  arXiv  Detail & Related papers  (2022-06-27T06:51:48Z)
• Red Blood Cell Segmentation with Overlapping Cell Separation and Classification on Imbalanced Dataset [1.7219362335740878]
  Overlapping cells can cause incorrect predicted results that have to separate into multiple single RBCs before classifying. To classify multiple classes with deep learning, imbalance problems are common in medical imaging because normal samples are always higher than rare disease samples. This paper presents a new method to segment and classify red blood cells from blood smear images, specifically to tackle cell overlapping and data imbalance problems.
  arXiv  Detail & Related papers  (2020-12-02T16:49:51Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.