中文

神经雕刻:通过剪枝与网络分析揭示神经网络中层次化模块化任务结构

机器学习 2023-10-31 v3

摘要

自然目标函数与任务通常表现出层次化模块化——它们可被分解为更简单的子函数,并以层次方式组织。此类子函数具有两个重要特征:它们拥有一组 distinct 的输入(输入可分离性),且它们在层次更高处被复用为输入(可复用性)。先前的研究已确立,本质稀疏的层次化模块化神经网络具有诸如学习效率、泛化、多任务学习和迁移等益处。然而,对于给定的任务,识别潜在的子函数及其层次结构可能具有挑战性。本工作的核心问题是:若我们使用足够深的神经网络学习一个任务,如何揭示该任务中潜在的子函数层次?作为起点,我们考察布尔函数领域,其中更易判断一个任务是否具有层次化模块化。我们提出一种基于迭代单元与边剪枝(在训练期间)并结合用于模块检测和层次推断的网络分析的方法。最后,我们证明该方法能够揭示广泛布尔函数及基于MNIST数字数据集的两个视觉任务的层次化模块化。

关键词

引用

@article{arxiv.2305.18402,
  title  = {Neural Sculpting: Uncovering hierarchically modular task structure in neural networks through pruning and network analysis},
  author = {Shreyas Malakarjun Patil and Loizos Michael and Constantine Dovrolis},
  journal= {arXiv preprint arXiv:2305.18402},
  year   = {2023}
}