arXiv:2503.10065cs.LGcs.CV2025-03CVPR被引 13

挑战神经网络简单性偏见,发现特定任务需更复杂激活函数

Do We Always Need the Simplicity Bias? Looking for Optimal Inductive Biases in the Wild

  • 通过元学习自适应生成任务专属激活函数,突破传统ReLU限制
  • 在表格数据、回归和捷径学习等场景中,新激活函数显著优于ReLU
  • 图像任务仍适合简单性偏见,但其他领域需更高复杂度先验

神经网络倾向于用相对简单的函数拟合数据,这种‘简单性偏见’被视为其成功的关键。本文探究该原则的边界。基于近期发现ReLU激活函数导致简单性偏见[96],我们提出一种方法,可元学习适配特定任务的新激活函数与归纳偏置。研究发现,在多个任务中,简单性偏见不适用,ReLU表现不佳;此时学习到的新激活函数通过引入更高复杂度先验,性能更优。这些场景包括表格数据、回归任务、捷径学习及算法式领悟任务,而这些恰是神经网络长期难以应对的领域。相比之下,在图像任务上,最优学习激活函数几乎与ReLU、GeLU一致。结论表明:尽管普遍认为ReLU的简单性偏见普适有效,但实际仅在图像分类任务中近似最优;其他场景可能需要更复杂的归纳偏置。我们证实激活函数可调控归纳偏置,未来定制化架构或带来更大提升。仍需进一步研究如何超越‘复杂度’这一维度,准确刻画模型归纳偏置与数据的适配性。

原文摘要 · Abstract (English)

Neural architectures tend to fit their data with relatively simple functions. This "simplicity bias" is widely regarded as key to their success. This paper explores the limits of this principle. Building on recent findings that the simplicity bias stems from ReLU activations [96], we introduce a method to meta-learn new activation functions and inductive biases better suited to specific tasks. Findings: We identify multiple tasks where the simplicity bias is inadequate and ReLUs suboptimal. In these cases, we learn new activation functions that perform better by inducing a prior of higher complexity. Interestingly, these cases correspond to domains where neural networks have historically struggled: tabular data, regression tasks, cases of shortcut learning, and algorithmic grokking tasks. In comparison, the simplicity bias induced by ReLUs proves adequate on image tasks where the best learned activations are nearly identical to ReLUs and GeLUs. Implications: Contrary to popular belief, the simplicity bias of ReLU networks is not universally useful. It is near-optimal for image classification, but other inductive biases are sometimes preferable. We showed that activation functions can control these inductive biases, but future tailored architectures might provide further benefits. Advances are still needed to characterize a model's inductive biases beyond "complexity", and their adequacy with the data.

神经网络激活函数归纳偏置元学习

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