arXiv:2410.08003cs.LG2024-10ICLR被引 2

用生物启发的固定路由机制,让模型自动共享相似输入的参数,提升效率和泛化能力。

More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routing

  • 采用固定随机投影代替可训练门控,实现生物启发的专家重叠路由。
  • 相似输入共享更多参数,学习速度更快,对新样本泛化能力更强。
  • 无需任务标识,适用于图像、语言等多类任务,扩展性强。

生物神经系统的演化催生了模块化与稀疏编码,带来能量效率与跨任务鲁棒性。相比之下,标准神经网络依赖密集、非专业化结构,所有参数同时更新以学习多任务,导致干扰。现有稀疏网络方法受限于:1)可训练门控函数引发表征坍塌;2)互不重叠的专家造成冗余计算与慢学习;3)依赖显式输入或任务标识,限制灵活性与可扩展性。本文提出条件重叠专家混合模型(COMET),通过指数级重叠专家构建模块化稀疏架构。COMET以生物启发的固定随机投影替代稀疏专家中的可训练门控,使专家重叠程度依赖输入相似性,相似输入共享更多参数。这提升了每步更新的学习速度并增强了样本外泛化能力。我们在图像分类、语言建模和回归等任务上,使用多种主流深度学习架构验证了COMET的有效性。

原文摘要 · Abstract (English)

The evolution of biological neural systems has led to both modularity and sparse coding, which enables energy efficiency and robustness across the diversity of tasks in the lifespan. In contrast, standard neural networks rely on dense, non-specialized architectures, where all model parameters are simultaneously updated to learn multiple tasks, leading to interference. Current sparse neural network approaches aim to alleviate this issue but are hindered by limitations such as 1) trainable gating functions that cause representation collapse, 2) disjoint experts that result in redundant computation and slow learning, and 3) reliance on explicit input or task IDs that limit flexibility and scalability. In this paper we propose Conditionally Overlapping Mixture of ExperTs (COMET), a general deep learning method that addresses these challenges by inducing a modular, sparse architecture with an exponential number of overlapping experts. COMET replaces the trainable gating function used in Sparse Mixture of Experts with a fixed, biologically inspired random projection applied to individual input representations. This design causes the degree of expert overlap to depend on input similarity, so that similar inputs tend to share more parameters. This results in faster learning per update step and improved out-of-sample generalization. We demonstrate the effectiveness of COMET on a range of tasks, including image classification, language modeling, and regression, using several popular deep learning architectures.

专家模型稀疏网络生物启发路由机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。