arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.23630cs.LG
收藏

BTTackler: 一种基于诊断的高效深度学习超参数优化框架

BTTackler: A Diagnosis-based Framework for Efficient Deep Learning Hyperparameter Optimization

School of Software, BNRist, \ University Beijing China;School of Software, EIRI, \ University Beijing China;School of Software, \ University Beijing China;School of Software, BNRist, \ University;School of Software, EIRI, \ University;School of Software, \ University

Zhongyi Pei, Zhiyao Cen, Yipeng Huang, Chen Wang, Lin Liu, Philip Yu, Mingsheng Long

AI总结:

BTTackler通过引入训练诊断自动识别训练问题,有效减少超参数优化时间并提升试验效率。

摘要:

超参数优化(HPO)在深度学习中被认为成本高昂,尤其是在使用自动化方法时。现有的大多数自动化HPO方法都是基于准确性的,即使用准确性指标来指导不同超参数配置在特定搜索空间中的试验。然而,许多试验可能会遇到严重的训练问题,如消失梯度和收敛不足,这些在训练初期难以通过准确性指标反映出来,往往导致性能不佳。这导致了不高效的优化轨迹,因为不良试验占用大量计算资源,并降低了在时间限制内找到优秀超参数配置的可能性。在本文中,我们提出了Bad Trial Tackler(BTTackler),一种新颖的HPO框架,它引入了训练诊断来自动识别训练问题,从而解决不良试验。BTTackler通过计算一组精心设计的量化指标来诊断每个试验,并在检测到任何训练问题时触发早期终止。在代表性的HPO任务上进行了评估,这些任务包括三种经典深度神经网络(DNN)和四种广泛使用的HPO方法。为了更好地量化自动化HPO方法的有效性,我们提出了两种基于准确性和时间消耗的新度量标准。结果表明,BTTackler在两个方面具有优势:(1)它将达到与基线方法相当的准确性的时间消耗减少了40.33%;(2)在给定的时间预算内,它比基线方法多进行了44.5%的前10名试验。我们还发布了一个开源的Python库,使用户能够轻松地将BTTackler应用于自动化HPO流程,只需最小的代码更改。

英文摘要:

Hyperparameter optimization (HPO) is known to be costly in deep learning, especially when leveraging automated approaches. Most of the existing automated HPO methods are accuracy-based, i.e., accuracy metrics are used to guide the trials of different hyperparameter configurations amongst a specific search space. However, many trials may encounter severe training problems, such as vanishing gradients and insufficient convergence, which can hardly be reflected by accuracy metrics in the early stages of the training and often result in poor performance. This leads to an inefficient optimization trajectory because the bad trials occupy considerable computation resources and reduce the probability of finding excellent hyperparameter configurations within a time limitation. In this paper, we propose \textbf{Bad Trial Tackler (BTTackler)}, a novel HPO framework that introduces training diagnosis to identify training problems automatically and hence tackles bad trials. BTTackler diagnoses each trial by calculating a set of carefully designed quantified indicators and triggers early termination if any training problems are detected. Evaluations are performed on representative HPO tasks consisting of three classical deep neural networks (DNN) and four widely used HPO methods. To better quantify the effectiveness of an automated HPO method, we propose two new measurements based on accuracy and time consumption. Results show the advantage of BTTackler on two-fold: (1) it reduces 40.33\% of time consumption to achieve the same accuracy comparable to baseline methods on average and (2) it conducts 44.5\% more top-10 trials than baseline methods on average within a given time budget. We also released an open-source Python library that allows users to easily apply BTTackler to automated HPO processes with minimal code changes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑