Clustering-Based Multi-instance Learning Network for Whole Slide Image Classification - IMT Mines Alès
Communication Dans Un Congrès Année : 2022

Clustering-Based Multi-instance Learning Network for Whole Slide Image Classification

Wei Wu
  • Fonction : Auteur
Zhonghang Zhu
  • Fonction : Auteur
Baptiste Magnier
Liansheng Wang
  • Fonction : Auteur

Résumé

Automated and accurate classification of Whole Slide Image (WSI) is of great significance for the early diagnosis and treatment of cancer, which can be realized by Multi-Instance Learning (MIL). However, the current MIL method easily suffers from over-fitting due to the weak supervision of slide-level labels. In addition, it is difficult to distinguish discriminative instances in a WSI bag in the absence of pixel-level annotations. To address these problems, we propose a novel Clustering-Based Multi-Instance Learning method (CBMIL) for WSI classification. The CBMIL constructs feature set from phenotypic clusters to augment data for training the aggregation network. Meanwhile, a contrastive learning task is incorporated into the CBMIL for multi-task learning, which helps to regularize the feature aggregation process. In addition, the centroid of each phenotypic cluster is updated by the model, and the weights of the WSI patches are calculated by their similarity to the phenotypic centroids to highlight the significant patches. Our method is evaluated on two public WSI datasets (CAMELYON16 and TCGA-Lung) for binary tumor and cancer sub-types classification and achieves better performance and great interpretability compared with the state-of-the-art methods. The code is available at: https://github.com/wwu98934/CBMIL.
Fichier principal
Vignette du fichier
Clustering_based_Multi_instance_Learning.pdf (4.52 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03790194 , version 1 (04-10-2022)

Identifiants

Citer

Wei Wu, Zhonghang Zhu, Baptiste Magnier, Liansheng Wang. Clustering-Based Multi-instance Learning Network for Whole Slide Image Classification. CMMCA 2022 - The 1st Workshop on Computational Mathematics Modeling in Cancer Analysis (part of MICCAI 2022), https://cmmca2022.casconf.cn/, Sep 2022, Singapour, Singapore. pp.100-109, ⟨10.1007/978-3-031-17266-3_10⟩. ⟨hal-03790194⟩
55 Consultations
242 Téléchargements

Altmetric

Partager

More