Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. ScoreNet: Learning Non-Uniform Attention and Augmentation for Transformer-Based Histopathological Image Classification
 
conference paper

ScoreNet: Learning Non-Uniform Attention and Augmentation for Transformer-Based Histopathological Image Classification

Stegmuller, Thomas  
•
Bozorgtabar, Behzad  
•
Spahr, Antoine  
Show more
January 1, 2023
2023 Ieee/Cvf Winter Conference On Applications Of Computer Vision (Wacv)
23rd IEEE/CVF Winter Conference on Applications of Computer Vision (WACV 2023)

Progress in digital pathology is hindered by high-resolution images and the prohibitive cost of exhaustive localized annotations. The commonly used paradigm to categorize pathology images is patch-based processing, which often incorporates multiple instance learning (MIL) to aggregate local patch-level representations yielding imagelevel prediction. Nonetheless, diagnostically relevant regions may only take a small fraction of the whole tissue, and current MIL-based approaches often process images uniformly, discarding the inter-patches interactions. To alleviate these issues, we propose ScoreNet, a new efficient transformer that exploits a differentiable recommendation stage to extract discriminative image regions and dedicate computational resources accordingly. The proposed transformer leverages the local and global attention of a few dynamically recommended high-resolution regions at an efficient computational cost. We further introduce a novel mixing data-augmentation, namely ScoreMix, by leveraging the image's semantic distribution to guide the data mixing and produce coherent sample-label pairs. ScoreMix is embarrassingly simple and mitigates the pitfalls of previous augmentations, which assume a uniform semantic distribution and risk mislabeling the samples. Thorough experiments and ablation studies on three breast cancer histology datasets of Haematoxylin & Eosin (H&E) have validated the superiority of our approach over prior arts, including transformer-based models on tumour regions-of-interest (TRoIs) classification. ScoreNet equipped with proposed ScoreMix augmentation demonstrates better generalization capabilities and achieves new state-of-the-art (SOTA) results with only 50% of the data compared to other mixing augmentation variants. Finally, ScoreNet yields high efficacy and outperforms SOTA efficient transformers, namely TransPath [37] and SwinTransformer [20], with throughput around 3x and 4x higher than the aforementioned architectures, respectively. Our code is publicly available(1).

  • Details
  • Metrics
Type
conference paper
DOI
10.1109/WACV56688.2023.00611
Web of Science ID

WOS:000971500206029

Author(s)
Stegmuller, Thomas  
Bozorgtabar, Behzad  
Spahr, Antoine  
Thiran, Jean-Philippe  
Date Issued

2023-01-01

Publisher

IEEE COMPUTER SOC

Publisher place

Los Alamitos

Published in
2023 Ieee/Cvf Winter Conference On Applications Of Computer Vision (Wacv)
ISBN of the book

978-1-6654-9346-8

Series title/Series vol.

IEEE Winter Conference on Applications of Computer Vision

Start page

6159

End page

6168

Subjects

Computer Science, Artificial Intelligence

•

Engineering, Electrical & Electronic

•

Imaging Science & Photographic Technology

•

Computer Science

•

Engineering

•

Imaging Science & Photographic Technology

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LTS5  
Event nameEvent placeEvent date
23rd IEEE/CVF Winter Conference on Applications of Computer Vision (WACV 2023)

Waikoloa, HI

Jan 03-07, 2023

Available on Infoscience
July 31, 2023
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/199493
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés