Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This study introduces a two-stage framework for leukemia cell classification that effectively addresses the challenges posed by domain shifts across multiple microscopy datasets. By leveraging a pretrained vision foundation model, the first stage performs binary classification on over 122,000 images, while the second stage refines subtype classification for Acute Lymphoblastic Leukemia and Acute Myeloid Leukemia using an additional 69,400 images. The results indicate that the proposed method, which incorporates retrieval-augmented classification, significantly enhances generalization performance across heterogeneous datasets, revealing the potential for cost-effective adaptation strategies in medical imaging tasks.
Retrieval-augmented classification boosts leukemia cell subtype identification accuracy across diverse datasets, challenging the need for expensive domain-specific pretraining.
Leukemia cell image classification is challenged by real-world domain shifts from acquisition, staining, illumination, and site protocols, causing single-dataset models to generalize poorly in real clinical scenarios. This work presents a robust framework for leukemia classification across multiple heterogeneous datasets using a two-stage pipeline with a pretrained vision foundation model. Stage 1 performs binary classification (leukemia vs. non-leukemia) and is trained using 122,167 single-cell images. Stage 2 is conditionally applied to Stage 1 positives to perform subtype classification into Acute Lymphoblastic Leukemia (ALL) and Acute Myeloid Leukemia (AML), trained using 69,400 single-cell images. Labels are harmonized across five heterogeneous datasets to enable cross-dataset training, and performance is evaluated on a held-out dataset protocol to assess domain-shift generalization. Within this pipeline, three encoders are benchmarked (DinoBloom, pretrained on single-cell images; BiomedCLIP, pretrained on biomedical data; and CLIP as a general-purpose model) under linear probing, Low-Rank Adaptation (LoRA), and a Retrieval-Augmented Classification (RAC) module that retrieves the top-k most similar cell images to provide cytomorphological grounding. The objective is to quantify how much domain-specific pretraining contributes to performance under domain shift, and whether cost-effective adaptation and retrieval can be a viable alternative to expensive domain-specialized pretraining. The held-out protocol additionally serves as a diagnostic tool, revealing when classification performance is attributable to dataset-specific artifacts rather than to cytomorphological features.