用非参数教学法加速注意力模型训练,不降精度还更快。
Nonparametric Teaching of Attention Learners
- 通过选择关键样本对注意力模型进行教学,提升学习效率。
- 大模型和视觉变换器训练时间减少13%至20.58%。
- 适合追求高效训练且不想牺牲性能的研究者。
注意力学习者(如Transformer)擅长捕捉序列与属性间的隐式关系,但训练成本高。本文提出一种新范式——注意力神经教学(AtteNT),从非参数教学视角重构学习过程。该方法将隐式映射视为密集的序列-属性对集合,由教师选择代表性子集以加速注意力模型收敛。通过分析注意力在参数梯度下降中的作用,并将参数更新演化重铸为非参数教学中的函数梯度下降,首次证明注意力学习是重要性自适应的非参数教学。实验显示,该方法在大语言模型(LLMs)和视觉变换器(ViTs)上分别实现13.01%和20.58%的训练时间降低,涵盖微调与从零训练场景,且性能保持甚至提升,在多种下游任务中表现稳定。
原文摘要 · Abstract (English)
Attention learners, neural networks built on the attention mechanism, e.g., transformers, excel at learning the implicit relationships that relate sequences to their corresponding properties, e.g., mapping a given sequence of tokens to the probability of the next token. However, the learning process tends to be costly. To address this, we present a novel paradigm named Attention Neural Teaching (AtteNT) that reinterprets the learning process through a nonparametric teaching perspective. Specifically, the latter provides a theoretical framework for teaching mappings that are implicitly defined (i.e., nonparametric) via example selection. Such an implicit mapping is embodied through a dense set of sequence-property pairs, with the AtteNT teacher selecting a subset to accelerate convergence in attention learner training. By analytically investigating the role of attention on parameter-based gradient descent during training, and recasting the evolution of attention learners, shaped by parameter updates, through functional gradient descent in nonparametric teaching, we show for the first time that teaching attention learners is consistent with teaching importance-adaptive nonparametric learners. These new findings readily commit AtteNT to enhancing learning efficiency of attention learners. Specifically, we observe training time reductions of 13.01% for LLMs and 20.58% for ViTs, spanning both fine-tuning and training-from-scratch regimes. Crucially, these gains are achieved without compromising accuracy; in fact, performance is consistently preserved and often enhanced across a diverse set of downstream tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。