arXiv:2410.20035cs.LGcs.AI2024-10NeurIPS被引 4

用引导网络给难训练模型注入先验,让全连接网也能在ImageNet上不崩溃

Training the Untrainable: Introducing Inductive Bias via Representational Alignment

  • 用冻结的引导网络通过表征相似性约束目标网络
  • 全连接网在ImageNet上过拟合问题被解决,准确率显著提升
  • 适合想改造旧架构或理解模型先验的研究者

我们证明,传统上被认为不适合特定任务的模型,可通过另一模型的归纳偏置实现有效训练。当模型在调参后仍过拟合、欠拟合或收敛到差结果时,称其为不可训练。例如,全连接网络在图像识别中过拟合,无残差连接的深层卷积网络则欠拟合。传统做法是更换架构以引入归纳偏置,但偏置的具体形式未知。我们提出‘引导’机制:一个引导网络通过神经距离函数指导目标网络,使其最小化任务损失与各层表征相似度。若引导网络已训练,则传递其架构先验和知识;若未训练,则仅传递部分架构先验。实验显示,该方法防止了全连接网络在ImageNet上的过拟合,缩小了普通RNN与Transformer的性能差距,使普通卷积网络逼近残差网络精度,并提升Transformer在序列任务中的表现。进一步发现,仅用引导驱动初始化即可缓解全连接网络过拟合。本方法提供数学工具研究模型先验与架构关系,长远可推动架构自动化设计。

原文摘要 · Abstract (English)

We demonstrate that architectures which traditionally are considered to be ill-suited for a task can be trained using inductive biases from another architecture. We call a network untrainable when it overfits, underfits, or converges to poor results even when tuning their hyperparameters. For example, fully connected networks overfit on object recognition while deep convolutional networks without residual connections underfit. The traditional answer is to change the architecture to impose some inductive bias, although the nature of that bias is unknown. We introduce guidance, where a guide network steers a target network using a neural distance function. The target minimizes its task loss plus a layerwise representational similarity against the frozen guide. If the guide is trained, this transfers over the architectural prior and knowledge of the guide to the target. If the guide is untrained, this transfers over only part of the architectural prior of the guide. We show that guidance prevents FCN overfitting on ImageNet, narrows the vanilla RNN-Transformer gap, boosts plain CNNs toward ResNet accuracy, and aids Transformers on RNN-favored tasks. We further identify that guidance-driven initialization alone can mitigate FCN overfitting. Our method provides a mathematical tool to investigate priors and architectures, and in the long term, could automate architecture design.

模型训练归纳偏置表征对齐架构改进

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。