arXiv:2410.05966cs.LGcs.AI2024-10ICLR被引 4

优化前向学习的采样策略,用最少查询提升梯度估计精度。

FLOPS: Forward Learning with OPtimal Sampling

  • 根据数据点重要性动态分配查询次数,降低冗余计算。
  • 在视觉变换器微调中显著减少查询量,保持精度不变。
  • 可直接嵌入现有模型,适用于黑盒场景如提示调优和多模态对齐。

由于反向传播的局限性,仅通过前向传递(即查询)进行学习的方法近年来受到关注。传统前向学习对每个数据点使用大量查询以获得准确的梯度估计,严重制约了算法的可扩展性。然而,并非所有数据点都需同等数量的查询。本文从最小化梯度估计方差成本的角度出发,提出在训练一个批次中为每个数据点分配最优查询数,实现估计精度与计算效率的平衡。通过简化代理目标函数和重参数化技术,设计了一种参数极少、可即插即用的查询分配器,并给出了理论证明其最优性。在多个数据集上对视觉变换器进行微调的实验表明,该方法显著提升了前向学习算法的可扩展性。进一步将分配器应用于两个黑盒任务:提示调优和基础模型的多模态对齐,结果均验证了其有效性,为前向学习的实际应用铺平道路。

原文摘要 · Abstract (English)

Given the limitations of backpropagation, perturbation-based gradient computation methods have recently gained focus for learning with only forward passes, also referred to as queries. Conventional forward learning consumes enormous queries on each data point for accurate gradient estimation through Monte Carlo sampling, which hinders the scalability of those algorithms. However, not all data points deserve equal queries for gradient estimation. In this paper, we study the problem of improving the forward learning efficiency from a novel perspective: how to reduce the gradient estimation variance with minimum cost? For this, we propose to allocate the optimal number of queries over each data in one batch during training to achieve a good balance between estimation accuracy and computational efficiency. Specifically, with a simplified proxy objective and a reparameterization technique, we derive a novel plug-and-play query allocator with minimal parameters. Theoretical results are carried out to verify its optimality. We conduct extensive experiments for fine-tuning Vision Transformers on various datasets and further deploy the allocator to two black-box applications: prompt tuning and multimodal alignment for foundation models. All findings demonstrate that our proposed allocator significantly enhances the scalability of forward-learning algorithms, paving the way for real-world applications.

前向学习梯度估计采样优化

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