arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.10550eess.IVcs.CV

LighTDiff:基于T-Diffusion的手术内窥镜图像低光照增强

LighTDiff: Surgical Endoscopic Image Low-Light Enhancement with T-Diffusion

The Chinese University of Hong Kong (CUHK)(香港中文大学);Shenzhen Research Institute, CUHK(香港中文大学深圳研究院);The University of Sydney(悉尼大学);Qilu Hospital of Shandong University(山东大学齐鲁医院)

Tong Chen, Qingcheng Lyu, Long Bai, Erjian Guo, Huxin Gao, Xiaoxiao Yang, Hongliang Ren, Luping Zhou

AI总结:

针对手术内窥镜低光照增强中传统DDPM计算效率低的问题,提出轻量级T形扩散模型LighTDiff,结合剪枝、TLU模块与CB模块,在提升训练稳定性与去噪效果的同时实现优于现有LLIE方法的性能与效率。

摘要:

内窥镜在外科手术中的应用进展面临着光照不足等挑战。深度学习,尤其是去噪扩散概率模型(Denoising Diffusion Probabilistic Model, DDPM),在医学领域的低光照图像增强方面具有应用前景。然而,DDPM计算量大、速度慢,限制了其在实际医疗场景中的应用。为弥补这一不足,我们提出了一种轻量级DDPM,命名为LighTDiff。该模型采用T形架构,通过低分辨率图像捕获全局结构信息,并在后续去噪步骤中逐步恢复细节。我们进一步对模型进行剪枝,在保留性能的同时大幅缩减模型尺寸。由于舍弃部分下采样操作以节省参数会导致训练过程不稳定、收敛效率低,我们引入了即插即用模块时序光照单元(Temporal Light Unit, TLU),以实现更稳定的训练和更优的性能。TLU将时间步与去噪后的图像特征关联,建立去噪步骤的时序依赖关系,提升去噪效果。此外,我们注意到使用扩散模型恢复图像时可能出现光谱偏移问题,为此进一步引入色度平衡器(Chroma Balancer, CB)来缓解该问题。我们的LighTDiff在计算效率上表现优异,性能优于多种主流的低光照图像增强(LLIE)方法。

英文摘要:

Advances in endoscopy use in surgeries face challenges like inadequate lighting. Deep learning, notably the Denoising Diffusion Probabilistic Model (DDPM), holds promise for low-light image enhancement in the medical field. However, DDPMs are computationally demanding and slow, limiting their practical medical applications. To bridge this gap, we propose a lightweight DDPM, dubbed LighTDiff. It adopts a T-shape model architecture to capture global structural information using low-resolution images and gradually recover the details in subsequent denoising steps. We further prone the model to significantly reduce the model size while retaining performance. While discarding certain downsampling operations to save parameters leads to instability and low efficiency in convergence during the training, we introduce a Temporal Light Unit (TLU), a plug-and-play module, for more stable training and better performance. TLU associates time steps with denoised image features, establishing temporal dependencies of the denoising steps and improving denoising outcomes. Moreover, while recovering images using the diffusion model, potential spectral shifts were noted. We further introduce a Chroma Balancer (CB) to mitigate this issue. Our LighTDiff outperforms many competitive LLIE methods with exceptional computational efficiency.

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

↑