arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.02452eess.IVcs.CVcs.RO

LLCaps:基于弯曲小波注意力与反向扩散的低光照胶囊内镜图像增强学习方法

LLCaps: Learning to Illuminate Low-Light Capsule Endoscopy with Curved Wavelet Attention and Reverse Diffusion

The University of Sydney(悉尼大学);The Chinese University of Hong Kong (CUHK)(香港中文大学);Northeastern University(东北大学);University College London(伦敦大学学院);Shun Hing Institute of Advanced Engineering, CUHK(香港中文大学信兴高等工程研究所)

Long Bai, Tong Chen, Yanan Wu, An Wang, Mobarakol Islam, Hongliang Ren

AI总结:

针对无线胶囊内镜低光照成像问题,提出结合弯曲小波注意力与反向扩散的多尺度增强框架,在定量定性指标及疾病分割任务上均优于现有SOTA方法,具备临床应用潜力。

摘要:

无线胶囊内镜(WCE)是一种无痛、无创的胃肠道(GI)疾病诊断工具。但受胃肠道解剖结构限制与硬件制造工艺约束,WCE的视觉信号可能存在光照不足问题,导致筛查与检查流程复杂。医学领域中基于深度学习的低光照图像增强(LLIE)逐渐受到研究者关注。鉴于去噪扩散概率模型(DDPM)在计算机视觉领域的蓬勃发展,我们提出一种基于多尺度卷积神经网络(CNN)与反向扩散过程的WCE低光照图像增强框架。多尺度设计使模型能够保留高分辨率表征并获取低分辨率下的上下文信息,同时我们提出弯曲小波注意力(CWA)模块用于高频与局部特征学习。此外,我们结合反向扩散过程进一步优化浅层输出,生成最具真实感的图像。我们将所提方法与10种当前最优(SOTA)的低光照图像增强方法进行对比,在定量与定性指标上均显著优于现有方法。该方法在胃肠道疾病分割任务上的优异表现进一步证明了其临床应用潜力。我们的代码已公开。

英文摘要:

Wireless capsule endoscopy (WCE) is a painless and non-invasive diagnostic tool for gastrointestinal (GI) diseases. However, due to GI anatomical constraints and hardware manufacturing limitations, WCE vision signals may suffer from insufficient illumination, leading to a complicated screening and examination procedure. Deep learning-based low-light image enhancement (LLIE) in the medical field gradually attracts researchers. Given the exuberant development of the denoising diffusion probabilistic model (DDPM) in computer vision, we introduce a WCE LLIE framework based on the multi-scale convolutional neural network (CNN) and reverse diffusion process. The multi-scale design allows models to preserve high-resolution representation and context information from low-resolution, while the curved wavelet attention (CWA) block is proposed for high-frequency and local feature learning. Furthermore, we combine the reverse diffusion procedure to further optimize the shallow output and generate the most realistic image. The proposed method is compared with ten state-of-the-art (SOTA) LLIE methods and significantly outperforms quantitatively and qualitatively. The superior performance on GI disease segmentation further demonstrates the clinical potential of our proposed model. Our code is publicly accessible.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑