SM2通过预训练识别低效配置,用更少能耗找到高性能超参
Spend More to Save More (SM2): An Energy-Aware Implementation of Successive Halving for Sustainable Hyperparameter Optimization
- 用预训练快速筛除低效超参组合,降低能源消耗
- 在多种数据集与硬件上验证,显著减少训练能耗
- 适合关注模型训练可持续性的研究者和工程师
机器学习模型开发中,超参数调优常需多次训练以寻找最优配置。随着模型复杂度提升,仅追求性能的调优方法已难满足可持续性需求。本文提出节能型超参优化方法SM2,基于广泛使用的逐次减半算法,引入探索性预训练机制,在极低能耗下识别无效配置。结合硬件特性与实时能耗监控,SM2在保障模型性能的同时实现高效训练。在多个数据集、模型及硬件环境下实验验证表明,该方法可有效避免超参配置训练过程中的能源浪费。
原文摘要 · Abstract (English)
A fundamental step in the development of machine learning models commonly involves the tuning of hyperparameters, often leading to multiple model training runs to work out the best-performing configuration. As machine learning tasks and models grow in complexity, there is an escalating need for solutions that not only improve performance but also address sustainability concerns. Existing strategies predominantly focus on maximizing the performance of the model without considering energy efficiency. To bridge this gap, in this paper, we introduce Spend More to Save More (SM2), an energy-aware hyperparameter optimization implementation based on the widely adopted successive halving algorithm. Unlike conventional approaches including energy-intensive testing of individual hyperparameter configurations, SM2 employs exploratory pretraining to identify inefficient configurations with minimal energy expenditure. Incorporating hardware characteristics and real-time energy consumption tracking, SM2 identifies an optimal configuration that not only maximizes the performance of the model but also enables energy-efficient training. Experimental validations across various datasets, models, and hardware setups confirm the efficacy of SM2 to prevent the waste of energy during the training of hyperparameter configurations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。