Integration of Self-Supervised BYOL in Semi-Supervised Medical Image Recognition

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Feng, Hao, Jia, Yuanzhe, Xu, Ruijia, Prasad, Mukesh, Anaissi, Ali, Braytee, Ali
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909171348144128
author Feng, Hao
Jia, Yuanzhe
Xu, Ruijia
Prasad, Mukesh
Anaissi, Ali
Braytee, Ali
author_facet Feng, Hao
Jia, Yuanzhe
Xu, Ruijia
Prasad, Mukesh
Anaissi, Ali
Braytee, Ali
contents Image recognition techniques heavily rely on abundant labeled data, particularly in medical contexts. Addressing the challenges associated with obtaining labeled data has led to the prominence of self-supervised learning and semi-supervised learning, especially in scenarios with limited annotated data. In this paper, we proposed an innovative approach by integrating self-supervised learning into semi-supervised models to enhance medical image recognition. Our methodology commences with pre-training on unlabeled data utilizing the BYOL method. Subsequently, we merge pseudo-labeled and labeled datasets to construct a neural network classifier, refining it through iterative fine-tuning. Experimental results on three different datasets demonstrate that our approach optimally leverages unlabeled data, outperforming existing methods in terms of accuracy for medical image recognition.
format Preprint
id arxiv_https___arxiv_org_abs_2404_10405
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Integration of Self-Supervised BYOL in Semi-Supervised Medical Image Recognition
Feng, Hao
Jia, Yuanzhe
Xu, Ruijia
Prasad, Mukesh
Anaissi, Ali
Braytee, Ali
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Image recognition techniques heavily rely on abundant labeled data, particularly in medical contexts. Addressing the challenges associated with obtaining labeled data has led to the prominence of self-supervised learning and semi-supervised learning, especially in scenarios with limited annotated data. In this paper, we proposed an innovative approach by integrating self-supervised learning into semi-supervised models to enhance medical image recognition. Our methodology commences with pre-training on unlabeled data utilizing the BYOL method. Subsequently, we merge pseudo-labeled and labeled datasets to construct a neural network classifier, refining it through iterative fine-tuning. Experimental results on three different datasets demonstrate that our approach optimally leverages unlabeled data, outperforming existing methods in terms of accuracy for medical image recognition.
title Integration of Self-Supervised BYOL in Semi-Supervised Medical Image Recognition
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2404.10405