梯度下降能稳健学习图像本质维度,抗背景噪声干扰。
Gradient Descent Robustly Learns the Intrinsic Dimension of Data in Training Convolutional Neural Networks
- 用梯度下降训练CNN,可自动捕捉图像本质维度。
- 即使背景噪声较大,网络秩仍保持稳定,远优于数据秩变化。
- 适合关注模型泛化与数据结构的从业者参考。
现代神经网络通常高度过参数化。人们普遍认为,若数据简单,训练后的网络将自动等价于一个简单预测器。基于此直觉,已有研究探讨了神经网络的“秩”与数据秩的关系。本文聚焦于梯度下降训练的卷积神经网络(CNN)的秩,特别研究其对图像背景噪声的鲁棒性。我们发现,当向图像添加背景噪声时,训练后CNN的秩受扰动远小于数据本身的秩。通过一个具体的数据模型分析,我们证明:即使存在相对较大的背景噪声,梯度下降训练的CNN仍能学习到干净图像的本质维度。我们在合成数据和真实数据集上进行了实验,进一步验证了该结论。
原文摘要 · Abstract (English)
Modern neural networks are usually highly over-parameterized. Behind the wide usage of over-parameterized networks is the belief that, if the data are simple, then the trained network will be automatically equivalent to a simple predictor. Following this intuition, many existing works have studied different notions of "ranks" of neural networks and their relation to the rank of data. In this work, we study the rank of convolutional neural networks (CNNs) trained by gradient descent, with a specific focus on the robustness of the rank to image background noises. Specifically, we point out that, when adding background noises to images, the rank of the CNN trained with gradient descent is affected far less compared with the rank of the data. We support our claim with a theoretical case study, where we consider a particular data model to characterize low-rank clean images with added background noises. We prove that CNNs trained by gradient descent can learn the intrinsic dimension of clean images, despite the presence of relatively large background noises. We also conduct experiments on synthetic and real datasets to further validate our claim.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。