ICLRFeb, 2024

控制方差的自适应采样高效反向传播

TL;DR通过使用采样算法进行神经网络训练,我们引入了一种方差控制的自适应采样(VCAS)方法,该方法可以在数据和权重梯度计算过程中使用细粒度的分层重要性采样和杠杆得分采样,以加快训练过程并降低计算量。通过在多个任务中进行评估,我们发现 VCAS 可以在保持准确性的同时,减少最高达 73.87% 的 BP 计算量和 49.58% 的整个训练过程计算量。