中文

Wav-KAN:小波Kolmogorov-Arnold网络

机器学习 2024-05-28 v2 人工智能 信号处理 机器学习

摘要

在本文中,我们介绍了Wav-KAN,一种创新的神经网络架构,利用小波Kolmogorov-Arnold网络(Wav-KAN)框架来增强可解释性和性能。传统的多层感知器(MLP)甚至最近的进展如Spl-KAN在可解释性、训练速度、鲁棒性、计算效率和性能方面面临挑战。Wav-KAN通过将小波函数纳入Kolmogorov-Arnold网络结构来解决这些限制,使网络能够高效地捕获输入数据的高频和低频分量。基于小波的近似采用正交或半正交基,并在准确表示底层数据结构和避免过拟合噪声之间保持平衡。虽然连续小波变换(CWT)具有很大潜力,我们还采用了离散小波变换(DWT)进行多分辨率分析,这避免了在寻找细节时重新计算先前步骤的需要。类似于水适应其容器的形状,Wav-KAN适应数据结构,从而相比Spl-KAN和MLP提高了准确性、更快的训练速度和更强的鲁棒性。我们的结果突出了Wav-KAN作为开发可解释和高性能神经网络的强大工具的潜力,其应用跨越多个领域。这项工作为进一步在PyTorch和TensorFlow等框架中探索和实现Wav-KAN奠定了基础,旨在使KAN中的小波像通用逼近理论(UAT)中的ReLU和sigmoid激活函数一样普及。重现模拟的代码可在https://github.com/zavareh1/Wav-KAN获取。

关键词

引用

@article{arxiv.2405.12832,
  title  = {Wav-KAN: Wavelet Kolmogorov-Arnold Networks},
  author = {Zavareh Bozorgasl and Hao Chen},
  journal= {arXiv preprint arXiv:2405.12832},
  year   = {2024}
}

备注

Work in progress; codes are available at are available at https://github.com/zavareh1/Wav-KAN