Skip to main content
QUICK REVIEW

[论文解读] A theoretical framework for deep and locally connected ReLU network

Yuandong Tian|arXiv (Cornell University)|Sep 27, 2018
Neural Networks and Applications参考文献 22被引用 7
一句话总结

该论文提出了一种新颖的理论框架,用于深度、局部连接的ReLU网络,采用教师-学生蒸馏方法,将学生的学习除了在自身计算图上,还延伸到教师的计算图上。通过避免不切实际的假设并整合批量归一化等正则化技术,该框架能够对实际深度网络中的泛化、过拟合和解耦表示进行严格分析。

ABSTRACT

Understanding theoretical properties of deep and locally connected nonlinear network, such as deep convolutional neural network (DCNN), is still a hard problem despite its empirical success. In this paper, we propose a novel theoretical framework for such networks with ReLU nonlinearity. The framework explicitly formulates data distribution, favors disentangled representations and is compatible with common regularization techniques such as Batch Norm. The framework is built upon teacher-student setting, by expanding the student forward/backward propagation onto the teacher's computational graph. The resulting model does not impose unrealistic assumptions (e.g., Gaussian inputs, independence of activation, etc). Our framework could help facilitate theoretical analysis of many practical issues, e.g. overfitting, generalization, disentangled representations in deep networks.

研究动机与目标

  • 为解决深度且局部连接的ReLU网络在实际场景中缺乏理论理解的问题。
  • 开发一种不依赖于不切实际假设(如高斯输入分布或独立激活)的理论框架。
  • 实现对过拟合、泛化和解耦表示等关键实际问题的理论分析。
  • 将批量归一化等常见正则化技术整合进理论分析中。
  • 为复杂、局部连接架构中的前向与反向传播提供统一的分析结构。

提出的方法

  • 该框架采用教师-学生训练范式,将学生的前向与反向传播过程显式映射到教师的计算图上。
  • 该映射使学生能够继承教师的表示结构,同时从自身的梯度中学习。
  • 该方法避免了对输入分布或激活独立性的假设,因而适用于真实世界数据。
  • 通过将正则化技术(如批量归一化)嵌入蒸馏过程,该框架自然地集成了这些技术。
  • 理论分析基于计算图的结构,使表示学习动态得以拆解。
  • 该框架通过追踪不同输入因素如何影响学习到的特征,支持对解耦表示的分析。

实验结果

研究问题

  • RQ1如何在不依赖强分布假设的前提下,对深度、局部连接的ReLU网络中的泛化与过拟合进行理论分析?
  • RQ2能否构建一个教师-学生框架,使得反向传播过程尊重教师的架构与表示结构?
  • RQ3批量归一化等正则化技术在多大程度上可以被正式整合进深度网络的理论模型中?
  • RQ4该框架如何支持深度非线性架构中解耦表示的出现?
  • RQ5该框架能否用于分析权重重用、局部连接与表示学习之间的关系?

主要发现

  • 所提出的框架能够在不依赖高斯输入或独立激活等不切实际假设的前提下,对深度、局部连接的ReLU网络进行理论分析。
  • 通过将学生的学习除了在自身计算图上,还嵌入教师的计算图中,该框架支持对前向与反向传播的一致且可解释的分析。
  • 该框架与批量归一化等常见正则化方法兼容,从而能够对这些技术在泛化性上的影响进行理论分析。
  • 该框架通过结构化的梯度流将输入因素与学习到的特征单元关联,为分析解耦表示提供了基础。
  • 该方法通过精确追踪表示复杂度与梯度动态,促进了对过拟合与泛化的研究。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。