Return
Frequency-Aware Causal Regularization for Multiple Instance Learning in Whole Slide Image Classification
DOI:10.1109/tmi.2026.3697015.png)
Abstract
En 中文
Whole slide image (WSI) classification is a critical task in computational pathology and is aimed at providing automated diagnostic support through high-resolution tissue image analysis. In weakly supervised WSI classification scenarios, the main challenge concerns the traditional multiple instance learning (MIL) methods, which rely on instance-level embeddings aggregated by an attention-based pooling mechanism. These methods often depend on data-driven statistical correlations, leading to misalignments between their attention allocation schemes and histopathological diagnostic regions and reducing the resulting prediction reliability. To address this, we propose frequency-aware causal regularized multiple instance learning (FC-MIL), an innovative framework combining that combines frequency-aware attention (FAA) and causal regularization (CR). FAA extracts more granular, fine-grained histological textures by jointly modeling spatial- and frequency- domain features, whereas CR introduces feature-level counterfactual perturbations as an intervention-inspired regularizer in the latent space, encouraging the model to rely less on spurious correlations and more on invariant pathological cues. Experimental results obtained on four WSI datasets show that FC-MIL outperforms the state-of-the-art MIL methods in terms of both accuracy and interpretability. Our source code is available at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/7FFDW/FCMIL</uri>
Keywords:
Whole slide image classification
frequency-aware
causal regularization
multiple instance learning
Journal
IF:
9.8
Papers:
6.2K
Citations:
3.7W

