Is Self-Supervision Enough? Benchmarking Foundation Models Against End-to-End Training for Mitotic Figure Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ganz, Jonathan, Ammeling, Jonas, Rosbach, Emely, Lausser, Ludwig, Bertram, Christof A., Breininger, Katharina, Aubreville, Marc
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915065156861952
author Ganz, Jonathan
Ammeling, Jonas
Rosbach, Emely
Lausser, Ludwig
Bertram, Christof A.
Breininger, Katharina
Aubreville, Marc
author_facet Ganz, Jonathan
Ammeling, Jonas
Rosbach, Emely
Lausser, Ludwig
Bertram, Christof A.
Breininger, Katharina
Aubreville, Marc
contents Foundation models (FMs), i.e., models trained on a vast amount of typically unlabeled data, have become popular and available recently for the domain of histopathology. The key idea is to extract semantically rich vectors from any input patch, allowing for the use of simple subsequent classification networks potentially reducing the required amounts of labeled data, and increasing domain robustness. In this work, we investigate to which degree this also holds for mitotic figure classification. Utilizing two popular public mitotic figure datasets, we compared linear probing of five publicly available FMs against models trained on ImageNet and a simple ResNet50 end-to-end-trained baseline. We found that the end-to-end-trained baseline outperformed all FM-based classifiers, regardless of the amount of data provided. Additionally, we did not observe the FM-based classifiers to be more robust against domain shifts, rendering both of the above assumptions incorrect.
format Preprint
id arxiv_https___arxiv_org_abs_2412_06365
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Is Self-Supervision Enough? Benchmarking Foundation Models Against End-to-End Training for Mitotic Figure Classification
Ganz, Jonathan
Ammeling, Jonas
Rosbach, Emely
Lausser, Ludwig
Bertram, Christof A.
Breininger, Katharina
Aubreville, Marc
Computer Vision and Pattern Recognition
Foundation models (FMs), i.e., models trained on a vast amount of typically unlabeled data, have become popular and available recently for the domain of histopathology. The key idea is to extract semantically rich vectors from any input patch, allowing for the use of simple subsequent classification networks potentially reducing the required amounts of labeled data, and increasing domain robustness. In this work, we investigate to which degree this also holds for mitotic figure classification. Utilizing two popular public mitotic figure datasets, we compared linear probing of five publicly available FMs against models trained on ImageNet and a simple ResNet50 end-to-end-trained baseline. We found that the end-to-end-trained baseline outperformed all FM-based classifiers, regardless of the amount of data provided. Additionally, we did not observe the FM-based classifiers to be more robust against domain shifts, rendering both of the above assumptions incorrect.
title Is Self-Supervision Enough? Benchmarking Foundation Models Against End-to-End Training for Mitotic Figure Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.06365