🤖 AI Summary
This work addresses the limitations of conventional adaptive optimizers, which often rely heavily on hyperparameter tuning and struggle to simultaneously achieve strong convergence and generalization. The authors propose a scalable adaptive cubic-regularized optimizer that dynamically adjusts the regularization strength in Newton-type updates by solving an auxiliary optimization problem with a cubic constraint. To mitigate computational costs, the method leverages the Hutchinson estimator for efficient approximation of the Hessian matrix. As the first successful application of cubic regularization to large-scale deep learning, the proposed optimizer guarantees local convergence without requiring meticulous hyperparameter tuning. Empirical results across computer vision, natural language processing, and signal processing benchmarks demonstrate that, with a fixed set of hyperparameters, it consistently matches or outperforms state-of-the-art optimizers.
📝 Abstract
A novel regularization technique, AdaCubic, is proposed that adapts the weight of the cubic term. The heart of AdaCubic is an auxiliary optimization problem with cubic constraints that dynamically adjusts the weight of the cubic term in Newton's cubic regularized method. We use Hutchinson's method to approximate the Hessian matrix, thereby reducing computational cost. We demonstrate that AdaCubic inherits the cubically regularized Newton method's local convergence guarantees. Our experiments in Computer Vision, Natural Language Processing, and Signal Processing tasks demonstrate that AdaCubic outperforms or competes with several widely used optimizers. Unlike other adaptive algorithms that require hyperparameter fine-tuning, AdaCubic is evaluated with a fixed set of hyperparameters, rendering it a highly attractive optimizer in settings where fine-tuning is infeasible. This makes AdaCubic an attractive option for researchers and practitioners alike. To our knowledge, AdaCubic is the first optimizer to leverage cubic regularization in scalable deep learning applications.