FAST: A Dual-tier Few-Shot Learning Paradigm for Whole Slide Image Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fu, Kexue, Luo, Xiaoyuan, Qu, Linhao, Wang, Shuo, Xiong, Ying, Maglogiannis, Ilias, Gao, Longxiang, Wang, Manning
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912050038439936
author Fu, Kexue
Luo, Xiaoyuan
Qu, Linhao
Wang, Shuo
Xiong, Ying
Maglogiannis, Ilias
Gao, Longxiang
Wang, Manning
author_facet Fu, Kexue
Luo, Xiaoyuan
Qu, Linhao
Wang, Shuo
Xiong, Ying
Maglogiannis, Ilias
Gao, Longxiang
Wang, Manning
contents The expensive fine-grained annotation and data scarcity have become the primary obstacles for the widespread adoption of deep learning-based Whole Slide Images (WSI) classification algorithms in clinical practice. Unlike few-shot learning methods in natural images that can leverage the labels of each image, existing few-shot WSI classification methods only utilize a small number of fine-grained labels or weakly supervised slide labels for training in order to avoid expensive fine-grained annotation. They lack sufficient mining of available WSIs, severely limiting WSI classification performance. To address the above issues, we propose a novel and efficient dual-tier few-shot learning paradigm for WSI classification, named FAST. FAST consists of a dual-level annotation strategy and a dual-branch classification framework. Firstly, to avoid expensive fine-grained annotation, we collect a very small number of WSIs at the slide level, and annotate an extremely small number of patches. Then, to fully mining the available WSIs, we use all the patches and available patch labels to build a cache branch, which utilizes the labeled patches to learn the labels of unlabeled patches and through knowledge retrieval for patch classification. In addition to the cache branch, we also construct a prior branch that includes learnable prompt vectors, using the text encoder of visual-language models for patch classification. Finally, we integrate the results from both branches to achieve WSI classification. Extensive experiments on binary and multi-class datasets demonstrate that our proposed method significantly surpasses existing few-shot classification methods and approaches the accuracy of fully supervised methods with only 0.22$\%$ annotation costs. All codes and models will be publicly available on https://github.com/fukexue/FAST.
format Preprint
id arxiv_https___arxiv_org_abs_2409_19720
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FAST: A Dual-tier Few-Shot Learning Paradigm for Whole Slide Image Classification
Fu, Kexue
Luo, Xiaoyuan
Qu, Linhao
Wang, Shuo
Xiong, Ying
Maglogiannis, Ilias
Gao, Longxiang
Wang, Manning
Computer Vision and Pattern Recognition
The expensive fine-grained annotation and data scarcity have become the primary obstacles for the widespread adoption of deep learning-based Whole Slide Images (WSI) classification algorithms in clinical practice. Unlike few-shot learning methods in natural images that can leverage the labels of each image, existing few-shot WSI classification methods only utilize a small number of fine-grained labels or weakly supervised slide labels for training in order to avoid expensive fine-grained annotation. They lack sufficient mining of available WSIs, severely limiting WSI classification performance. To address the above issues, we propose a novel and efficient dual-tier few-shot learning paradigm for WSI classification, named FAST. FAST consists of a dual-level annotation strategy and a dual-branch classification framework. Firstly, to avoid expensive fine-grained annotation, we collect a very small number of WSIs at the slide level, and annotate an extremely small number of patches. Then, to fully mining the available WSIs, we use all the patches and available patch labels to build a cache branch, which utilizes the labeled patches to learn the labels of unlabeled patches and through knowledge retrieval for patch classification. In addition to the cache branch, we also construct a prior branch that includes learnable prompt vectors, using the text encoder of visual-language models for patch classification. Finally, we integrate the results from both branches to achieve WSI classification. Extensive experiments on binary and multi-class datasets demonstrate that our proposed method significantly surpasses existing few-shot classification methods and approaches the accuracy of fully supervised methods with only 0.22$\%$ annotation costs. All codes and models will be publicly available on https://github.com/fukexue/FAST.
title FAST: A Dual-tier Few-Shot Learning Paradigm for Whole Slide Image Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.19720