Wave-Aware Weakly Supervised Histopathological Tissue Segmentation With Cross-Scale Logits Distillation.

Abstract

Weakly supervised learning based on image-level labels can effectively reduce annotation costs, making it a popular choice for histopathological tissue segmentation. However, this pattern still face some challenges: 1) inaccurate class activation maps (CAM) make pseudo masks quality insufficient; 2) noisy pixels in pseudo masks will mislead the segmentation model's decision-making. To deal with these problems, we propose a novel weakly supervised semantic segmentation (WSSS) framework. First, we introduce Local Spatial Affine Perturbation to strengthen the model's utilization of weak supervision signals and improve its robustness to noisy regions within CAM. Second, we propose Wave-aware Dynamic Feature Aggregation to adaptively enhance the information-aware representation of target regions to obtain fine-grained pseudo masks enriched with positive semantic information. Third, we train a segmentation model with a noise-suppression scheme called Cross-scale Logits Distillation to reduce the inevitable false positive pixels in pseudo masks. We conduct extensive experiments to validate our method and set new state-of-the-art segmentation performances on five histopathological tissue segmentation datasets. Moreover, we will introduce a new dataset, GCSS-WSSS for gastric cancer, to promote the diversification for the research community of computational pathology. Code and data will be released at: https://github.com/director87/WaWeHis.