用数学变换提升无监督去雾效果,模型更轻更快。
Unpaired Image Dehazing via Kolmogorov-Arnold Transformation of Latent Features
- 引入柯尔莫哥洛夫-阿诺德变换建模特征关系
- 在多个数据集上超越现有无监督方法性能
- 适合需要轻量级、无需配对数据的去雾场景
本文提出一种基于柯尔莫哥洛夫-阿诺德变换的无监督图像去雾框架(UID-KAT)。图像去雾是一项具有挑战性且病态的视觉任务,需在特征空间中进行复杂变换与解释。受柯尔莫哥洛夫-阿诺德网络(KANs)启发,该方法利用其多项式基础高效逼近复杂函数,相比传统多层感知机(MLPs)所需层数更少。通过结合对抗训练与对比学习,模型能生成高保真图像,并聚焦关键特征、抑制无关信息。所提框架在无监督设置下训练,充分利用真实世界数据,克服成对雾图/清晰图标注稀缺的难题。实验表明,UID-KAT在多个数据集和场景中达到当前最优去雾性能,同时降低模型复杂度。代码已开源:https://github.com/tranleanh/uid-kat。
原文摘要 · Abstract (English)
This paper proposes an innovative framework for Unsupervised Image Dehazing via Kolmogorov-Arnold Transformation, termed UID-KAT. Image dehazing is recognized as a challenging and ill-posed vision task that requires complex transformations and interpretations in the feature space. Recent advancements have introduced Kolmogorov-Arnold Networks (KANs), inspired by the Kolmogorov-Arnold representation theorem, as promising alternatives to Multi-Layer Perceptrons (MLPs) since KANs can leverage their polynomial foundation to more efficiently approximate complex functions while requiring fewer layers than MLPs. Motivated by this potential, this paper explores the use of KANs combined with adversarial training and contrastive learning to model the intricate relationship between hazy and clear images. Adversarial training is employed due to its capacity in producing high-fidelity images, and contrastive learning promotes the model's emphasis on significant features while suppressing the influence of irrelevant information. The proposed UID-KAT framework is trained in an unsupervised setting to take advantage of the abundance of real-world data and address the challenge of preparing paired hazy/clean images. Experimental results show that UID-KAT achieves state-of-the-art dehazing performance across multiple datasets and scenarios, outperforming existing unpaired methods while reducing model complexity. The source code for this work is publicly available at https://github.com/tranleanh/uid-kat.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。