中文精炼导读

核心观点

  • 这是 Distill Circuits 线程的结构性现象系列之一:作者把可解释性看作"神经网络的解剖学",之前的工作都在研究"小静脉"(单个神经元与小电路),而分支特化(branch specialization)研究的是更大尺度的结构——神经网络的"器官"或"脑区"。
  • 分支特化发生在"网络层被拆成分支"时:神经元与电路倾向于自组织,把相关功能聚到各分支,形成更大的功能单元——一种"神经网络脑区";并且这些结构在"没有分支的网络"里也隐含存在,分支只是把原本就存在的结构"具象化"出来。
  • 最早、也是文献里唯一被讨论过的例子来自 AlexNet:它的前两层被分成两个无法通信、直到第二层后才合并的分支(这是为在两块 GPU 上高效训练而设计的),结果第一层神经元自发组织成两组——一支是黑白 Gabor 滤波器、另一支是低频颜色检测器。
  • 统计上这几乎不可能是偶然:InceptionV1 的 mixed3a_5x5 分支只占全层 256 个神经元中的 32 个,但层里全部 9 个"黑白 vs 颜色"检测器都在它里面(巧合概率 <1/10⁸);mixed3b_5x5 占 480 中的 96 个,却装下了全部 30 个曲线相关特征(概率 <1/10²⁰)。
  • 作者认为训练中存在"正反馈回路":把网络切成通信受限的几块时,把相似特征放在一起是合理的——它们更需要共享信息;文章还提出了与神经科学的对话:大脑中单个神经元的激活函数在不同电路间被精细调优,若人为网络的"分支特化"也能类似微调,或许能增强性能,并有望通过操纵分支来对照脑疾病(如阿尔茨海默病早期受损的特定神经元群体)。
AlexNet 前两层分支特化
AlexNet 前两层分支特化

内容精讲

文章用"解剖学"的隐喻开场:如果可解释性是"神经网络的解剖学",Circuits 线程此前大多在观察"细小的血管"——小尺度、单个神经元及其连接。但生物学解剖中最突出的抽象是更大尺度的结构:心脏这样的器官、呼吸系统这样的器官系统。作者由此发问:人工神经网络里有没有"呼吸系统""心脏"或"脑区"?是否存在比电路更大的、可研究的涌现结构?分支特化就是他们观察到的三个较大"结构性现象"之一(另两个是等变性 motif 与权重带状)。其定义是:当神经网络层被分成若干分支时,神经元与电路倾向于自组织、把相关功能聚进各分支、形成更大的功能单元;而且有证据表明这些结构在没有分支的网络里也隐含存在——分支只是把原本就有的结构显现出来。

最早的、文献中唯一讨论过的例子来自 AlexNet。AlexNet 以推动深度学习革命著称,但论文里埋藏着一个迷人而少被讨论的细节:它的前两层被拆成两个分支,这两支直到第二层之后才能重新通信——这个结构是为了在两块 GPU 上最大化训练效率。有趣的是,作者们注意到由此产生了一个奇怪的结果:第一层的神经元自发组织成两组——黑白 Gabor 滤波器聚集在一支、低频颜色检测器聚集在另一支。作者用特征可视化(而非第一层那种直接看 RGB 权重的方法)展示了分支特化如何延伸到每个分支的第二层。

"什么是分支?"许多神经网络架构都有分支——一系列层暂时无法访问"并行的"、仍会传给后续层的信息。显式带分支的模型在过去很流行(AlexNet、Inception 家族),近年变少,但残差网络(可看作在残差块里隐式地有分支)变得非常普遍;神经架构搜索有时也会自动演化出分支结构。残差网络的隐式分支有些微妙:乍看每层都是双向分支,但因为分支靠加法合并,可以把模型改写成"并行分支"来理解。在很深的残差网络(如 ResNet-152)里通常能看到残差块特化;作者猜测原因是:在这些模型里,层的精确深度不重要,分支性比顺序性更突出。普通分支模型有一个概念弱点——虽然分支能省参数,但"在分支间混合值"仍需大量参数;而如果把残差网络按分支解释,就能看作一种绕开这个问题的策略:残差网络用低秩连接(把所有块投影到同一个和再投影回来)来混合分支——一种优雅的处理方式。这从实践上提示:分析残差网络时应密切关注块里的单元,且可以预期残差流异常地"多语义"(polysemantic)。

分支特化为什么发生?定义上,分支特化就是特征在分支间的组织方式不同:普通层里特征随机组织,一个特征等可能落在层里任何神经元上;而分支层里,某种类型的特征常常聚集到某一个分支——分支特化到了那种特征上。作者的直觉是训练期间存在正反馈回路(文中配图示意了这种假想回路):换个角度想,如果必须把神经网络切成"彼此通信能力有限"的几块,把相似特征放在一起是合理的——因为它们可能更需要共享信息。

InceptionV1 各 5x5 分支中的特征特化
InceptionV1 各 5x5 分支中的特征特化

分支特化不止发生在第一层。AlexNet 的后面层也被拆成分支,但在更深的层里,分支特化很微妙——不再是整体一刀切,而是"几十个小簇、每簇被分配到一个分支",很难确信不是噪声。但其他模型在后层有非常清晰的分支特化,这通常发生在"分支只占层的一小部分"时(要么分支很多、要么某个分支远小于其他)。这时分支可以特化到层中存在的很小一部分特征、显现出清晰模式。InceptionV1 大多数层都有分支结构:各分支单元数与卷积尺寸不同,5x5 分支最小但卷积尺寸最大,且常常高度特化。文中给出惊人的统计:mixed3a_5x5 只有 32/256 个神经元,却装下了全部 9 个黑白/颜色检测器(巧合概率 <1/10⁸);mixed3b_5x5 只有 96/480 个神经元,却装下了全部 30 个曲线特征(概率 <1/10²⁰)。作者还排除了一个混淆因素:5x5 分支既最小、卷积也更大(5x5 而非 3x3 或 1x1);但 mixed3a 与 mixed3b 是相邻层、包含相对相似的特征、在同一尺度——如果只是卷积尺寸的作用,为什么 mixed3a_5x5 里没有曲线、mixed3b_5x5 里没有颜色?

最令人惊讶的是分支特化的一致性:同样的特化在不同架构、不同任务上反复出现(如 AlexNet 第一层黑白 Gabor 分支 vs 低频颜色分支的拆分,是个异常鲁棒的现象)。文章结尾展望了与神经科学的互惠:从构建人工网络的角度,分支特化的单元与连接规则的"按分支微调"能否提升性能?大脑有证据表明单个神经元的激活函数在甚至不同神经电路内部被精细调优;若这种调优给大脑带来收益,人工网络也可能受益。从理解大脑的角度,分支特化能否产生可实验检验的预测?如果人工网络能工程化出与大脑分支通路组织相似的分支,对网络的操纵就能与光遗传/化学遗传实验对照——许多脑疾病涉及特定神经群体的变化(如阿尔茨海默病早期被破坏的特定神经元群体),在网络模型中破坏对应单元可以探索由此产生的计算缺陷与恢复认知功能的可能策略。

阅读价值

适合可解释性、机制性可解释性与神经网络结构研究方向的读者:这篇文章把"分支特化"这个整层尺度的结构性现象讲得很清楚,既包含 AlexNet 的历史细节与 InceptionV1 的惊人统计,也包含"残差块即并行分支"的重新解读与正反馈回路假设;其"从器官尺度看网络"的视角,以及把人工网络分支与脑电路/脑疾病对照的想法,对思考更大尺度涌现结构很有启发性。

阅读英文原文 → 返回 AI 技术文档

本文为中文精炼导读,由 AI 基于原文整理,内容与图片版权归原作者所有。原文: https://distill.pub/2020/circuits/branch-specialization