GPU-accelerated iterative method for FD-OCT image reconstruction with an image-level cross-domain regularizer

Abstract

The image reconstruction for Fourier-domain optical coherence tomography (FD-OCT) could be achieved by iterative methods, which offer a more accurate estimation than the traditional inverse discrete Fourier transform (IDFT) reconstruction. However, the existing iterative methods are mostly A-line-based and are developed on CPU, which causes slow reconstruction. Besides, A-line-based reconstruction makes the iterative methods incompatible with most existing image-level image processing techniques. In this paper, we proposed an iterative method that enables B-scan-based OCT image reconstruction, which has three major advantages: (1) Large-scale parallelism of the OCT dataset is achieved by using GPU acceleration. (2) A novel image-level cross-domain regularizer was developed, such that the image processing could be performed simultaneously during the image reconstruction; an enhanced image could be directly generated from the OCT interferogram. (3) The scalability of the proposed method was demonstrated for 3D OCT image reconstruction. Compared with the state-of-the-art (SOTA) iterative approaches, the proposed method achieves higher image quality with reduced computational time by orders of magnitude. To further show the image enhancement ability, a comparison was conducted between the proposed method and the conventional workflow, in which an IDFT reconstructed OCT image is later processed by a total variation-regularized denoising algorithm. The proposed method can achieve a better performance evaluated by metrics such as signal-to-noise ratio (SNR) and contrast-to-noise ratio (CNR), while the speed is improved by more than 30 times. Real-time image reconstruction at more than 20 B-scans per second was realized with a frame size of 4096 (axial) × 1000 (lateral), which showcases the great potential of the proposed method in real-world applications.