Sep, 2023
贝叶斯模型简化的深度神经网络的贝叶斯稀疏化
Bayesian sparsification for deep neural networks with Bayesian model reduction
Dimitrije Marković, Karl J. Friston, Stefan J. Kiebel
TL;DR深度学习的复杂模型限制了其巨大潜力的发挥，需要高效的稀疏化技术。贝叶斯稀疏化是一种关键方法，能够设计出在各种深度学习应用中既计算效率高又性能竞争力强的模型。本研究指出贝叶斯模型简化是一种更高效的模型参数修剪方法，相对于现有的基于随机变分推断的方案，具有更好的计算效率和修剪率。研究中通过对各种深度学习架构的实例进行了验证，包括经典的网络如 LeNet 以及现代框架如视觉 Transformer 和 MLP-Mixer。