中文

高阶互信息揭示用于多神经元重要性的协同子网络

机器学习 2022-11-09 v2 信息论 math.IT 神经元与认知

摘要

量化哪些神经元对于已训练神经网络的分类决策是重要的,对于理解其内在机制至关重要。先前工作主要将重要性归因于单个神经元。本工作中,我们使用一种称为 O-信息 的多变量互信息方法,研究哪些神经元组包含协同或冗余信息。我们观察到第一层由冗余主导,表明通用的共享特征(如检测边缘),而最后一层由协同主导,表明局部的类别特定特征(如概念)。最后,我们展示 O-信息可用于多神经元重要性。这可通过重新训练一个协同子网络来演示,其导致性能的最小变化。这些结果表明我们的方法可用于剪枝和无监督表示学习。

关键词

引用

@article{arxiv.2211.00416,
  title  = {Higher-order mutual information reveals synergistic sub-networks for multi-neuron importance},
  author = {Kenzo Clauw and Sebastiano Stramaglia and Daniele Marinazzo},
  journal= {arXiv preprint arXiv:2211.00416},
  year   = {2022}
}

备注

Paper presented at InfoCog @ NeurIPS 2022