Fisher Contrastive Learning to Mitigate Feature Suppression: A Case Study in Acute Promyelocytic Leukemia Diagnosis
Organizations: Department of Statistics, University of Georgia · Department of Radiology, University of North Carolina at Chapel Hill · Department of Statistics, Harvard University
Abstract
Self-supervised contrastive learning (SSCL) learns representations by contrasting augmented views of the same sample, but it can exhibit feature suppression, where dominant or easy-to-learn factors suppress other class-relevant signals, degrading the performance of representations, particularly in high-dimensional data with heterogeneous features. We introduce Fisher Contrastive Learning (FCL), which uses a regularized Fisher trace criterion to address feature suppression. FCL learns embeddings by maximizing the discriminative strength of Fisher discriminant functions, producing representations that preserve class-relevant information implied by the augmentation process. We prove that the representations learned by FCL could alleviate the feature suppression effects. We demonstrate FCL on simulated benchmarks designed to induce feature suppression and on a real-world biomedical application, pretraining on large white blood cell microscopy image dataset and transfer learning for diagnosis of acute promyelocytic leukemia in an external cohort. In all experiments, FCL outperforms the self-supervised baselines and yields more robust and discriminative representations.
Figures & tables
| Task | SimCLR | FCL | VICReg | SimSiam | BYOL | DINO |
|---|---|---|---|---|---|---|
| Cell Type | 98.45% | 99.42% | 97.87% | 98.64% | 98.64% | 98.84% |
| Accuracy | AUROC | AUPRC | |
|---|---|---|---|
| MLP | 0.5833 | 0.7014 | 0.7494 |
| SimCLR | 0.6667 | 0.7153 | 0.7263 |
| FCL | 0.7500 | 0.8264 | 0.8405 |
| VICReg | 0.7083 | 0.7986 | 0.8332 |
| SimSiam | 0.6667 | 0.7014 | 0.7260 |
| BYOL | 0.6667 | 0.7292 | 0.7445 |
Contrastive Learning
Contrastive learning trains representations by bringing related examples together and separating unrelated ones for recognition, retrieval, and transfer.
See all 326 papers