更改

跳到导航 跳到搜索
第54行: 第54行:  
==实验设置==
 
==实验设置==
   −
网络架构:
+
===网络架构===
 
(1)Iris任务:我们使用了一个四层神经网络,结构为4→5→5→3。输入层有4个神经元,对应Iris数据集的4个特征(如花萼长度、花萼宽度等);两个隐藏层各有5个神经元;输出层有3个神经元,对应3种鸢尾花的分类。
 
(1)Iris任务:我们使用了一个四层神经网络,结构为4→5→5→3。输入层有4个神经元,对应Iris数据集的4个特征(如花萼长度、花萼宽度等);两个隐藏层各有5个神经元;输出层有3个神经元,对应3种鸢尾花的分类。
 
(2)MNIST任务:我们使用了一个四层神经网络,结构为25→6→6→5。输入层有25个神经元,对应5x5的输入图像(原始MNIST图像为28x28,我们进行了降维处理);两个隐藏层各有6个神经元;输出层有5个神经元,对应数字0到4的分类(我们删除了数字5到9以简化任务)。
 
(2)MNIST任务:我们使用了一个四层神经网络,结构为25→6→6→5。输入层有25个神经元,对应5x5的输入图像(原始MNIST图像为28x28,我们进行了降维处理);两个隐藏层各有6个神经元;输出层有5个神经元,对应数字0到4的分类(我们删除了数字5到9以简化任务)。
   −
训练参数:
+
===训练参数===
 
我们使用均方误差(MSE)作为损失函数,梯度下降作为优化算法,学习率为0.01。Iris网络训练了4000个epoch,批量大小为10;MNIST网络训练了500个epoch,批量大小为50。权重初始化采用均匀分布,确保初始权重在合理范围内。权重从以下分布中采样:
 
我们使用均方误差(MSE)作为损失函数,梯度下降作为优化算法,学习率为0.01。Iris网络训练了4000个epoch,批量大小为10;MNIST网络训练了500个epoch,批量大小为50。权重初始化采用均匀分布,确保初始权重在合理范围内。权重从以下分布中采样:
−
<math>W_{ij} = \mathscr{U}\left([ -\frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}}, \frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}} ]\right)</math>
+
<math>W_{ij} = \mathscr{U}\left([ -\frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}}, \frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}} ]\right)</math>。为了计算EI,我们使用了固定长度的噪声注入(10^7个样本),以确保结果的收敛性。
−
为了计算EI,我们使用了固定长度的噪声注入(10^7个样本),以确保结果的收敛性。
     −
因果结构的变化
+
===实验结果===
−
 
+
如下图所示。(a)和(c)分别展示了在Iris和MNIST任务中,均方误差损失(MSE Loss)随训练epoch的变化情况。在训练初期,两个任务的训练损失和测试损失都迅速下降,表明模型正在快速学习任务;随着 epoch 的增加,损失逐渐趋于平稳,表明模型的学习速度减慢。在后期,训练损失继续下降,但测试损失趋于稳定甚至略有上升,表明模型可能开始过拟合。Iris 任务由于相对简单,损失下降较快且过拟合现象不明显;而 MNIST 任务更复杂,损失下降较慢且过拟合现象更为明显。下图(b)和(d)展示了Iris和MNIST任务中EI随训练epoch的变化情况。Iris任务:图中显示了输入层到第一个隐藏层(X→T1)、第一个隐藏层到第二个隐藏层(T1→T2)、以及第二个隐藏层到输出层(T2→Y)的EI变化。可以看到,EI在训练初期变化显著,随着网络逐渐拟合任务,EI趋于稳定。MNIST任务:图中显示了第一个隐藏层到第二个隐藏层(T1→T2)、第二个隐藏层到第三个隐藏层(T2→T3)、以及第三个隐藏层到输出层(T3→Y)的EI变化。与Iris任务相比,MNIST任务的EI变化更加显著,表明复杂任务导致网络层的因果结构分化更大。以上结果说明任务复杂度对因果结构的影响:对于更复杂的任务(如MNIST),网络层的因果结构在训练过程中表现出更大的分化,不同层的EI变化更加显著。而对于较简单的任务(如Iris),因果结构的分化较少,EI的变化也相对较小。
−
EI与损失曲线的关系:
  −
在训练过程中,我们发现EI的最大变化与损失曲线的最陡部分(即网络学习最快的阶段)显著匹配。这意味着在网络快速学习的阶段,因果结构发生了显著变化。在过拟合阶段,当训练性能与测试性能分离时,EI在所有层上趋于平坦,表明网络的因果结构在这一阶段没有发生显著变化。这意味着网络已经拟合了任务,因果结构趋于稳定。
  −
 
  −
任务复杂度的影响:
  −
对于更复杂的任务(如MNIST),网络层的因果结构在训练过程中表现出更大的分化,不同层的EI变化更加显著。而对于较简单的任务(如Iris),因果结构的分化较少,EI的变化也相对较小。
  −
 
  −
 
  −
下图展示了Iris和MNIST任务中EI随训练epoch的变化情况:
  −
Iris任务:图中显示了输入层到第一个隐藏层(X→T1)、第一个隐藏层到第二个隐藏层(T1→T2)、以及第二个隐藏层到输出层(T2→Y)的EI变化。可以看到,EI在训练初期变化显著,随着网络逐渐拟合任务,EI趋于稳定。
  −
MNIST任务:图中显示了第一个隐藏层到第二个隐藏层(T1→T2)、第二个隐藏层到第三个隐藏层(T2→T3)、以及第三个隐藏层到输出层(T3→Y)的EI变化。与Iris任务相比,MNIST任务的EI变化更加显著,表明复杂任务导致网络层的因果结构分化更大。
  −
 
  −
 
  −
 
  −
为了了解DNN的因果结构在学习过程中是如何变化的,我们跟踪了用两个基准分类任务训练的EIin网络:Iris<ref>R. A. Fisher.  The Design of Experiments. The American Mathematical Monthly, 43(3):180, 1936.  ISSN 00029890. doi: 10.2307/2300364.</ref>和MNIST<ref>Yann  LeCun,  Corinna  Cortes,  and  CJ  Burges.  Mnist  handwritten  digit  database. ATT  Labs  [Online].  Available: <nowiki>http://yann.lecun.com/exdb/mnist</nowiki>, 2, 2010.</ref>。对于Iris,我们训练了具有三个密集连接层4→5→5→3的网络,对于MNIST,我们使用了具有四个紧密连接层25→6→6→5的网络,使用sigmoid激活函数,并且两个任务都没有偏差。对于MNIST,我们将输入从28x28整形为5x5,并从数据集中删除了数字5-9的示例,这样最后一层只有5个节点——这是为了降低准确计算EI的计算成本。这两个任务的网络都用MSE损失和香草梯度下降训练,学习率为0.01。我们训练了批量大小为10的Irisnetworks 4000个epoch和批量大小为50的MNIST网络500个epoch。我们通过从均匀分布<math>W_{ij} = \mathscr{U}\left([ -\frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}}, \frac{1}{\sqrt{\mathrm{fan}_{\mathrm{in}}}} ]\right)</math>中采样来初始化权重。对于每个任务和体系结构,我们执行三次具有不同初始化的运行。使用相同的网络架构,我们还训练了具有tanh和ReLU激活函数的网络——结果可以在SI第7.2节中找到。为了计算EI,我们使用107个样本的固定噪声注入长度。我们发现,在我们的网络中,注入这个长度就足以确保收敛(见SI第7.1节)。然而,请注意,更宽的网络层可能需要更多的样本。     
  −
 
  −
从定性上讲,我们观察到,EI的最大变化与训练过程中损失曲线的最陡部分显著匹配,并且在最大学习期间,EI通常是动态的(如下图所示)。在训练性能与测试性能分离的过拟合期间,EI在所有层上通常都是平坦的,这表明在网络适当拟合后,因果结构中的信息在此期间没有变化。
      
[[文件:图 2.png|替代=|无框|500x500像素]]
 
[[文件:图 2.png|替代=|无框|500x500像素]]
5,443

个编辑

导航菜单