Yao Bin, Jin Lujia, Hu Jiakui, Liu Yuzhao, Yan Yuepeng, Li Qing, Lu Yanye
Institute of Microelectronics of the Chinese Academy of Sciences, Beijing 100029, China.
University of Chinese Academy of Sciences, Beijing 101408, China.
Biomed Opt Express. 2024 Apr 11;15(5):2958-2976. doi: 10.1364/BOE.521453. eCollection 2024 May 1.
Optical coherence tomography (OCT), owing to its non-invasive nature, has demonstrated tremendous potential in clinical practice and has become a prevalent diagnostic method. Nevertheless, the inherent speckle noise and low sampling rate in OCT imaging often limit the quality of OCT images. In this paper, we propose a lightweight Transformer to efficiently reconstruct high-quality images from noisy and low-resolution OCT images acquired by short scans. Our method, PSCAT, parallelly employs spatial window self-attention and channel attention in the Transformer block to aggregate features from both spatial and channel dimensions. It explores the potential of the Transformer in denoising and super-resolution for OCT, reducing computational costs and enhancing the speed of image processing. To effectively assist in restoring high-frequency details, we introduce a hybrid loss function in both spatial and frequency domains. Extensive experiments demonstrate that our PSCAT has fewer network parameters and lower computational costs compared to state-of-the-art methods while delivering a competitive performance both qualitatively and quantitatively.
光学相干断层扫描(OCT)由于其非侵入性,在临床实践中已展现出巨大潜力,并成为一种普遍的诊断方法。然而,OCT成像中固有的斑点噪声和低采样率常常限制了OCT图像的质量。在本文中,我们提出了一种轻量级Transformer,以有效地从短扫描获取的噪声和低分辨率OCT图像中重建高质量图像。我们的方法PSCAT在Transformer模块中并行采用空间窗口自注意力和通道注意力,以聚合来自空间和通道维度的特征。它探索了Transformer在OCT去噪和超分辨率方面的潜力,降低了计算成本并提高了图像处理速度。为了有效地辅助恢复高频细节,我们在空间和频域中引入了混合损失函数。大量实验表明,与现有方法相比,我们的PSCAT具有更少的网络参数和更低的计算成本,同时在定性和定量方面都具有竞争力的性能。