The loss landscapes of neural networks contain minima and saddle points that may be connected in flat regions or appear in isolation. We identify and characterize a special structure in the loss landscape: channels along which the loss decreases extremely slowly, while the output weights of at least two neurons, $a_i$ and $a_j$, diverge to $\pm$infinity, and their input weight vectors, $\mathbf{w_i}$ and $\mathbf{w_j}$, become equal to each other. At convergence, the two neurons implement a gated linear unit: $a_iσ(\mathbf{w_i} \cdot \mathbf{x}) + a_jσ(\mathbf{w_j} \cdot \mathbf{x}) \rightarrow σ(\mathbf{w} \cdot \mathbf{x}) + (\mathbf{v} \cdot \mathbf{x}) σ'(\mathbf{w} \cdot \mathbf{x})$. Geometrically, these channels to infinity are asymptotically parallel to symmetry-induced lines of critical points. Gradient flow solvers, and related optimization methods like SGD or ADAM, reach the channels with high probability in diverse regression settings, but without careful inspection they look like flat local minima with finite parameter values. Our characterization provides a comprehensive picture of these quasi-flat regions in terms of gradient dynamics, geometry, and functional interpretation. The emergence of gated linear units at the end of the channels highlights a surprising aspect of the computational capabilities of fully connected layers.


翻译:神经网络的损失景观包含可能通过平坦区域相互连接或以孤立形式存在的最小值和鞍点。我们识别并刻画了损失景观中的一种特殊结构:沿着这些通道损失下降极为缓慢,同时至少两个神经元的输出权重$a_i$和$a_j$发散至$\pm$无穷大,且它们的输入权重向量$\mathbf{w_i}$和$\mathbf{w_j}$彼此相等。在收敛时,这两个神经元实现了一个门控线性单元:$a_iσ(\mathbf{w_i} \cdot \mathbf{x}) + a_jσ(\mathbf{w_j} \cdot \mathbf{x}) \rightarrow σ(\mathbf{w} \cdot \mathbf{x}) + (\mathbf{v} \cdot \mathbf{x}) σ'(\mathbf{w} \cdot \mathbf{x})$。从几何角度看,这些通往无穷大的通道与对称性诱导的临界点线渐近平行。梯度流求解器以及相关的优化方法(如SGD或Adam)在多种回归设置中以高概率到达这些通道,但若不经仔细检查,它们看起来像是参数值有限的平坦局部最小值。我们的刻画从梯度动力学、几何和函数解释的角度为这些准平坦区域提供了全面的图景。通道末端门控线性单元的出现突显了全连接层计算能力的一个令人惊讶的方面。

0
下载
关闭预览

相关内容

专知会员服务
74+阅读 · 2021年5月28日
图神经网络最近这么火,不妨看看我们精选的这七篇
人工智能前沿讲习班
37+阅读 · 2018年12月10日
CosFace: Large Margin Cosine Loss for Deep Face Recognition论文笔记
统计学习与视觉计算组
44+阅读 · 2018年4月25日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
最新内容
《拥抱新兴技术:面向未来军官的教育革新》
专知会员服务
0+阅读 · 4分钟前
《无人地面战车(UGV)的崛起》报告
专知会员服务
7+阅读 · 7月16日
美陆军任务式指挥人工智能解决方案
专知会员服务
11+阅读 · 7月16日
相关VIP内容
专知会员服务
74+阅读 · 2021年5月28日
Top
微信扫码咨询专知VIP会员