arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.04455cs.LGcs.AIstat.ME

用于具有可训练激活函数的稀疏神经网络的贝叶斯优化

Bayesian optimization for sparse neural networks with trainable activation functions

University of Sfax(斯法克斯大学);Toulouse INP(图卢兹国立理工学院);University of Toulouse(图卢兹大学)

Mohamed Fakhfakh, Lotfi Chaari

AI总结:

针对可训练激活函数的参数估计与稀疏神经网络优化问题,提出全贝叶斯模型和基于 MCMC 的全局收敛采样方案,并在多数据集与 CNN 上验证其准确率提升。

摘要:

在深度神经网络文献中,开发能够提升神经网络性能的激活函数受到了广泛关注。近年来,提出可在整个学习过程中训练的激活函数重新引起了科学兴趣,因为这类函数似乎能够改善网络性能,尤其是通过减少过拟合。本文提出一种参数需要估计的可训练激活函数。作者建立了一个全贝叶斯模型,以从学习数据中自动估计模型权重和激活函数参数。文中开发了一种基于 MCMC 的优化方案来构建推断。所提方法旨在解决上述问题,并通过使用可保证收敛到全局最大值的高效采样方案来改善收敛时间。该方案在三个数据集上结合三种不同的 CNN 进行了测试。令人鼓舞的结果表明,由于所提出的激活函数和参数的贝叶斯估计,本文方法在提升模型准确率方面具有实用性。

英文摘要:

In the literature on deep neural networks, there is considerable interest in developing activation functions that can enhance neural network performance. In recent years, there has been renewed scientific interest in proposing activation functions that can be trained throughout the learning process, as they appear to improve network performance, especially by reducing overfitting. In this paper, we propose a trainable activation function whose parameters need to be estimated. A fully Bayesian model is developed to automatically estimate from the learning data both the model weights and activation function parameters. An MCMC-based optimization scheme is developed to build the inference. The proposed method aims to solve the aforementioned problems and improve convergence time by using an efficient sampling scheme that guarantees convergence to the global maximum. The proposed scheme is tested on three datasets with three different CNNs. Promising results demonstrate the usefulness of our proposed approach in improving model accuracy due to the proposed activation function and Bayesian estimation of the parameters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑