Collaborative masking based speckle disentanglement for self-supervised optical coherence tomography image despeckling

Abstract

Optical coherence tomography (OCT) has proven to be an effective and safe diagnostic tool in clinical settings due to its unique advantages. Nonetheless, the OCT images are unavoidably corrupted with speckle noise. Despeckling has become a vital preliminary step in OCT based image processing and analysis tasks. The supervised deep learning (DL) based denoising algorithms have recently attracted much attention due to their promising performance, but the lack of the noise-free OCT images renders it difficult to provide the effective supervision for DL model training. This paper has presented a self-supervised OCT image despeckling approach which does not depend on the reference clean image and combines the disentangled representation strategy with speckle noise estimation for the effective denoising. The proposed algorithm firstly disentangles the corrupted image into the speckle noise and clean image. Then, the multiple generated clean images are multiplied by the disentangled speckle noise to synthesize the speckle-corrupted image for self-supervised learning. To make the generated clean image and extracted speckle noise more statistically realistic, a collaborative masking strategy and a noise calibration module are applied to explore the spatial correlation of the clean image and refine speckle noise. The proposed method has been validated on two public datasets. Experimental results demonstrate that our method has superior speckle reduction performance to some traditional and DL-based despeckling approaches in terms of visual evaluation and quantitative metrics. Particularly, our method provides contrast-to-noise ratio (CNR) and signal-to-noise ratio (SNR) improvements of at least 0.35 dB and 14.37 dB over the compared methods, respectively. Additionally, the retinal image classification results provide further evidence for the exceptional despeckling performance of our approach against other compared methods. It is obvious that the developed method has significant advantages in keeping the fidelity of retinal structure and enhancing the speckle suppression level. Meanwhile, the presented method holds certain reference value for improving diagnostic efficiency in clinical.