Learning Hybrid Negative Probability Model for Weakly-Supervised Whole Slide Image Recognition.

Yining Qiu,Yuxi Li,Jiafu Wu,Zhenye Gan,Mingmin Chi,Yabiao Wang,Chengjie Wang,Pei Wang
DOI: https://doi.org/10.1109/ICASSP48485.2024.10448100
2024-01-01
Abstract:Classifying an entire Whole Slide Image (WSI) in a single forward pass is challenging due to its vast resolution. Consequently, current effort on WSI classification resorts to multiple instance learning (MIL), using patch-wise instances to predict categories under image-wise supervision. However, recent MIL methods usually follow implicit instance selection strategy and ignore the effect from inherent patch category imbalances. In a statistical sense, negative patches dominate in WSIs and provide sufficient samples for accurate density estimation. Therefore, in this paper, we learn from anomaly detection and propose a deep MIL framework which learns a hybrid negative probability model to bootstrap discovery of potential positive lesion. We associate attention-based MIL approach with a regularization loss function to explicitly improve patch selection process in positive images. Experiments conducted on benchmarks of WSI recognition demonstrate that our method brings significant improvement to classic attention-based MIL baseline and achieves state-of-the-art performance.
What problem does this paper attempt to address?