arXiv:2502.00240stat.MLcs.LG2025-02被引 5

用可证明收敛的非凸正则化方法提升图像重建效果

Learning Difference-of-Convex Regularizers for Inverse Problems: A Flexible Framework with Theoretical Guarantees

  • 将正则项设计为凸函数之差,兼顾性能与理论保证
  • 在稀疏和有限视角的CT重建中优于现有方法
  • 适合需要稳定性和可解释性的逆问题研究者

学习有效的正则化对解决广泛存在于科学与工程中的不适定逆问题至关重要。尽管基于深度神经网络参数化正则化的数据驱动方法表现出强大性能,但通常导致高度非凸的优化问题,缺乏理论保障。近期研究表明,将结构化非凸性(如弱凸性)引入神经网络正则化,可在性能与理论可处理性之间取得平衡。本文证明更广泛的非凸函数——凸函数之差(DC)类,能在保持强收敛性的同时提升实际表现。DC结构支持使用成熟的优化算法,如差分凸算法(DCA)和近端次梯度法(PSM),超越标准梯度下降。我们还提供了最优正则化器可表示为DC函数的理论条件。在计算机断层扫描(CT)重建任务上的大量实验表明,该方法在稀疏采样和有限视角设置下均表现优异,持续优于其他弱监督学习正则化方法。代码已公开于\url{https://github.com/YasminZhang/ADCR}。

原文摘要 · Abstract (English)

Learning effective regularization is crucial for solving ill-posed inverse problems, which arise in a wide range of scientific and engineering applications. While data-driven methods that parameterize regularizers using deep neural networks have demonstrated strong empirical performance, they often result in highly nonconvex formulations that lack theoretical guarantees. Recent work has shown that incorporating structured nonconvexity into neural network-based regularizers, such as weak convexity, can strike a balance between empirical performance and theoretical tractability. In this paper, we demonstrate that a broader class of nonconvex functions, difference-of-convex (DC) functions, can yield improved empirical performance while retaining strong convergence guarantees. The DC structure enables the use of well-established optimization algorithms, such as the Difference-of-Convex Algorithm (DCA) and a Proximal Subgradient Method (PSM), which extend beyond standard gradient descent. Furthermore, we provide theoretical insights into the conditions under which optimal regularizers can be expressed as DC functions. Extensive experiments on computed tomography (CT) reconstruction tasks show that our approach achieves strong performance across sparse and limited-view settings, consistently outperforming other weakly supervised learned regularizers. Our code is available at \url{https://github.com/YasminZhang/ADCR}.

逆问题正则化非凸优化CT重建

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