让神经网络权重随任务变化,在低维流形上平滑调节,提升泛化能力。
Walking the Weight Manifold: a Topological Approach to Conditioning Inspired by Neuromodulation
- 将权重设为任务变量的光滑流形函数,而非固定点
- 不同拓扑结构(如直线、椭圆)在测试中优于传统输入拼接
- 适合需要跨任务迁移与鲁棒泛化的场景
人们常希望高效学习一系列相似任务,复用知识。传统方法通过输入注入任务上下文来实现条件化;而大脑则通过血清素等神经调质动态调节参数。受此启发,本文提出将权重设计为任务变量的光滑函数,即优化一个具有预设拓扑结构的权重空间流形。通过最小化损失泛函并约束体积运动,实现类似梯度下降的优化过程。推理时,选择流形上一点作为特定子任务的有效权重矩阵。该策略有两大优势:其一,流形拓扑(如线、环、环面)可作为任务间关系的归纳偏置;其二,单个状态的学习会平滑影响整个流形,促进跨状态泛化。实验验证了直线(如噪声水平)、椭圆(如图像旋转)等拓扑结构的有效性,尽管形式简单,仍优于输入拼接的条件化模型,且对分布外样本泛化更优。结果表明,基于低维流形的权重调制是一种原则性强、有效的替代方案。
原文摘要 · Abstract (English)
One frequently wishes to learn a range of similar tasks as efficiently as possible, re-using knowledge across tasks. In artificial neural networks, this is typically accomplished by conditioning a network upon task context by injecting context as input. Brains have a different strategy: the parameters themselves are modulated as a function of various neuromodulators such as serotonin. Here, we take inspiration from neuromodulation and propose to learn weights which are smoothly parameterized functions of task context variables. Rather than optimize a weight vector, i.e. a single point in weight space, we optimize a smooth manifold in weight space with a predefined topology. To accomplish this, we derive a formal treatment of optimization of manifolds as the minimization of a loss functional subject to a constraint on volumetric movement, analogous to gradient descent. During inference, conditioning selects a single point on this manifold which serves as the effective weight matrix for a particular sub-task. This strategy for conditioning has two main advantages. First, the topology of the manifold (whether a line, circle, or torus) is a convenient lever for inductive biases about the relationship between tasks. Second, learning in one state smoothly affects the entire manifold, encouraging generalization across states. To verify this, we train manifolds with several topologies, including straight lines in weight space (for conditioning on e.g. noise level in input data) and ellipses (for rotated images). Despite their simplicity, these parameterizations outperform conditioning identical networks by input concatenation and better generalize to out-of-distribution samples. These results suggest that modulating weights over low-dimensional manifolds offers a principled and effective alternative to traditional conditioning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。