用单张图像训练的KAN网络做图像修复,效果优于传统方法。
You KAN Do It in a Single Shot: Plug-and-Play Methods with Single-Instance Priors
- 用KAN网络作为单样本先验的去噪器,替代传统数据依赖型模型
- 在超分辨率和联合优化任务中,仅需一次样本即达到更高精度
- 理论保证收敛性,适合数据稀缺场景下的图像恢复应用
Plug-and-Play(PnP)方法已成为解决逆问题的核心范式,其中去噪器作为正则化先验引导优化过程以获得干净解。本文提出KAN-PnP,将基于柯尔莫哥洛夫-阿诺德表示定理的Kolmogorov-Arnold网络(KANs)引入PnP框架,专为单实例先验问题设计——仅需一张含噪观测即可完成建模,无需大规模数据集。我们证明,KAN去噪器具有Lipschitz连续性,在PnP-ADMM等优化算法中确保稳定性与收敛性。同时提供理论保障:在数据保真项凸性、去噪器Lipschitz连续性和正则项有界性的条件下,可实现收敛。实验表明,在超分辨率与联合优化任务中,KAN-PnP显著优于现有方法,展现出优异的单次学习性能与快速收敛能力,以最少迭代次数达成高精度结果。
原文摘要 · Abstract (English)
The use of Plug-and-Play (PnP) methods has become a central approach for solving inverse problems, with denoisers serving as regularising priors that guide optimisation towards a clean solution. In this work, we introduce KAN-PnP, an optimisation framework that incorporates Kolmogorov-Arnold Networks (KANs) as denoisers within the Plug-and-Play (PnP) paradigm. KAN-PnP is specifically designed to solve inverse problems with single-instance priors, where only a single noisy observation is available, eliminating the need for large datasets typically required by traditional denoising methods. We show that KANs, based on the Kolmogorov-Arnold representation theorem, serve effectively as priors in such settings, providing a robust approach to denoising. We prove that the KAN denoiser is Lipschitz continuous, ensuring stability and convergence in optimisation algorithms like PnP-ADMM, even in the context of single-shot learning. Additionally, we provide theoretical guarantees for KAN-PnP, demonstrating its convergence under key conditions: the convexity of the data fidelity term, Lipschitz continuity of the denoiser, and boundedness of the regularisation functional. These conditions are crucial for stable and reliable optimisation. Our experimental results show, on super-resolution and joint optimisation, that KAN-PnP outperforms exiting methods, delivering superior performance in single-shot learning with minimal data. The method exhibits strong convergence properties, achieving high accuracy with fewer iterations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。