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

UCMNet: 用于下屏摄像头图像修复的不确定性感知上下文记忆网络

UCMNet: Uncertainty-Aware Context Memory Network for Under-Display Camera Image Restoration

Hanyang University(汉阳大学);Agency for Defense Development (ADD)(国防发展局(ADD))

Daehyun Kim, Youngmin Kim, Yoon Ju Oh, Tae Hyun Kim

AI总结:

UCMNet通过不确定性感知的自适应处理,有效恢复下屏摄像头图像中因光衍射和散射导致的高频细节,采用轻量级架构在多个基准上实现SOTA性能,参数量减少30%。

摘要:

下屏摄像头(UDCs)通过将成像传感器置于显示屏下方实现全屏设计,但光衍射和散射导致的空间变化和复杂退化显著降低了高频细节。当前基于点扩散函数的物理建模技术和频域分离网络在重建低频结构和保持整体色彩一致性方面有效,但在处理复杂、空间变化的退化时仍面临恢复细节数的挑战。为此,我们提出了一种轻量级的不确定性感知上下文记忆网络(UCMNet),用于UDC图像修复。与先前方法不同,UCMNet通过不确定性感知的自适应处理,在退化变化的区域恢复高频细节。估计的不确定性图通过不确定性驱动损失学习,量化由衍射和散射引起的空间不确定性,并指导内存银行从上下文银行检索区域自适应的上下文。此过程使有效建模UDC成像固有的非均匀退化特性。利用此不确定性作为先验,UCMNet在多个基准上实现了SOTA性能,参数量比先前模型少30%。项目页面:https://kdhrick2222.github.io/projects/UCMNet/

英文摘要:

Under-display cameras (UDCs) allow for full-screen designs by positioning the imaging sensor underneath the display. Nonetheless, light diffraction and scattering through the various display layers result in spatially varying and complex degradations, which significantly reduce high-frequency details. Current PSF-based physical modeling techniques and frequency-separation networks are effective at reconstructing low-frequency structures and maintaining overall color consistency. However, they still face challenges in recovering fine details when dealing with complex, spatially varying degradation. To solve this problem, we propose a lightweight \textbf{U}ncertainty-aware \textbf{C}ontext-\textbf{M}emory \textbf{Network} (\textbf{UCMNet}), for UDC image restoration. Unlike previous methods that apply uniform restoration, UCMNet performs uncertainty-aware adaptive processing to restore high-frequency details in regions with varying degradations. The estimated uncertainty maps, learned through an uncertainty-driven loss, quantify spatial uncertainty induced by diffraction and scattering, and guide the Memory Bank to retrieve region-adaptive context from the Context Bank. This process enables effective modeling of the non-uniform degradation characteristics inherent to UDC imaging. Leveraging this uncertainty as a prior, UCMNet achieves state-of-the-art performance on multiple benchmarks with 30\% fewer parameters than previous models. Project page: \href{https://kdhrick2222.github.io/projects/UCMNet/}{https://kdhrick2222.github.io/projects/UCMNet}.

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

↑