中文

神经网络中的学习阶段:从拟合多数到拟合少数

机器学习 2022-02-18 v1

摘要

深度神经网络的学习动力学存在争议。基于信息瓶颈(IB)理论,有人提出了分离的拟合与压缩阶段,但此后受到激烈争论。我们通过分析训练过程中参数的演化,以及一层对输入的重构能力与预测性能,来研究学习动力学。我们证明,在数据的温和假设下,存在先降低重构损失的原型阶段,随后降低少数样本的分类损失,而这会增大重构损失。除对单层分类网络给出数学分析外,我们还使用计算机视觉中如 ResNet 和 VGG 等常用数据集与架构评估了该行为。

关键词

引用

@article{arxiv.2202.08299,
  title  = {The learning phases in NN: From Fitting the Majority to Fitting a Few},
  author = {Johannes Schneider},
  journal= {arXiv preprint arXiv:2202.08299},
  year   = {2022}
}