arXiv:2410.04123eess.IVcs.CV2024-10

用注意力UNet直接重建波长域OCT图像,省去复杂校准步骤。

WAVE-UNET: Wavelength based Image Reconstruction method using attention UNET for OCT images

  • 用改进的带注意力门控的UNet,直接从波长域干涉条纹恢复图像。
  • 相比传统方法,重建图像质量更高,处理时间减少超过40%。
  • 适合需要快速高质量OCT成像的医学影像场景。

本文提出WAVE-UNET框架,利用深度学习直接从波长(λ)空间干涉条纹重建高质量扫频源光学相干断层成像(SS-OCT)图像。传统方法需将λ空间线性条纹通过k空间线性化与插值转换为均匀网格,再进行逆离散傅里叶变换(IDFT),但此过程易导致信息损失并增加系统复杂度。同时,低相干干涉系统固有的散斑噪声影响图像质量。WAVE-UNET跳过线性化步骤,采用含注意力门控和残差连接的改进UNet结构,以原始λ空间条纹为输入,显著提升重建图像的真实感与清晰度。实验表明,该方法在保持高图像质量的同时,处理时间复杂度降低超40%,优于传统OCT系统,适用于实时、高精度医学成像需求。

原文摘要 · Abstract (English)

In this work, we propose to leverage a deep-learning (DL) based reconstruction framework for high quality Swept-Source Optical Coherence Tomography (SS-OCT) images, by incorporating wavelength (λ) space interferometric fringes. Generally, the SS-OCT captured fringe is linear in wavelength space and if Inverse Discrete Fourier Transform (IDFT) is applied to extract depth-resolved spectral information, the resultant images are blurred due to the broadened Point Spread Function (PSF). Thus, the recorded wavelength space fringe is to be scaled to uniform grid in wavenumber (k) space using k-linearization and calibration involving interpolations which may result in loss of information along with increased system complexity. Another challenge in OCT is the speckle noise, inherent in the low coherence interferometry-based systems. Hence, we propose a systematic design methodology WAVE-UNET to reconstruct the high-quality OCT images directly from the λ-space to reduce the complexity. The novel design paradigm surpasses the linearization procedures and uses DL to enhance the realism and quality of raw λ-space scans. This framework uses modified UNET having attention gating and residual connections, with IDFT processed λ-space fringes as the input. The method consistently outperforms the traditional OCT system by generating good-quality B-scans with highly reduced time-complexity.

OCT图像重建深度学习注意力机制波长域

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。