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.
Figures & tables
ρk=F(xk)−mνk+1(sk+1)F(xk)−F(xk+sk+1)
Algorithm 1 AdaCubic algorithm
AdaCubic Hyperparameters
Hyperparameter
η1
η2
α1
α2
κeasy
ϵm
Assigned Value
0.05
0.75
2.5
0.25
0.01
10−6
Table 1: Universal AdaCubic hyperparameter settings. All hyperparameters in Algorithm 2 are fixed across benchmarks. ϵm denotes a numerical safeguard used in Algorithm 1 .
Task
Dataset
Model
Batch
Epochs
Optimizers
CV
CIFAR-10
ResNet20 / ResNet32
256
500
SGD , Adam , AdaHessian , AdaCubic
CIFAR-100
ResNet18
256
200
SGD , Adam , AdaHessian , AdaCubic
NLU
SST-2 , QNLI , RTE , WNLI
SqueezeBERT
32
15
SGD , AdaHessian , AdaCubic
MRPC , QQP
SqueezeBERT
32
15
SGD , AdaHessian , AdaCubic
STS-B , MNLI
SqueezeBERT
32
15
SGD , AdaHessian , AdaCubic
LM
WikiText-2
RoBERTa / BERT / DistilBERT
8
6
SGD , AdaHessian , AdaCubic
Table 2: Summary of model architectures, training settings, and optimizers used in all experiments.
Optimizer
Task
Dataset(s)
Initial LR
LR Schedule / Tuning
SGD
CV
CIFAR-10 / CIFAR-100
0.1
Step decay (tuned)
Adam
CV
CIFAR-10 / CIFAR-100
10−3
Step decay (tuned)
AdaHessian
CV
CIFAR-10 / CIFAR-100
0.15
Step decay (tuned)
AdaCubic
CV
CIFAR-10 / CIFAR-100
no LR
Universal parameters
SGD
NLU
SST-2 , QNLI , RTE , WNLI
2×10−2
Tuned
SGD
NLU
STS-B
2×10−3
Tuned
Table 3: Summary of LRs used in all experiments. For the CV benchmark, LRs are decayed by a factor of 10 at epochs 80 and 120 on CIFAR-10 , and by a factor of 20 at epochs 60 , 120 , and 160 on CIFAR-100 . For NLU, LM, and CMI benchmarks, no LR decay is applied. For CMI, LR is decayed by a factor of 10 at epochs 80 and 120 . AdaCubic is used with a fixed universal parameter set and does not require LR tuning.
CIFAR-10
CIFAR-100
ResNet20
ResNet32
ResNet18
SGD
88.52 ± 0.24
89.02 ± 0.20
72.62 ± 0.002
Adam
90.26 ± 0.19
91.24 ± 0.20
72.04 ± 0.13
AdaHessian
91.64 ± 0.46
93.15 ± 0.12
72.45 ± 0.16
-
-
72.59 ± 0.271
AdaCubic
91.49 ± 0.46
92.65 ± 0.19
71.81 ± 0.003
Table 4: Accuracy (%) and std of the accuracy measures for ResNet18/20/32 models on CIFAR-10 and CIFAR-100 datasets. Δ reports the gap between the strongest competing optimizer and AdaCubic .
Figure 1: Training loss curve of ResNet20 (top) and ResNet32 (bottom) on CIFAR-10 for Adam , AdaHessian , and AdaCubic optimizers.
Optimizer
RoBERTa
BERT
DistilBERT
SGD
3.547
13.380
6.118
AdaCubic
3.756
5.759
6.565
AdaHessian
4.374
16.151
6.822
Table 6: Perplexity achieved by SGD , AdaHessian , and AdaCubic on wikitext-2 dataset.
Optimizer
RoBERTa
BERT
DistilBERT
SGD
4.345
17.344
8.299
AdaCubic
5.145
14.170
7.334
AdaHessian
7.582
20.851
10.182
Table 7: Perplexity using SGD , AdaHessian , and AdaCubic on the PTB dataset.
Figure 2: Perplexity vs. epochs for RoBERTa , BERT , and DistilBERT models on wikitext-2 dataset.
Figure 3: Perplexity vs. epochs for RoBERTa , BERT , and DistilBERT models on PTB dataset.
AdaCubic
Adam
Native
WhatsApp
YouTube
Native
WhatsApp
YouTube
Fold 0
97.40
96.10
94.59
96.10
93.50
91.9
Fold 1
93.51
93.51
93.24
94.80
90.90
93.24
Fold 2
94.81
92.22
94.59
90.90
88.31
95.94
Fold 3
93.42
93.43
91.89
93.42
94.73
82.43
Fold 4
94.73
88.16
93.24
94.73
88.15
95.94
Table 8: CMI accuracy (%) using ResNet18 .
Figure 4: Cumulative time vs. epochs for SGD , AdaHessian , and AdaCubic for ResNet20 on CIFAR-10 .
Figure 5: Comparison of SGD , AdaHessian , and AdaCubic on ResNet20 and CIFAR-10 . Training loss vs. cumulative time over epochs (Left). Training loss vs. epochs (Right).
SGD
AdaHessian
AdaCubic
Epoch
Time
Epoch
Time
Epoch
Time
Result
83
35.16
81
61.85
55
42.40
Table 9: Execution time in minutes required to reach a target loss threshold when ResNet20 is trained on CIFAR-10 .
Optimizer
Order
Sensitivity
Extra Backward Pass
Time Cost
Memory Footprint
SGD
First
High
No
d
d
Adam
First
High
No
d
3d
AdaHessian
Second
Medium
Yes
\pazocalSd
4d
AdaCubic
Second
Low
Yes
\pazocalSd
2d
Table 10: Comparison of optimization methods used in the experimental evaluation. Recall that d denotes the number of model parameters and \pazocalS the number of random vectors used in the diagonal Hessian approximation.
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Figure 6: Logical connection between key lemmata, theorems, and corollaries throughout Sections 2 , 3 and 4 and Appendices B.1 , B.2 , B.3 , B.4 , B.5 , B.6 , B.7 , B.8 , B.9 , B.10 , B.11 , B.12 and B.13 .
Department of Computer Science & Engineering, Indian Institute of Technology Indore, India. · Leibniz Universität Hannover, Institut für Angewandte Mathematik, Germany. · National Remote Sensing Centre, Indian Space Research Organisation, India.