分析瓶颈层与捷径连接对线性去噪自编码器泛化能力的影响
Impact of Bottleneck Layers and Skip Connections on the Generalization of Linear Denoising Autoencoders
- 构建双层线性去噪自编码器,引入瓶颈层和捷径连接
- 发现瓶颈宽度影响偏差与方差的权衡,捷径可降低方差
- 适用于研究深度模型泛化机制的科研人员
现代深度神经网络在高度过参数化情况下仍表现出强大泛化能力。尽管监督学习中的该现象已取得显著进展,但无监督任务如去噪仍存在诸多未解问题。现有研究多局限于单层线性模型,本文聚焦于在梯度流下训练的两层线性去噪自编码器,引入两个现代深度架构关键组件:低维瓶颈层(实现秩约束)和绕过瓶颈的捷径连接。在乘积正则化下,推导出所有临界点的闭式表达式,特别是最小范数原则下的全局最小值。进一步在过参数化情形下,推导出含与不含捷径连接模型的测试风险公式。分析揭示两个现象:其一,瓶颈层引入类似偏差-方差权衡的复杂度度量——增大瓶颈宽度降低偏差但增加方差;其二,捷径连接可缓解去噪自编码器的方差,尤其在适度过参数化时。通过随机矩阵理论分析捷径影响,并以数值实验验证结论。
原文摘要 · Abstract (English)
Modern deep neural networks exhibit strong generalization even in highly overparameterized regimes. Significant progress has been made to understand this phenomenon in the context of supervised learning, but for unsupervised tasks such as denoising, several open questions remain. While some recent works have successfully characterized the test error of the linear denoising problem, they are limited to linear models (one-layer network). In this work, we focus on two-layer linear denoising autoencoders trained under gradient flow, incorporating two key ingredients of modern deep learning architectures: A low-dimensional bottleneck layer that effectively enforces a rank constraint on the learned solution, as well as the possibility of a skip connection that bypasses the bottleneck. We derive closed-form expressions for all critical points of this model under product regularization, and in particular describe its global minimizer under the minimum-norm principle. From there, we derive the test risk formula in the overparameterized regime, both for models with and without skip connections. Our analysis reveals two interesting phenomena: Firstly, the bottleneck layer introduces an additional complexity measure akin to the classical bias-variance trade-off -- increasing the bottleneck width reduces bias but introduces variance, and vice versa. Secondly, skip connection can mitigate the variance in denoising autoencoders -- especially when the model is mildly overparameterized. We further analyze the impact of skip connections in denoising autoencoder using random matrix theory and support our claims with numerical evidence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。