用AutoML优化深度移位网络,兼顾性能与能耗
Leveraging AutoML for Sustainable Deep Learning: A Multi-Objective HPO Approach on Deep Shift Neural Networks

- 结合多保真度与多目标优化,自动寻找模型设计最优权衡
- 性能提升约20%,部分场景碳排放降低超60%
- 发现小范围低精度量化反而最省电,适用于资源受限场景
深度学习在从大数据中提取复杂模式方面推动了多个领域的发展,但其计算需求带来了环境与资源挑战。深度移位神经网络(DSNNs)通过移位操作降低推理时的计算复杂度,但仍缺乏充分理解与优化。本文利用AutoML技术,针对图像分类任务,在低资源环境下探索DSNNs的设计潜力。考虑到准确率与能耗等多重目标,融合先进的多保真度(MF)超参数优化(HPO)与多目标优化方法,寻找帕累托最优解。实验表明,相比默认配置,新方案在损失与碳排放上均有显著改善:性能提升约20%,部分情况下碳排放减少超过60%。通过分析量化网络在能耗与准确率上的表现,发现模型特异性权衡现象,例如在某些情况下,仅对网络小部分采用低精度量化可实现最佳能效,同时保持甚至提升性能。该结论在多个骨干网络架构中得到验证,揭示了量化策略的细微差别,并提供了一种自动化平衡能效与性能的方法。
原文摘要 · Abstract (English)
Deep Learning (DL) has advanced various fields by extracting complex patterns from large datasets. However, the computational demands of DL models pose environmental and resource challenges. Deep Shift Neural Networks (DSNNs) present a solution by leveraging shift operations to reduce computational complexity at inference. Compared to common DNNs, DSNNs are still less well understood and less well optimized. By leveraging AutoML techniques, we provide valuable insights into the potential of DSNNs and how to design them in a better way. We focus on image classification, a core task in computer vision, especially in low-resource environments. Since we consider complementary objectives such as accuracy and energy consumption, we combine state-of-the-art multi-fidelity (MF) hyperparameter optimization (HPO) with multi-objective optimization to find a set of Pareto optimal trade-offs on how to design DSNNs. Our approach led to significantly better configurations of DSNNs regarding loss and emissions compared to default DSNNs. This includes simultaneously increasing performance by about 20% and reducing emissions, in some cases by more than 60%. Investigating the behavior of quantized networks in terms of both emissions and accuracy, our experiments reveal surprising model-specific trade-offs, yielding the greatest energy savings. For example, in contrast to common expectations, quantizing smaller portions of the network with low precision can be optimal with respect to energy consumption while retaining or improving performance. We corroborated these findings across multiple backbone architectures, highlighting important nuances in quantization strategies and offering an automated approach to balancing energy efficiency and model performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。