May, 2024

分布式学习中的创新网络

TL;DR本论文介绍了 Wavelet Kolmogorov-Arnold Networks(Wav-KAN)在联邦学习中的开发和应用,并在客户端中实现了 Wav-KAN,通过连续小波变换和离散小波变换实现了多分辨率能力,帮助在客户端之间进行异构数据分布。通过在不同数据集上进行广泛实验,证明了 Wav-KAN 在可解释性、计算速度、训练和测试精度方面具有优越性。我们的联邦学习算法集成了基于小波的激活函数,通过权重、尺度和平移参数化,以提高本地和全局模型性能。结果显示,在计算效率、鲁棒性和准确性方面都有显著改进,突出了小波选择在可扩展神经网络设计中的有效性。