A filtering scheme for confocal laser endomicroscopy (CLE)-video sequences for self-supervised learning

TL;DR

This paper introduces a filtering scheme for CLE video sequences to enhance self-supervised learning efficiency and accuracy, demonstrating significant improvements in training speed and model performance on medical imaging datasets.

Contribution

It proposes a novel filtering method for CLE videos that reduces redundancy, improving SSL training convergence and efficiency, with validation on multiple medical datasets.

Findings

Filtered SSL models outperform non-filtered baselines in accuracy.

Training time is reduced by 67% using the proposed filtering scheme.

SSL proves effective for CLE pretraining on medical datasets.

Abstract

Confocal laser endomicroscopy (CLE) is a non-invasive, real-time imaging modality that can be used for in-situ, in-vivo imaging and the microstructural analysis of mucous structures. The diagnosis using CLE is, however, complicated by images being hard to interpret for non-experienced physicians. Utilizing machine learning as an augmentative tool would hence be beneficial, but is complicated by the shortage of histopathology-correlated CLE imaging sequences with respect to the plurality of patterns in this domain, leading to overfitting of machine learning models. To overcome this, self-supervised learning (SSL) can be employed on larger unlabeled datasets. CLE is a video-based modality with high inter-frame correlation, leading to a non-stratified data distribution for SSL training. In this work, we propose a filter functionality on CLE video sequences to reduce the dataset redundancy in SSL training and improve SSL training convergence and training efficiency. We use four state-of-the-art baseline networks and a SSL teacher-student network with a vision transformer small backbone for the evaluation. These networks were evaluated on downstream tasks for a sinonasal tumor dataset and a squamous cell carcinoma of the skin dataset. On both datasets, we found the highest test accuracy on the filtered SSL-pretrained model, with 67.48% and 73.52%, both considerably outperforming their non-SSL baselines. Our results show that SSL is an effective method for CLE pretraining. Further, we show that our proposed CLE video filter can be utilized to improve training efficiency in self-supervised scenarios, resulting in a reduction of 67% in training time.