Search papers, labs, and topics across Lattice.
This paper introduces SpurCon, a novel framework that utilizes a weighted supervised contrastive loss to address the challenges of spurious correlations in medical imaging datasets. By leveraging metadata and predicted spurious labels, SpurCon enhances model robustness without requiring extensive retraining, employing a fast few-shot procedure to estimate spurious labels from expert-annotated samples. The results demonstrate that SpurCon significantly improves worst-group and overall accuracy across various medical imaging datasets, including CheXpert and ISIC 2020, thereby increasing clinical trust in AI applications.
Achieving the best spurious-mitigation performance in medical imaging, SpurCon reshapes representation geometry to enhance model reliability with minimal retraining.
Despite the rapid progress of deep neural networks in visual recognition, their adoption in high-risk medical applications remains limited due to reliability and robustness concerns. Models may exploit spurious correlations, particularly in medical imaging, where devices or treatment artifacts often co-occur with pathology. In small or imbalanced datasets, such cues further reduce worst-group performance and undermine clinical trust. To solve these issues, two major challenges should be addressed: identifying dataset-specific spurious cues, which typically require domain knowledge, and mitigating reliance on them. To tackle both, we propose SpurCon, a lightweight framework based on a novel supervised contrastive loss formulation that leverages available metadata and predicted spurious labels to enhance robustness. We introduce a fast few-shot procedure, without network training, to estimate spurious labels using a small number of expert-annotated samples. We then propose a weighted supervised contrastive objective, WtSupCon, that reshapes the representation geometry by assigning sample-specific weights that depend on the [pathology, spurious, metadata] combination. For example, the highest weight is assigned to samples that differ only in their spurious label. This yields highly similar representations for images with the same metadata and pathology, differing only in the predicted spurious label. Our method operates on pretrained image encoders (such as BiomedCLIP) and trains only a lightweight projection head. We evaluate SpurCon on a synthetic setting and on Waterbirds, CheXpert, a chest X-ray classification dataset, and ISIC 2020, a skin cancer classification dataset. Our approach delivers the best spurious-mitigation performance, balancing well worst-group and overall accuracy on multiple datasets.
