mlsanityData to 14 Oct 2026

Polyp Segmentation Generalisability of Pretrained Backbones

arXiv 2405.15524 ↗First posted 24 May 2024cs.CV cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

It has recently been demonstrated that pretraining backbones in a self-supervised manner generally provides better fine-tuned polyp segmentation performance, and that models with ViT-B backbones typically perform better than models with ResNet50 backbones. In this paper, we extend this recent work to consider generalisability. I.e., we assess the performance of models on a different dataset to that used for fine-tuning, accounting for variation in network architecture and pretraining pipeline (algorithm and dataset). This reveals how well models with different pretrained backbones generalise to data of a somewhat different distribution to the training data, which will likely arise in deployment due to different cameras and demographics of patients, amongst other factors. We observe that the previous findings, regarding pretraining pipelines for polyp segmentation, hold true when considering generalisability. However, our results imply that models with ResNet50 backbones typically generalise better, despite being outperformed by models with ViT-B backbones in evaluation on the test set from the same dataset used for fine-tuning.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 14 references that are in the map. First posted on arXiv 24 May 2024.

Builds on

11 of 14 references.

References · 11

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  4. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  5. 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,223
  6. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
  7. 2021Vision Transformers for Dense Predictioncited by 1,803
  8. 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
  9. 2019Kvasir-SEG: A Segmented Polyp Datasetcited by 589
  10. 2022FCN-Transformer Feature Fusion for Polyp Segmentationcited by 37
  11. 2024A Study on Self-Supervised Pretraining for Vision Problems in Gastrointestinal Endoscopycited by 2

Built on by

Not cited yet.