用镜面下降法让深度均衡模型更好解决泊松成像逆问题。
Deep Equilibrium models for Poisson Imaging Inverse problems via Mirror Descent
- 基于非欧几何的镜面下降构造新DEQ,适配泊松数据项结构。
- 理论保证收敛性,实验显示性能优于传统方法且无需调参。
- 适合图像重建研究者,尤其关注高效无超参方法的场景。
深度均衡模型(DEQ)是具有固定点的隐式神经网络,近年来因其在学习图像正则化泛函方面的潜力而受到关注,尤其适用于高斯保真度情形,此时前向算子的假设可保证标准(近端)梯度下降算子的压缩性。本文将DEQ拓展至泊松逆问题,其中数据保真项更适于用KL散度建模。为此,我们提出一种基于镜面下降的新DEQ框架,其定义在定制的非欧几里得几何上,能自然适应数据项结构。该框架支持在合理训练范式下学习神经正则器。我们基于子解析函数在非闭域上的Kurdyka–Łojasiewicz框架,推导出充分条件并建立精细收敛结果,确保所学重建方案的收敛性,并提出计算策略以实现高效训练与无参数推理。数值实验表明,本方法性能超越传统模型基方法,与Bregman Plug-and-Play方法相当,同时避免了其典型缺点,如超参数调优耗时。代码公开于 https://github.com/christiandaniele/DEQ-MD。
原文摘要 · Abstract (English)
Deep Equilibrium Models (DEQs) are implicit neural networks with fixed points, which have recently gained attention for learning image regularization functionals, particularly in settings involving Gaussian fidelities, where assumptions on the forward operator ensure contractiveness of standard (proximal) Gradient Descent operators. In this work, we extend the application of DEQs to Poisson inverse problems, where the data fidelity term is more appropriately modeled by the Kullback--Leibler divergence. To this end, we introduce a novel DEQ formulation based on Mirror Descent defined in terms of a tailored non-Euclidean geometry that naturally adapts with the structure of the data term. This enables the learning of neural regularizers within a principled training framework. We derive sufficient conditions and establish refined convergence results based on the Kurdyka--Lojasiewicz framework for subanalytic functions with non-closed domains to guarantee the convergence of the learned reconstruction scheme and propose computational strategies that enable both efficient training and parameter-free inference. Numerical experiments show that our method outperforms traditional model-based approaches and it is comparable to the performance of Bregman Plug-and-Play methods, while mitigating their typical drawbacks, such as time-consuming tuning of hyper-parameters. The code is publicly available at https://github.com/christiandaniele/DEQ-MD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。