Exploiting the potential of unlabeled endoscopic video data with self-supervised learning

TL;DR

This paper demonstrates that self-supervised learning using unlabeled endoscopic videos, specifically through a GAN-based re-colorization auxiliary task, can significantly reduce manual annotation needs while maintaining high segmentation performance.

Contribution

It introduces a novel self-supervised pre-training method for CNNs in medical imaging using GAN-based re-colorization of unlabeled videos, reducing annotation effort.

Findings

Reduces labeled data requirement by up to 75%.

Outperforms pre-training on external datasets.

Effective for instrument segmentation in endoscopy.

Abstract

Surgical data science is a new research field that aims to observe all aspects of the patient treatment process in order to provide the right assistance at the right time. Due to the breakthrough successes of deep learning-based solutions for automatic image annotation, the availability of reference annotations for algorithm training is becoming a major bottleneck in the field. The purpose of this paper was to investigate the concept of self-supervised learning to address this issue. Our approach is guided by the hypothesis that unlabeled video data can be used to learn a representation of the target domain that boosts the performance of state-of-the-art machine learning algorithms when used for pre-training. Core of the method is an auxiliary task based on raw endoscopic video data of the target domain that is used to initialize the convolutional neural network (CNN) for the target task. In this paper, we propose the re-colorization of medical images with a generative adversarial network (GAN)-based architecture as auxiliary task. A variant of the method involves a second pre-training step based on labeled data for the target task from a related domain. We validate both variants using medical instrument segmentation as target task. The proposed approach can be used to radically reduce the manual annotation effort involved in training CNNs. Compared to the baseline approach of generating annotated data from scratch, our method decreases exploratively the number of labeled images by up to 75% without sacrificing performance. Our method also outperforms alternative methods for CNN pre-training, such as pre-training on publicly available non-medical or medical data using the target task (in this instance: segmentation). As it makes efficient use of available (non-)public and (un-)labeled data, the approach has the potential to become a valuable tool for CNN (pre-)training.