Exploring the Effect of Dataset Diversity in Self-Supervised Learning for Surgical Computer Vision
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866912084860600320 |
|---|---|
| author | Jaspers, Tim J. M. de Jong, Ronald L. P. D. Khalil, Yasmina Al Zeelenberg, Tijn Kusters, Carolus H. J. Li, Yiping van Jaarsveld, Romy C. Bakker, Franciscus H. A. Ruurda, Jelle P. Brinkman, Willem M. De With, Peter H. N. van der Sommen, Fons |
| author_facet | Jaspers, Tim J. M. de Jong, Ronald L. P. D. Khalil, Yasmina Al Zeelenberg, Tijn Kusters, Carolus H. J. Li, Yiping van Jaarsveld, Romy C. Bakker, Franciscus H. A. Ruurda, Jelle P. Brinkman, Willem M. De With, Peter H. N. van der Sommen, Fons |
| contents | Over the past decade, computer vision applications in minimally invasive surgery have rapidly increased. Despite this growth, the impact of surgical computer vision remains limited compared to other medical fields like pathology and radiology, primarily due to the scarcity of representative annotated data. Whereas transfer learning from large annotated datasets such as ImageNet has been conventionally the norm to achieve high-performing models, recent advancements in self-supervised learning (SSL) have demonstrated superior performance. In medical image analysis, in-domain SSL pretraining has already been shown to outperform ImageNet-based initialization. Although unlabeled data in the field of surgical computer vision is abundant, the diversity within this data is limited. This study investigates the role of dataset diversity in SSL for surgical computer vision, comparing procedure-specific datasets against a more heterogeneous general surgical dataset across three different downstream surgical applications. The obtained results show that using solely procedure-specific data can lead to substantial improvements of 13.8%, 9.5%, and 36.8% compared to ImageNet pretraining. However, extending this data with more heterogeneous surgical data further increases performance by an additional 5.0%, 5.2%, and 2.5%, suggesting that increasing diversity within SSL data is beneficial for model performance. The code and pretrained model weights are made publicly available at https://github.com/TimJaspers0801/SurgeNet. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2407_17904 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Exploring the Effect of Dataset Diversity in Self-Supervised Learning for Surgical Computer Vision Jaspers, Tim J. M. de Jong, Ronald L. P. D. Khalil, Yasmina Al Zeelenberg, Tijn Kusters, Carolus H. J. Li, Yiping van Jaarsveld, Romy C. Bakker, Franciscus H. A. Ruurda, Jelle P. Brinkman, Willem M. De With, Peter H. N. van der Sommen, Fons Computer Vision and Pattern Recognition Over the past decade, computer vision applications in minimally invasive surgery have rapidly increased. Despite this growth, the impact of surgical computer vision remains limited compared to other medical fields like pathology and radiology, primarily due to the scarcity of representative annotated data. Whereas transfer learning from large annotated datasets such as ImageNet has been conventionally the norm to achieve high-performing models, recent advancements in self-supervised learning (SSL) have demonstrated superior performance. In medical image analysis, in-domain SSL pretraining has already been shown to outperform ImageNet-based initialization. Although unlabeled data in the field of surgical computer vision is abundant, the diversity within this data is limited. This study investigates the role of dataset diversity in SSL for surgical computer vision, comparing procedure-specific datasets against a more heterogeneous general surgical dataset across three different downstream surgical applications. The obtained results show that using solely procedure-specific data can lead to substantial improvements of 13.8%, 9.5%, and 36.8% compared to ImageNet pretraining. However, extending this data with more heterogeneous surgical data further increases performance by an additional 5.0%, 5.2%, and 2.5%, suggesting that increasing diversity within SSL data is beneficial for model performance. The code and pretrained model weights are made publicly available at https://github.com/TimJaspers0801/SurgeNet. |
| title | Exploring the Effect of Dataset Diversity in Self-Supervised Learning for Surgical Computer Vision |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2407.17904 |