Collaborative instance-level and bag-level multiple instance learning with label disambiguation for whole slide image...

Artificial intelligence (AI)-driven histopathological image analysis has shown significant advantages for disease diagnosis, prognosis, and treatment planning, and it is receiving growing attention in modern healthcare. Due to the gigapixel size of whole slide images (WSIs), multiple instance learning (MIL) methods are widely employed in their analysis. Existing MIL approaches primarily rely on either instance-level or bag-level supervision, each facing challenges related to noisy pseudo-labels and suboptimal feature aggregation, respectively. In this paper, we present a novel MIL method for WSI analysis, termed CIB-MIL, which integrates collaborative instance-level and bag-level supervision. We introduce a label disambiguation module within the instance-level supervision channel that employs a noisy-label learning strategy to refine instance pseudo-labels and mitigate the impact of noisy labels. Additionally, we propose a collaborative supervision framework that promotes communication and interaction between the attention mechanism in the bag-level supervision channel and the pseudo-label mechanism in the instance-level supervision channel, enabling cooperative optimization of supervision in both channels. Extensive experiments conducted on five datasets, including three public datasets and two in-house datasets, demonstrate the state-of-the-art performance of CIB-MIL. The code is available at https://github.com/TencentAILabHealthcare/CIB-MIL.