Saved in:
Bibliographic Details
Main Authors: Zheng, Hongye, Wang, Bingxing, Xiao, Minheng, Qin, Honglin, Wu, Zhizhong, Tan, Lianghao
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.11839
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909293361496064
author Zheng, Hongye
Wang, Bingxing
Xiao, Minheng
Qin, Honglin
Wu, Zhizhong
Tan, Lianghao
author_facet Zheng, Hongye
Wang, Bingxing
Xiao, Minheng
Qin, Honglin
Wu, Zhizhong
Tan, Lianghao
contents Adaptive optimizers are pivotal in guiding the weight updates of deep neural networks, yet they often face challenges such as poor generalization and oscillation issues. To counter these, we introduce sigSignGrad and tanhSignGrad, two novel optimizers that integrate adaptive friction coefficients based on the Sigmoid and Tanh functions, respectively. These algorithms leverage short-term gradient information, a feature overlooked in traditional Adam variants like diffGrad and AngularGrad, to enhance parameter updates and convergence.Our theoretical analysis demonstrates the wide-ranging adjustment capability of the friction coefficient S, which aligns with targeted parameter update strategies and outperforms existing methods in both optimization trajectory smoothness and convergence rate. Extensive experiments on CIFAR-10, CIFAR-100, and Mini-ImageNet datasets using ResNet50 and ViT architectures confirm the superior performance of our proposed optimizers, showcasing improved accuracy and reduced training time. The innovative approach of integrating adaptive friction coefficients as plug-ins into existing optimizers, exemplified by the sigSignAdamW and sigSignAdamP variants, presents a promising strategy for boosting the optimization performance of established algorithms. The findings of this study contribute to the advancement of optimizer design in deep learning.
format Preprint
id arxiv_https___arxiv_org_abs_2408_11839
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Adaptive Friction in Deep Learning: Enhancing Optimizers with Sigmoid and Tanh Function
Zheng, Hongye
Wang, Bingxing
Xiao, Minheng
Qin, Honglin
Wu, Zhizhong
Tan, Lianghao
Machine Learning
Artificial Intelligence
Adaptive optimizers are pivotal in guiding the weight updates of deep neural networks, yet they often face challenges such as poor generalization and oscillation issues. To counter these, we introduce sigSignGrad and tanhSignGrad, two novel optimizers that integrate adaptive friction coefficients based on the Sigmoid and Tanh functions, respectively. These algorithms leverage short-term gradient information, a feature overlooked in traditional Adam variants like diffGrad and AngularGrad, to enhance parameter updates and convergence.Our theoretical analysis demonstrates the wide-ranging adjustment capability of the friction coefficient S, which aligns with targeted parameter update strategies and outperforms existing methods in both optimization trajectory smoothness and convergence rate. Extensive experiments on CIFAR-10, CIFAR-100, and Mini-ImageNet datasets using ResNet50 and ViT architectures confirm the superior performance of our proposed optimizers, showcasing improved accuracy and reduced training time. The innovative approach of integrating adaptive friction coefficients as plug-ins into existing optimizers, exemplified by the sigSignAdamW and sigSignAdamP variants, presents a promising strategy for boosting the optimization performance of established algorithms. The findings of this study contribute to the advancement of optimizer design in deep learning.
title Adaptive Friction in Deep Learning: Enhancing Optimizers with Sigmoid and Tanh Function
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2408.11839