Mitigating Spurious Correlations in Patch-wise Tumor Classification on High-Resolution Multimodal Images

TL;DR

This paper addresses the challenge of spurious correlations in patch-wise tumor classification on high-resolution multimodal images, proposing a debiasing method that improves model fairness and accuracy, especially for minority cases.

Contribution

The study introduces a debiasing strategy using GERNE to mitigate spurious correlations in patch-wise binary tumor classification, enhancing worst-group accuracy.

Findings

WGA improved by approximately 7% with GERNE

Model performance increased on minority cases

Spurious correlation mitigation enhances fairness

Abstract

Patch-wise multi-label classification provides an efficient alternative to full pixel-wise segmentation on high-resolution images, particularly when the objective is to determine the presence or absence of target objects within a patch rather than their precise spatial extent. This formulation substantially reduces annotation cost, simplifies training, and allows flexible patch sizing aligned with the desired level of decision granularity. In this work, we focus on a special case, patch-wise binary classification, applied to the detection of a single class of interest (tumor) on high-resolution multimodal nonlinear microscopy images. We show that, although this simplified formulation enables efficient model development, it can introduce spurious correlations between patch composition and labels: tumor patches tend to contain larger tissue regions, whereas non-tumor patches often consist mostly of background with small tissue areas. We further quantify the bias in model predictions caused by this spurious correlation, and propose to use a debiasing strategy to mitigate its effect. Specifically, we apply GERNE, a debiasing method that can be adapted to maximize worst-group accuracy (WGA). Our results show an improvement in WGA by approximately 7% compared to ERM for two different thresholds used to binarize the spurious feature. This enhancement boosts model performance on critical minority cases, such as tumor patches with small tissues and non-tumor patches with large tissues, and underscores the importance of spurious correlation-aware learning in patch-wise classification problems.