arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.15212cs.LGcs.CV
收藏

ZENITH: 自动化梯度范数信息驱动的随机优化

ZENITH: Automated Gradient Norm Informed Stochastic Optimization

Dhrubo Saha

AI总结:

提出ZENITH优化器,利用梯度范数的时间演化自动调整学习率,在图像分类、目标检测等任务中实现更高精度和更快速度,且兼容正则化。

摘要:

训练深度计算机视觉模型需要手动监督或超参数调整学习率(LR)调度。虽然现有的自适应优化器自动调度LR,但它们存在计算和内存开销、与正则化不兼容以及次优LR选择的问题。在这项工作中,我们引入了ZENITH(使用范数信息训练历史进行零开销演化)优化器,它利用梯度范数的时间演化来调整LR。跨越6个CNN架构和6个基准的图像分类实验表明,ZENITH在更短的墙钟时间内实现了比基线更高的测试准确率。在使用R-CNN系列模型的MS COCO上,它在目标检测、关键点检测和实例分割中也产生了优越的mAP。此外,它与正则化的兼容性使其能够实现更好的泛化。

英文摘要:

Training deep computer vision models requires manual oversight or hyperparameter tuning of the learning rate (LR) schedule. While existing adaptive optimizers schedule the LR automatically, they suffer from computational and memory overhead, incompatibility with regularization, and suboptimal LR choices. In this work, we introduce the ZENITH (Zero-overhead Evolution using Norm-Informed Training History) optimizer, which adapts the LR using the temporal evolution of the gradient norm. Image classification experiments spanning 6 CNN architectures and 6 benchmarks demonstrate that ZENITH achieves higher test accuracy in lower wall-clock time than baselines. It also yielded superior mAP in object detection, keypoint detection, and instance segmentation on MS COCO using the R-CNN family of models. Furthermore, its compatibility with regularization enables even better generalization.

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

↑