arXiv:2602.05688cs.LGcs.AI2026-02被引 3

用进化搜索发现通用激活函数,能提升性能并控制模型偏好。

Mining Generalizable Activation Functions

  • 用大模型做变异算子,自动探索所有符合算力约束的Python函数空间。
  • 小规模合成数据即可让AlphaEvolve找到有效激活函数,且在分布外数据上表现更好。
  • 可指定目标如泛化能力,让模型学习特定数据结构偏好,适合研究架构偏置者。

激活函数的选择是当前研究热点,不同设计旨在提升优化效率并保持表达能力。此外,激活函数会显著改变模型的隐式归纳偏置,从而调控其非线性行为。本文延续先前工作,认为进化搜索是发现新激活函数的有效框架,并提出两个新见解:首先,现代流水线(如AlphaEvolve)利用前沿大模型作为变异操作符,可覆盖任意符合特定浮点运算预算的Python函数空间,无需人工设计搜索空间;同时,因大模型具备常识表征能力,搜索过程更偏向有意义的激活函数,提高搜索效率。其次,通过该框架,不仅能优化性能,还能主动设计具有特定归纳偏置的激活函数,例如以分布外数据表现作为适应度函数,反映模型对数据内在结构的尊重程度,而独立于分布漂移。我们进行了实证探索,表明相对小规模的合成数据即可使AlphaEvolve发现有效激活函数。

原文摘要 · Abstract (English)

The choice of activation function is an active area of research, with different proposals aimed at improving optimization, while maintaining expressivity. Additionally, the activation function can significantly alter the implicit inductive bias of the architecture, controlling its non-linear behavior. In this paper, in line with previous work, we argue that evolutionary search provides a useful framework for finding new activation functions, while we also make two novel observations. The first is that modern pipelines, such as AlphaEvolve, which relies on frontier LLMs as a mutator operator, allows for a much wider and flexible search space; e.g., over all possible python functions within a certain FLOP budget, eliminating the need for manually constructed search spaces. In addition, these pipelines will be biased towards meaningful activation functions, given their ability to represent common knowledge, leading to a potentially more efficient search of the space. The second observation is that, through this framework, one can target not only performance improvements but also activation functions that encode particular inductive biases. This can be done by using performance on out-of-distribution data as a fitness function, reflecting the degree to which the architecture respects the inherent structure in the data in a manner independent of distribution shifts. We carry an empirical exploration of this proposal and show that relatively small scale synthetic datasets can be sufficient for AlphaEvolve to discover meaningful activations.

激活函数进化计算归纳偏置AlphaEvolve

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