通过浅层特征动态剪枝,实现无损训练加速
Partial Forward Blocking: A Novel Data Pruning Paradigm for Lossless Training Acceleration
- 基于目标模型浅层特征评估样本重要性,仅保留关键样本继续计算
- 在ImageNet上剪掉40%数据,训练时间减少33%,准确率提升0.5%
- 无需梯度或代理模型,自动优先处理稀有样本,适合大规模训练优化
训练数据规模的持续增长提升了现代机器学习模型的泛化能力,但也带来了高昂的计算成本。现有数据剪枝方法通常依赖梯度或代理模型,导致反向传播和模型训练开销巨大。本文提出部分前向阻断(PFB)框架,实现无损训练加速。其核心在于自适应剪枝流程:基于目标模型浅层提取的特征评估样本重要性,对不重要样本进行剪枝,仅让保留样本继续后续前向传播与损失反向传播。该机制显著降低深层前向与反向计算开销,同时避免辅助反向计算和代理模型训练。PFB引入概率密度作为样本重要性指标,并结合自适应分布估计模块,动态优先处理相对稀有的样本,与训练状态变化同步。大量实验表明,PFB在性能与速度上均具显著优势:在ImageNet上,剪掉40%数据后,准确率提升0.5%,训练时间减少33%。
原文摘要 · Abstract (English)
The ever-growing size of training datasets enhances the generalization capability of modern machine learning models but also incurs exorbitant computational costs. Existing data pruning approaches aim to accelerate training by removing those less important samples. However, they often rely on gradients or proxy models, leading to prohibitive additional costs of gradient back-propagation and proxy model training. In this paper, we propose Partial Forward Blocking (PFB), a novel framework for lossless training acceleration. The efficiency of PFB stems from its unique adaptive pruning pipeline: sample importance is assessed based on features extracted from the shallow layers of the target model. Less important samples are then pruned, allowing only the retained ones to proceed with the subsequent forward pass and loss back-propagation. This mechanism significantly reduces the computational overhead of deep-layer forward passes and back-propagation for pruned samples, while also eliminating the need for auxiliary backward computations and proxy model training. Moreover, PFB introduces probability density as an indicator of sample importance. Combined with an adaptive distribution estimation module, our method dynamically prioritizes relatively rare samples, aligning with the constantly evolving training state. Extensive experiments demonstrate the significant superiority of PFB in performance and speed. On ImageNet, PFB achieves a 0.5% accuracy improvement and 33% training time reduction with 40% data pruned.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。