Dataset Distillation for Histopathology Image Classification

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Cong, Cong, Xuan, Shiyu, Liu, Sidong, Pagnucco, Maurice, Zhang, Shiliang, Song, Yang
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910570202005504
author Cong, Cong
Xuan, Shiyu
Liu, Sidong
Pagnucco, Maurice
Zhang, Shiliang
Song, Yang
author_facet Cong, Cong
Xuan, Shiyu
Liu, Sidong
Pagnucco, Maurice
Zhang, Shiliang
Song, Yang
contents Deep neural networks (DNNs) have exhibited remarkable success in the field of histopathology image analysis. On the other hand, the contemporary trend of employing large models and extensive datasets has underscored the significance of dataset distillation, which involves compressing large-scale datasets into a condensed set of synthetic samples, offering distinct advantages in improving training efficiency and streamlining downstream applications. In this work, we introduce a novel dataset distillation algorithm tailored for histopathology image datasets (Histo-DD), which integrates stain normalisation and model augmentation into the distillation progress. Such integration can substantially enhance the compatibility with histopathology images that are often characterised by high colour heterogeneity. We conduct a comprehensive evaluation of the effectiveness of the proposed algorithm and the generated histopathology samples in both patch-level and slide-level classification tasks. The experimental results, carried out on three publicly available WSI datasets, including Camelyon16, TCGA-IDH, and UniToPath, demonstrate that the proposed Histo-DD can generate more informative synthetic patches than previous coreset selection and patch sampling methods. Moreover, the synthetic samples can preserve discriminative information, substantially reduce training efforts, and exhibit architecture-agnostic properties. These advantages indicate that synthetic samples can serve as an alternative to large-scale datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2408_09709
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dataset Distillation for Histopathology Image Classification
Cong, Cong
Xuan, Shiyu
Liu, Sidong
Pagnucco, Maurice
Zhang, Shiliang
Song, Yang
Computer Vision and Pattern Recognition
Deep neural networks (DNNs) have exhibited remarkable success in the field of histopathology image analysis. On the other hand, the contemporary trend of employing large models and extensive datasets has underscored the significance of dataset distillation, which involves compressing large-scale datasets into a condensed set of synthetic samples, offering distinct advantages in improving training efficiency and streamlining downstream applications. In this work, we introduce a novel dataset distillation algorithm tailored for histopathology image datasets (Histo-DD), which integrates stain normalisation and model augmentation into the distillation progress. Such integration can substantially enhance the compatibility with histopathology images that are often characterised by high colour heterogeneity. We conduct a comprehensive evaluation of the effectiveness of the proposed algorithm and the generated histopathology samples in both patch-level and slide-level classification tasks. The experimental results, carried out on three publicly available WSI datasets, including Camelyon16, TCGA-IDH, and UniToPath, demonstrate that the proposed Histo-DD can generate more informative synthetic patches than previous coreset selection and patch sampling methods. Moreover, the synthetic samples can preserve discriminative information, substantially reduce training efforts, and exhibit architecture-agnostic properties. These advantages indicate that synthetic samples can serve as an alternative to large-scale datasets.
title Dataset Distillation for Histopathology Image Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.09709