MLCN: Metric Learning Constrained Network for Whole Slide Image Classification with Bilinear Gated Attention Mechanism

Baorong Shi, Xinyu Liu, Fa Zhang*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Citations (Scopus)

Abstract

Whole Slide Image (WSI) classification is an important part of pathological diagnosis. Although previous approaches (such as DSMIL and CLAM) have achieved good results, the classification performance is still unsatisfactory because the learned features of WSI lack discrimination and the correlation among sub-characteristics of tumor images are ignored. In this paper, we proposed a Metric Learning Constraint Network (referred to as MLCN). Particularly, MLCN benefits from two aspects: 1) It enhances the discriminative power of features by enlarging inter-class distance and narrowing intra-class distance in both slide-level and patch-level. 2) It learns a more powerful feature aggregator by proposing the bilinear gated attention mechanism to capture relations among sub-characteristics of tumor issues. Experiments on CAMELYON16 and TCGA Kidney datasets validate the effectiveness of our approach, and we achieved state-of-the-art performance compared to other popular methods. The codes will be available soon.

Original languageEnglish
Title of host publicationComputational Mathematics Modeling in Cancer Analysis - 1st International Workshop, CMMCA 2022, Held in Conjunction with MICCAI 2022, Proceedings
EditorsWenjian Qin, Nazar Zaki, Fa Zhang, Jia Wu, Fan Yang
PublisherSpringer Science and Business Media Deutschland GmbH
Pages35-46
Number of pages12
ISBN (Print)9783031172656
DOIs
Publication statusPublished - 2022
Externally publishedYes
Event1st International Workshop on Computational Mathematics Modeling in Cancer Analysis, CMMCA 2022, held in conjunction with the 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022 - Virtual, Online
Duration: 18 Sept 202218 Sept 2022

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13574 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference1st International Workshop on Computational Mathematics Modeling in Cancer Analysis, CMMCA 2022, held in conjunction with the 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022
CityVirtual, Online
Period18/09/2218/09/22

Keywords

  • Attention mechanism
  • Deep learning
  • Metric learning
  • Pathological image
  • Whole Slide Image classification

Fingerprint

Dive into the research topics of 'MLCN: Metric Learning Constrained Network for Whole Slide Image Classification with Bilinear Gated Attention Mechanism'. Together they form a unique fingerprint.

Cite this