Frequency-Aware Causal Regularization for Multiple Instance Learning in Whole Slide Image Classification
全切片图像分类中多实例学习的频率感知因果正则化
Abstract / 摘要
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 https://github.com/7FFDW/FCMIL.
全切片图像(WSI)分类是计算病理学中的一项关键任务,旨在通过高分辨率组织图像分析提供自动化诊断支持。在弱监督的WSI分类场景中,主要挑战在于传统的多实例学习(MIL)方法,这些方法依赖于基于注意力的池化聚合的实例级嵌入...