arXiv:2410.09298cs.LG2024-10被引 1

提出无需自回归的可学习上下文模型,用集合不变性提升推理速度与精度。

DeepOSets: Non-Autoregressive In-Context Learning with Permutation-Invariance Inductive Bias

  • 基于集合不变性设计非自回归架构,突破传统注意力机制依赖。
  • 高维场景下准确率显著提升,参数量仅为Transformer的十分之一。
  • 适合需要快速推理且数据规模小的部署场景,如边缘计算。

上下文学习(ICL)是某些机器学习模型在不更新参数的情况下,仅通过用户提示中的示例进行学习的能力。尽管最初在大语言模型中被观察到,且普遍认为源于自回归Transformer的注意力机制,本文通过简化的回归任务证明:即使在非自回归神经架构中,只要引入硬编码的排列不变性归纳偏置,也能实现ICL。该新架构称为DeepOSets,结合了DeepSets的集合学习能力与DeepONets的算子学习特性。我们给出了排列不变回归学习算子的表示定理,并证明DeepOSets是该类算子的通用逼近器。通过全面数值实验,在不同噪声水平、维度和样本量下评估其对线性、多项式及浅层神经网络回归的学习能力。在高维情形下,将DeepSets层替换为Set Transformer后性能进一步提升。结果表明,DeepOSets在参数量减少一个数量级的同时,仍能实现高精度、快速推理。

原文摘要 · Abstract (English)

In-context learning (ICL) is the remarkable ability displayed by some machine learning models to learn from examples provided in a user prompt without any model parameter updates. ICL was first observed in the domain of large language models, and it has been widely assumed that it is a product of the attention mechanism in autoregressive transformers. In this paper, using stylized regression learning tasks, we demonstrate that ICL can emerge in a non-autoregressive neural architecture with a hard-coded permutation-invariance inductive bias. This novel architecture, called DeepOSets, combines the set learning properties of the DeepSets architecture with the operator learning capabilities of Deep Operator Networks (DeepONets). We provide a representation theorem for permutation-invariant regression learning operators and prove that DeepOSets are universal approximators of this class of operators. We performed comprehensive numerical experiments to evaluate the capabilities of DeepOSets in learning linear, polynomial, and shallow neural network regression, under varying noise levels, dimensionalities, and sample sizes. In the high-dimensional regime, accuracy was enhanced by replacing the DeepSets layer with a Set Transformer. Our results show that DeepOSets deliver accurate and fast results with an order of magnitude fewer parameters than a comparable transformer-based alternative.

上下文学习非自回归集合不变性高效模型

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