SNA-SKAN: Unpaired learning for SDOCT speckle noise removal based on self noise assist and kolmogorov-arnold network.

Journal: Computerized medical imaging and graphics : the official journal of the Computerized Medical Imaging Society
Published Date:

Abstract

Optical Coherence Tomography (OCT) will inevitably be contaminated by speckle noise when imaging, resulting in a decrease in the visual quality of images and affecting clinical diagnosis. Existing unsupervised denoising methods often rely on complex model architectures or extensive data preprocessing. This paper proposes an unpaired Spectral-Domain OCT (SDOCT) denoising framework named SNA-SKAN. The Self Noise Assist (SNA) module leverages wavelet transform and singular value decomposition to extract noise components directly from noisy OCT images. These components are then fused into a new noise representation, which guides the neural network in effectively learning speckle noise patterns. Furthermore, to more effectively model speckle noise in OCT images, this paper exploits the Kolmogorov-Arnold Network (KAN) for its superior capacity to represent complex distributions, and proposes a KAN-based speckle noise generation network (SKAN). The SNA-SKAN framework is built upon the Generative Adversarial Network (GAN) architecture, employing a single generator and a single discriminator. Extensive experiments conducted on an unpaired public dataset for training and two public datasets for evaluation demonstrate that the proposed method outperforms existing unsupervised methods and state-of-the-art unpaired methods, in terms of denoising capability and detail preservation. SNA-SKAN can achieve efficient OCT denoising while preserving edges and details, demonstrating strong potential to meet clinical needs. The code is publicly available at: https://github.com/zhencunjiang/SNA-SKAN.

Authors

  • Zhencun Jiang
    School of Electrical and Electronic Engineering, Shanghai Institute of Technology, 100 Haiquan Road, Shanghai, China.
  • Kangrui Ren
    Department of Control Science and Engineering, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, China.
  • Zixiong Hao
    Department of Control Science and Engineering, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, China; School of Computation, Information and Technology, Technical University of Munich, Munich 80333, Germany.
  • Zhongjie Wang
    Shanghai Research Institute for Intelligent Autonomous Systems, Tongji University, Shanghai 200092, China; Department of Control Science and Engineering, College of Electronics and Information Engineering, Tongji University, Shanghai 201804, China. Electronic address: wang_zhongjie@tongji.edu.cn.