Search papers, labs, and topics across Lattice.
This paper investigates one-class representation learning (OCRL) methods, specifically DSVDD and DROC, for detecting rare malignant cells in whole-slide cytological images where malignant cells are extremely rare and instance-level annotations are limited. The OCRL models are trained only on slide-negative patches to learn a representation of normal cells, enabling the detection of deviations indicative of malignancy during inference. Experiments on bone marrow and oral cancer datasets demonstrate that DSVDD achieves state-of-the-art instance-level abnormality ranking, particularly in ultra-low witness-rate regimes, and can even surpass fully supervised learning.
Forget fully supervised learning: one-class representation learning can now outperform it for detecting extremely rare malignant cells in cytological images, even with witness rates as low as 1%.
In computational cytology, detecting malignancy on whole-slide images is difficult because malignant cells are morphologically diverse yet vanishingly rare amid a vast background of normal cells. Accurate detection of these extremely rare malignant cells remains challenging due to large class imbalance and limited annotations. Conventional weakly supervised approaches, such as multiple instance learning (MIL), often fail to generalize at the instance level, especially when the fraction of malignant cells (witness rate) is exceedingly low. In this study, we explore the use of one-class representation learning techniques for detecting malignant cells in low-witness-rate scenarios. These methods are trained exclusively on slide-negative patches, without requiring any instance-level supervision. Specifically, we evaluate two OCC approaches, DSVDD and DROC, and compare them with FS-SIL, WS-SIL, and the recent ItS2CLR method. The one-class methods learn compact representations of normality and detect deviations at test time. Experiments on a publicly available bone marrow cytomorphology dataset (TCIA) and an in-house oral cancer cytology dataset show that DSVDD achieves state-of-the-art performance in instance-level abnormality ranking, particularly in ultra-low witness-rate regimes ($\leq 1\%$) and, in some cases, even outperforming fully supervised learning, which is typically not a practical option in whole-slide cytology due to the infeasibility of exhaustive instance-level annotations. DROC is also competitive under extreme rarity, benefiting from distribution-augmented contrastive learning. These findings highlight one-class representation learning as a robust and interpretable superior choice to MIL for malignant cell detection under extreme rarity.