UGM: A Unified Framework and New Perspectives for Accelerated Gradient Methods in Smooth and Strongly Convex Optimization
Abstract
In this paper, we propose a unified framework for accelerated gradient methods, dubbed UGM, which subsumes a wide range of accelerated and conventional gradient-type methods designed for minimizing -smooth and -strongly convex functions. We demonstrate that the iteration update of the proposed framework can be intrinsically interpreted as a hybrid combination of the heavy-ball method and vanilla gradient descent. This interpretation reveals that classical accelerated gradient methods essentially integrate a conservative gradient descent step into the fast yet unstable heavy-ball dynamics, which achieves a favorable trade-off between acceleration and stability. We further establish a unified convergence analysis using Lyapunov functions. Guided by our analysis, we develop a family of enhanced accelerated gradient algorithms that leverage the inner product relationship between gradient information and iterative variables to optimize iterative updates. Extensive numerical experiments on unconstrained quadratic optimization and logistic regression validate that the proposed algorithms achieve superior performance compared with existing baseline methods under typical structural conditions.
Mathematical Subject Classifications: 15A18, 15A69, 65F15, 90C33
1 Introduction
Let and be constants satisfying , and let be the set of -smooth and -strongly convex functions on . In this paper, we consider the following unconstrained minimization problem
| (1) |
Throughout this paper, we denote as the condition number, as the unique minimizer of , and . Gradient descent (GD) method is a standard algorithm for solving (1). With fixed step size and initial point , the GD method iterates as
| (2) |
A typical choice of the step size that is optimal for maximizing the per-iteration decrease in objective value is , leading to the update
| (3) |
whereas optimizes the decay of the squared optimality gap [25, 21]; these two step sizes yield convergence rates of and , respectively. Moreover, momentum-based accelerated algorithms have been extensively developed to solve the problem (1). The key principle of such methods lies in incorporating historical iterative information alongside the instantaneous gradient to construct the update direction, which enables improved convergence speed. Representative algorithms within this category contain Polyak’s classic Heavy-Ball (HB) accelerated scheme [28], Nesterov’s Fast Gradient Method (FGM) [24, 25], as well as numerous subsequent variants proposed in [7, 13, 17, 33, 27, 35, 32, 10]. While the body of related work is extensive, we focus, throughout this paper, our discussion on the FGM [24, 25], the Strongly Convex Optimized Gradient Method (SC-OGM) [27], the Triple Momentum (TM) method [33], and the Accelerated Over-Relaxation Heavy Ball method (AOR-HB) [35].
Given , , and , the general form of Polyak’s HB method iterates via
| (gHB) |
Setting and yields
| (HB) |
which is theoretically proven optimal for convex quadratic minimization with the convergence rate , since its convergence complexity attains the known lower complexity bounds [23]. However, (HB) is not guaranteed to converge for ; see counterexamples given in [19, 16]. Moreover, it is shown in [16] that for any , the general HB method (gHB) either fails to attain an accelerated convergence rate (i.e., slower than ) over strongly convex quadratic functions, or admits some together with an initialization for which the iteration diverges. Compared with the GD method (2), FGM achieves the faster convergence rates on and on for ; by contrast, convergence of (HB) is guaranteed only for strongly convex quadratic functions. The convergence rate of FGM over is further improved to in [33, 32], which matches the exact lower complexity bound in [12]. Moreover, the idea of accelerating gradient methods via momentum was extended to the proximal point method in [5] and to composite optimization problems in [7, 26]. Furthermore, the fundamental question “Why can FGM achieve accelerated convergence?” has attracted substantial research attention, and its acceleration mechanism has been interpreted from diverse viewpoints; we review several representative interpretations in what follows.
- •
ODE-based interpretation. Su et al. [31] established an approximate equivalence between FGM and a second-order ordinary differential equation (ODE). Based on this ODE model, they found that oscillations in late iterations degrade the algorithm’s convergence. Accordingly, they proposed an adaptive restart strategy for cases where the gradient direction conflicts with the momentum direction. A large body of follow-up research has adopted the ODE perspective to analyze and design accelerated algorithms, including [6, 22, 11, 20, 29, 35].
- •
Linear-coupling-based interpretation. Zhu et al. [3] attributed the acceleration of FGM to the linear coupling of two basic algorithms: GD (3) and mirror descent. The combination integrates their respective strengths to attain accelerated convergence. Later, this linear-coupling framework was extended to design faster and more concise algorithms for packing and covering linear programs [2, 34], semidefinite programming solvers [1], and coordinate descent methods [4].
- •
Dissipativity theory from the control perspective. Lessard et al. [19] modeled FGM as a dynamical system and adopted the integral quadratic constraints (IQC) framework from robust control theory to rigorously analyze its accelerated convergence rate for strongly convex problems. Their work established a unified paradigm for automated algorithm design and performance evaluation. Later, this framework was extended to general convex settings via non-quadratic Lyapunov functions [14].
1.1 Our proposed framework and contributions
Given , and , we initialize . In this paper, we propose the following unified framework, dubbed UGM, for both non-accelerated and accelerated gradient methods
| (4a) | ||||
| (4b) | ||||
| (4c) | ||||
| (4d) | ||||
A more compact yet equivalent formulation of the UGM update (4) is given in (49), which matches the general form of the optimized gradient method presented in [17, 18], [32, Eq. (2)]. Our main contributions are summarized below.
- (i)
We show that UGM can be interpreted as a combination of (HB) and the GD method (3). Moreover, a broad class of popular accelerated and non-accelerated methods is unified within the UGM framework. Hence, accelerated methods such as FGM can be interpreted as integrating a conservative GD step (3) into the fast yet unstable HB method (HB) via an appropriate combination of coefficients, thereby balancing acceleration and convergence. Furthermore, we elucidate the roles of the parameters and : mainly regulates the convergence rate, while controls how closely the method resembles (3). These findings provide useful insights for constructing Lyapunov functions and are discussed in detail in Section 2.
- (ii)
In Section 3, we conduct a unified convergence analysis for a broad class of popular accelerated methods with fixed parameters and under the UGM framework, using intuitively structured Lyapunov functions. The main convergence results are summarized in the fixed-parameter portion of Table 1. Under our analysis framework, we develop new -linear convergence guarantees for FGM, TM, and AOR-HB methods.
- (iii)
Since our analysis indicates that the choice of is, to some extent, independent of , we propose new accelerated UGM algorithms with varying parameters. We also develop adaptive variants, UGM-II and UGM-III, which select based on the inner product to improve the iteration updates. To the best of our knowledge, these adaptive selection mechanisms are new to the literature. Full details are given in Section 4, and the main convergence results are summarized in the varying parameters portion of Table 1.
| Param. | Method | Lyapunov function11 1 In this column, we only present the Lyapunov functions for Q-linear convergence, which satisfy for some . | Rate | Type | Ref. |
| fixed | FGM | — | R22 2 In [25], convergence is established via the estimate sequence , which is lower bounded by , leading to the result . | [25] | |
| (19) with | Q | this work | |||
| SC-OGM | (19) with | Q | [27] | ||
| this work | |||||
| TM method | — | R | [33] | ||
| (19) with | Q | this work | |||
| AOR-HB | see footnote33 3 Here, the Lyapunov function is , where . The authors also provide convergence analysis, but with a more complex Lyapunov function. In contrast, our analysis achieves the same rate using a simple Lyapunov function. | Q | [35] | ||
| (18) with | Q | this work | |||
| varying | ITEM | — | R | [32] | |
| UGM-I | (33) | Q | this work | ||
| UGM-II | |||||
| UGM-III |
1.2 Preliminaries and notation
Throughout the paper, we adopt the following notation. Let be the -dimensional Euclidean space, with inner product and induced norm denoted by and , respectively. A function is said to be -smooth if it is differentiable and satisfies for all . Following [25, Thm. 2.1.5], for convex , -smoothness is equivalent to either of the following statements:
| (5) | ||||
| (6) |
A differentiable function is said to be -strongly convex if
For any , there holds
| (7) |
For any and , the following identity holds:
| (8) |
All algebraic identities and equivalent reformulations in this paper have been verified by manual calculation and symbolic computation using SymPy. The verification notebook is available at https://github.com/dq-kyo/UGM-a-Unified-Framework-for-Gradient-Method. Statements whose proofs rely on these code-verified algebraic identities are marked with footnotes in the text.
2 UGM unifies diverse gradient methods
We now explain the motivation behind UGM (4) from two perspectives.
Perspective 1: Embedding the GD step (3) into (HB).
First, the UGM iteration can be viewed as a combination of (HB) and the GD method (3). To see this, we can rewrite (HB) as
| (9a) | ||||
| (9b) | ||||
| (9c) | ||||
Here , and . This formulation shows that (HB) consists of a large GD step with step size , followed by two moving-average updates; the proof of equivalence between (HB) and (9) is provided in Appendix A. As shown in [19, 16], (HB) fails to converge for general strongly convex and smooth functions. To address this issue, we introduce the standard GD update (3), i.e., , into the second moving-average step (9c), which yields the UGM iteration (4). By properly tuning the combination coefficients for (3) and (HB), existing accelerated methods (e.g., FGM) can be naturally embedded into the UGM framework. Hence, these methods essentially integrate the stable GD step (3) into the fast but possibly divergent scheme (HB), achieving a favorable trade-off between acceleration and convergence guarantee.
Perspective 2: Unifying many existing gradient methods.
Second, the UGM framework unifies a large family of classic accelerated and non-accelerated gradient methods. This unified perspective helps us clarify the roles of parameters and . Such understanding further guides the construction of Lyapunov functions and the design of new algorithms. Below, we list some representative algorithms encompassed by the proposed UGM framework.
- 1.
- 2.
GD with step size : UGM reduces to this scheme when setting and ; see Lemma B.3 for the equivalence proof. Here we also have for all ;
- 3.
- 4.
- 5.
- 6.
Remark 2.1 (The role of and ).
Note that for the HB method (9), GD method (2) with or , and FGM, the associated exactly matches their respective convergence rates. On the other hand, the convergence rates of SC-OGM, AOR-HB, and TM methods approximate their corresponding values. From this observation, we conclude that primarily characterizes the convergence rate. On the other hand, the parameter governs how similar the iteration behaves to the GD method (3). In fact, as can be seen from (4d), taking sufficiently large forces to mimic the GD iteration (3).
3 Unified analysis via Lyapunov functions
In this section, we derive the accelerated convergence rates for all the aforementioned accelerated algorithms via an intuitive unified Lyapunov function framework.
When conducting convergence analysis for gradient-type methods over the class with , it is standard to study two quantities: the objective gap and the squared distance to the optimal solution . Standard proofs for Nesterov’s FGM require rather intricate technical arguments to balance these two quantities.
In our UGM analysis, we instead track the objective gap and the squared distance separately, and then balance these two terms. Our choice to employ draws inspiration from (2) with step sizes and , for which holds for all when regarded as special instances of UGM. We further derive an inductive relation for , motivated by [27] and the inductive relation for in (14).
Lemma 3.1 (Inductive relation for and ).
Proof.
Lemma 3.2 (Inductive relation for ).
For all , the update defined in (4) satisfies
| (17) | ||||
Proof.
Next, we introduce the following Lyapunov functions, which play a central role in our linear convergence analysis:
| (18) | ||||
| (19) |
Here denotes an undetermined parameter serving as a balancing weight between the squared distance term and either the objective gap or . For , we write and for brevity. By combining the inductive bounds for from (14), for from (15), and for from (17), we arrive at the following theorem.
Theorem 3.1.
Proof.
Let . It follows from the definition of in (18) and the condition that
| (22) |
Using (14) to bound the first term on the right-hand side of (22), substituting (17) (with replaced by ) into the second term, and rearranging all terms immediately yields the desired result (20). Similarly, (21) can be derived using (19), (15) and (17). ∎
Remark 3.1.
Remark 3.2.
Equations (18) and (19) each define a Lyapunov function by combining a function-value-based error term with the squared distance . Specifically, (18) adopts the standard objective gap , whereas (19) employs the modified residual term . Note that the nonnegativity of follows directly from the -smoothness of . Alternatively, one may replace the function-value error term with its strongly convex counterpart , or adopt the tighter inequality as studied in [35, 32]. Nonnegativity of these two expressions hinges on the -strong convexity of . Moreover, both expressions explicitly couple the objective gap and squared distance, which conflicts with our aim to decouple these two quantities.
Based on (20) and (21), we subsequently present new convergence analysis for FGM, SC-OGM, TM method and AOR-HB, using the unified Lyapunov functions.
- (i)
FGM [24, 25]: As established in the analysis of Section 2, FGM is a special case of UGM with and . Under this parameter setup, the identities and hold; these equalities have been verified both analytically and via the symbolic computation script check_proof.ipynb. Set and . Substituting these into (21), we obtain
From this inequality, we conclude that Nesterov’s FGM attains linear convergence with convergence rate .
- (ii)
SC-OGM [27]: SC-OGM is a special case of UGM with , and . With these parameter choices, the equalities , and hold. Again, these identities have been verified both analytically and using the symbolic computation script check_proof.ipynb. Set and . Substituting these into (21), we obtain
Hence, SC-OGM achieves linear convergence with rate .
- (iii)
TM method [33]: The TM method is a special instance of UGM with and . Then, using and applying the weighted AM-GM inequality, we obtain
Furthermore, invoking (7) with and , and using , we derive
For convenience, we define
(23) (24) If inequality
(25) holds, then from (21) we obtain
(26) where is defined in (23). Next, we show that there exists a pair satisfying (25) and such that . We focus on the nontrivial case . First, we set to minimize the expression . Since , it follows that . Substituting the variable transformation into (24), along with the relations and , and performing elementary algebraic computations, we obtain
(27) where and . Next, we choose such that , i.e., . From Lemma C.1, we have . Hence, via elementary algebraic calculations, we derive . Since and , we get , which verifies that the condition (25) holds for . Substituting into (26) and noting that , we derive the linear convergence result . Finally, we have
(28) where the Taylor expansion is adopted for the asymptotic approximation. Accordingly, we establish a Q-linear convergence rate of , while the original result in [33] achieves an R-linear convergence rate of . Note that the accelerated convergence behavior is retained. Finally, we note that all algebraic identities: the expression for in (23)-(24), the form of in (27), and the alternative representation of in (28) have been formally verified in the script check_proof.ipynb.
- (iv)
AOR-HB [35]: From the discussion in Section 2, AOR-HB is a special instance of UGM with parameters and . Lemma B.7 guarantees and . Let and . One can manually verify the relation . Alternatively, these equalities are also validated via symbolic computation in check_proof.ipynb. Hence, we have
(29) For the given , the following equivalence holds:
This equivalence is verified via symbolic computation in check_proof.ipynb. Given , the polynomial on the right-hand side is obviously strictly positive, which directly yields . Recall our choice . We thus have . Meanwhile, -strongly convexity of delivers the bound . This observation guarantees nonnegativity of the second term appearing on the right-hand side of (20). Combining the relations (20), (29), alongside the identity , we deduce that
Hence, we have established the linear convergence result . Furthermore, one may check that our selected satisfies . In comparison, the original analysis of the AOR-HB algorithm in [35]44 4 The authors also mention a rate in their slides, yet the corresponding Lyapunov function is far more intricate than the one constructed here. establishes a linear convergence rate of .
4 UGM with varying parameters and
Let . This section investigates the UGM framework (4) equipped with iteration-dependent parameters and :
| (30) |
In particular, Subsection 4.1 presents simple iteration-varying parameter choices, while Subsections 4.2–4.3 further develop adaptive, problem-dependent parameter variants. First, using analogous arguments based on (15) and (17), we derive the following iterate estimates for the updates in (30).
Lemma 4.1.
Theorem 4.1.
Let and satisfy for all . Define the Lyapunov function by
| (33) |
Then, for the iterates generated by (30), it holds for all that
| (34) | ||||
Proof.
4.1 UGM-I: Parameter rule
Consider the four terms on the right-hand side of (34). Among them and aside from their coefficients, the only term with an undetermined sign is “”, whose coefficient is . We therefore set the coefficient preceding this term to zero, yielding , and select the parameters involved in the iterative scheme (30) and Lyapunov function (33) as follows:
UGM-I. Choose and any sequence such that , , and for all , and (36a) (36b) (36c) (36d)Theorem 4.2.
Proof.
We next provide concrete parameter choices that satisfy the conditions in (36).
To verify that the parameters in (37) satisfy the constraints in (36), we first introduce a useful lemma.
Lemma 4.2.
Let be a sequence satisfying for all , with . If , then for all , we have
| (38) |
Moreover, for , we have .
Proof.
For , the inequality trivially holds due to . Furthermore, solving the recurrence relation for all yields the closed-form expression . The subsequent proof proceeds by considering two separate cases: and .
- (i)
If , the sequence is decreasing and converges to , which implies for all . The condition ensures (see Lemma C.2 for a rigorous verification). Consequently, holds for all . In addition, since the function is monotonically increasing and , we further obtain .
- (ii)
If , the sequence is increasing and converges to , so for all . This further implies and .
Thus, we have shown that and for all . ∎
Proof.
First, direct computation using (37c) yields , which verifies condition (36b). Second, it is straightforward to verify from (37) that and , which validates (36a) and (36d). Finally, using the foregoing relations and the update rules in (37), one can directly derive that
where the inequality follows from (38). This verifies condition (36c). ∎
4.2 UGM-II: -aware parameter rule
In UGM-I, we choose so that the coefficient of vanishes, since this inner-product term carries an indefinite sign. A natural question arises: if, under suitable conditions, we can determine the sign of , do we gain greater flexibility in selecting ? Suppose for all . Substituting and into (5) and invoking (30), we can derive
| (39) |
Hence, the term is regarded as a “good” term whenever it is nonnegative, since (39) then guarantees a reduction in the objective function value.
Below, we derive two estimates for . First, we have
| (40) |
Second, rearranging yields . Substituting this identity into (40) and rearranging the terms, we obtain
| (41) | ||||
Lemma 4.4.
Let be fixed constants. Suppose for all . Then the following hold:
- (i)
If , then ;
- (ii)
If , then .
Proof.
In view of (34), if , then it is desirable to choose satisfying . Furthermore, (39) suggests that a smaller may be preferable, as it tends to yield a larger decrease in the objective function value. Conversely, if , then it is desirable to choose so that the reverse inequality holds. On the other hand, the parameters and are selected following the scheme UGM-I (explicit) presented in (37). Recall that from (37) and Lemma 4.2, for all we have
| (42) |
Then, we have the following lemma.
Lemma 4.5.
Choose and by following the scheme UGM-I (explicit) presented in (37). Assume be fixed constants satisfying and . For , define via
| (43) |
Then, we have for all .
Proof.
Based on Lemma 4.5, we are now ready to present our -aware parameter rule as follows.
UGM-II: -aware parameter rule. Choose parameters such that , , , . For all , compute , and via (43), (37b) and (37c), respectively.Theorem 4.3.
For the iterations generated by (30) with the parameters specified in UGM-II, we have for all .
4.3 UGM-III: - and -aware parameter rule
We further note the term on the right-hand side of (34) also depends implicitly on the choice of , since . In this subsection, we select by relating it to this previously neglected term; specifically, we seek to find such that
| (44) |
Next, we focus on the case .
Lemma 4.6.
Proof.
Fix any . Recall that we always have and the identity . If , then , and hence , which implies that (44) holds. Next, we assume . From the definitions of , , , the relation , and the left-hand side inequality in (41), we obtain
where is a quadratic function of . Note that , and the axis of symmetry of is . Next, we split the discussion into three mutually disjoint cases.
- (i)
- (ii)
. In this case, the coefficient of the quadratic term in is positive, and straightforward calculations show that defined in (45) is the larger root of the quadratic equation . Since and , we have . Combined with , this implies that the axis of symmetry of , namely , is bounded above . Recalling , we further obtain , and for all , so (44) holds;
- (iii)
. In this case, and straightforward calculations yields , which solves the linear equation . Hence, for all , and thus (44) follows.
Thus, we complete the proof. ∎
Based on Lemma 4.6 and in analogy with (43), we now propose the following more flexible update rule for selecting :
| (46) |
We are now ready to present our - and -aware parameter rule as follows.
UGM-III: & -aware parameter rule. Choose parameters such that , , , . For all , compute , and via (46), (37b) and (37c), respectively.Theorem 4.4.
For the iterations generated by (30) with the parameters specified in UGM-III, we have for all .
5 Numerical experiments
This section presents three groups of numerical experiments to comprehensively validate the performance and practical merits of the proposed algorithms. First, we test the proposed methods on unconstrained quadratic optimization problems to explore their favorable application scenarios and analyze the functional role of the parameter . Second, we evaluate the algorithm performance on the logistic regression task, verifying that our methods achieve competitive results in practical real-world learning problems. Third, we conduct experiments on softmax regression to investigate whether the iteration efficiency advantages of UGM-II and UGM-III can be converted into CPU time savings in distributed environments, leveraging cached-logit computation and scalar aggregation techniques. All experiments are implemented in Python 3.11.0 and run on a personal computer equipped with an AMD Ryzen 7 5800H processor, Radeon Graphics, and 16 GB of RAM.
5.1 Unconstrained quadratic optimization
For the first experiment, we consider the unconstrained optimization of quadratic functions. Let be a symmetric positive definite matrix and . Consider minimizing over . We evaluate four accelerated gradient methods in this setting, including the classical FGM algorithm [25] and our three proposed schemes, namely UGM-I, UGM-II, and UGM-III, which are developed in Section 4. To simplify the experimental setup, we construct and in a controlled manner. We first randomly generate an optimal solution whose entries are uniformly sampled from , and then set . The matrix is defined as the diagonal matrix with ordered eigenvalues . For a fixed dimension , we set the extreme eigenvalues to and , which ensures that is -strongly convex and -smooth. The intermediate eigenvalues are sampled from five predefined distributions, rescaled to the interval , and finally sorted in ascending order. The detailed configurations of the five eigenvalue distributions are presented as follows.
- (Case 1)
Geometric progression: set for with .
- (Case 2)
U-shaped distribution: generate samples from .
- (Case 3)
Central-concentrated distribution: generate samples from .
- (Case 4)
Lower-end skewed distribution: generate samples from .
- (Case 5)
Upper-end skewed distribution: generate samples from .
For Cases 2-5, all samples are linearly rescaled to the interval via the mapping , and the resulting values are then sorted in ascending order to form the intermediate eigenvalues. In this experiment, we set . The five eigenvalue distributions are visualized in Figure 1.
We consider five combinations of strong convexity and smoothness parameters : , , , , and . These configurations produce condition numbers ranging from to . To ensure fair comparison across different conditioning levels, the maximum iteration number is adaptively determined according to . Specifically, we specify three anchor points and adopt linear interpolation on the scale to compute the iteration budget for each tested condition number.
For the proposed UGMs, we adopt the following parameter settings. We set for all the three UGMs, for UGM-I and for both UGM-II and UGM-III. The lower bound parameter is set to for and otherwise, which satisfies the theoretical constraint . Furthermore, we set and define the upper bound parameter . All compared algorithms are initialized at the origin.
Comparison results.
Each configuration is run five times with independent random seeds. For each eigenvalue distribution and each tested pair, we report the objective suboptimality against iteration count and CPU time (s), alongside the evolution of the extrapolation parameter for UGM-II and UGM-III. Numerical results for Cases 1–5 are presented respectively in Figures 2–6.
In terms of iteration counts, UGM‑II and UGM‑III attain slightly lower suboptimality than FGM. However, since UGM‑II and UGM‑III require one extra gradient evaluation per iteration, they do not offer a clear advantage in CPU time in most test cases. Some exceptions include Figure 4 (column (c)), and Figure 6 (column (a), (c), (d)), where the faster convergence of UGM‑II and UGM‑III in terms of iterations offsets the additional gradient evaluations, rendering them competitive in CPU‑time performance. Across all test cases, the proposed UGMs achieve performance comparable to FGM.
The role of in UGM‑I.
We next investigate the effect of the parameter on the performance of UGM‑I. We select four representative values of from the interval . The smallest value is taken at the lower bound endpoint, while the largest is set to to stay safely below the strict upper limit. The remaining two values are equally spaced between these two extremes. We compare the four resulting variants of UGM‑I on two contrasting eigenvalue distributions, Case 1 and Case 5, using three configurations: , , and . The numerical results are presented in Figure 7.
It is clear from the results in Figure 7 that the parameter dictates the convergence speed of UGM‑I: the smaller the value of , the steeper the decrease of the suboptimality .
5.2 Logistic regression
In this subsection, we consider the regularized logistic regression problem
where is the dataset, denotes the sigmoid function, is the number of samples, and is the regularization parameter. For this problem, the gradient of is given by . Consequently, is -smooth with Lipschitz constant , where and denotes the largest eigenvalue of . In this experiment, all competing algorithms are run for iterations. We test on the mushrooms dataset from LIBSVM [9], which has sample size and feature dimension . The Lipschitz constant is , and the regularization parameter is set to . Consistent with the experiments in Section 5.1, every algorithm is initialized at the origin. For the proposed UGMs, the parameters are chosen as follows. We take , matching the convergence‑rate parameter of the FGM method. We further set for UGM‑I and for UGM‑II and UGM‑III. We then let , , and .
In Figure 8, we plot the objective suboptimality against iteration count and CPU time (in seconds), along with the evolution of the extrapolation parameter for UGM‑II and UGM‑III. From Figures 8(a) and 8(b), we observe that both UGM‑II and UGM‑III consistently outperform FGM in terms of both iteration efficiency and computational cost on the tested logistic regression instance. This performance gain arises from the adaptive updating rule of the extrapolation parameter . As shown in Figure 8(c), eventually converges to its lower bound after a certain number of iterations, which suppresses parameter oscillations and substantially accelerates the decay of objective suboptimality.
5.3 Softmax regression
The numerical results above demonstrate that UGM‑II and UGM‑III achieve faster convergence in terms of iteration count, yet this benefit does not always yield a corresponding wall‑clock improvement, owing to the extra gradient evaluation required at each iteration. In this subsection, we examine whether their iteration‑wise advantage can be converted into better computational efficiency when the inner‑product scalar is computed without constructing an additional full‑gradient vector. To this end, we consider the following -regularized multiclass logistic regression problem, commonly referred to as softmax regression:
| (47) |
Here denotes the score of sample for class , is the score corresponding to its true class label , denotes the Frobenius norm, and . Given a prescribed condition number , we set , with . With this choice, the objective function in (47) is -smooth with Lipschitz constant and -strongly convex. Its resulting condition number is therefore .
We investigate two ways to convert UGM‑II and UGM‑III’s iteration‑count advantage into CPU‑time savings. The first uses cached logits to evaluate without computing an additional full gradient. The second employs scalar aggregation in a distributed setting, replacing communication of high‑dimensional gradient matrices with a single scalar transmitted from each worker. Both experiments use the Digits dataset from sklearn.datasets.load_digits, which contains grayscale images belonging to digit classes. Each image is flattened into a 64‑dimensional vector with pixel values in . We standardize the original features and use all samples to construct the finite‑sum problem. The two experiments differ only in the nonlinear feature map subsequently applied to the standardized data. Furthermore, the UGM variants in these experiments adopt the same parameter settings for , , , , and as specified in Section 5.2.
Scalar evaluation using cached logits.
Let denote the matrix of softmax probabilities corresponding to the logits , with entries
Then, the scalar required by the update can be expressed as
| (48) |
where is the one-hot encoding matrix of the class labels. Here, for matrix variables, denotes the trace inner product. The quantity can be computed directly from the cached logits and . Thus, this directional implementation only evaluates the Frobenius inner products in (48), without explicitly constructing the full‑gradient term . The directional and full‑gradient implementations produce identical iterates up to floating‑point round‑off error, yet they incur different per‑iteration computational costs.
Our first experiment maps the handwritten‑digit samples to random Fourier features (Digits‑RFF) and considers classes.
The experimental results are presented in Figure 9. In our tests, UGM‑II and UGM‑III attain the stopping criterion in and iterations, respectively, compared with iterations for FGM and iterations for UGM‑I. With the cached‑logit directional implementation, their CPU times are and seconds. Compared with FGM, these results yield speedups of and . These findings demonstrate that the cached‑logit implementation successfully converts the iteration‑count advantages of UGM‑II and UGM‑III into substantial CPU‑time savings.
Digits‑Poly2 with scalar aggregation.
Our second experiment expands the same samples into standardized degree‑2 polynomial features (Digits‑Poly2), again with classes, to further evaluate the benefits of scalar aggregation in a distributed setting. The samples are partitioned across four logical workers. For the directional computation, each worker returns a single double‑precision scalar, resulting in a total communication payload of bytes. In contrast, full‑gradient aggregation requires transmitting four matrices, corresponding to bytes. Therefore, scalar aggregation reduces the communication payload by a factor of .
The experimental results are presented in Figure 10. In our tests, UGM‑II and UGM‑III attain the stopping criterion in and iterations, while FGM and UGM‑I require and iterations. With scalar aggregation across the distributed workers, their CPU times are reduced to and seconds, whereas FGM requires seconds. Relative to FGM, this yields speedups of and , respectively. These findings demonstrate that scalar aggregation can convert the lower iteration counts of UGM‑II and UGM‑III into measurable computational gains in distributed settings.
Moreover, the panels (last plots in Figures 9 and 10) show a brief initial adaptive transient phase, after which the algorithm stably selects the lower bound . Therefore, the observed improvement in iteration‑count performance does not arise merely from employing a fixed momentum parameter. Instead, it stems from the branch‑aware update rule, while cached‑logit evaluation and scalar aggregation reduce the overhead of this extra selection information sufficiently to achieve an end‑to‑end runtime advantage.
6 Concluding remarks
In this paper, we study a unified first‑order method that encompasses a broad family of accelerated and non‑accelerated algorithms for minimizing . We show that the proposed method can be naturally interpreted as a combination of the heavy‑ball method and gradient descent. This reveals that classical acceleration schemes effectively incorporate a conservative gradient‑descent step into the fast yet potentially divergent heavy‑ball dynamics to balance acceleration We then present a unified convergence analysis built upon a Lyapunov‑function framework, which further motivates the design of problem‑dependent accelerated algorithms and their variants. Applications to unconstrained quadratic optimization, logistic regression, and softmax regression demonstrate that the proposed methods can outperform existing approaches under specific structural conditions.
The following directions are worthy of further investigation.
- (i)
One interesting direction is to estimate the sign of without relying on Lemma 4.4, which requires either additional full-gradient evaluations or the corresponding directional computations. Directly determining the sign of this inner product would reduce computational overhead and improve practicality. Such a technique is consistent with our unified Lyapunov framework and may support the design of additional accelerated optimization variants.
- (ii)
Estimating the -smoothness and -strong convexity parameters without prior problem information remains challenging [8, 36]. It is worth investigating whether adaptive iteration-dependent parameters and can be incorporated into the UGM framework (4) to achieve accelerated convergence, following the parameter strategy of Nesterov’s FGM [25]. The corresponding update scheme is provided below:
- (iii)
We note that if , any negative satisfies the condition (44), which allows negative momentum while maintaining convergence guarantees. Previous studies have shown that negative momentum may cause oscillation and instability in minimax optimization [15, 30]. Examining its behavior under the current algorithmic setting constitutes another interesting research direction.
References
- [ALO16a] (2016) Using optimization to obtain a width-independent, parallel, simpler, and faster positive sdp solver. In Proceedings of the twenty-seventh annual ACM-SIAM symposium on Discrete algorithms, pp. 1824–1831. Cited by: 2nd item.
- [AO15] (2015) Nearly-linear time positive lp solver with faster convergence rate. In STOC, pp. 229–236. Cited by: 2nd item.
- [AO17] (2017) Linear coupling: an ultimate unification of gradient and mirror descent. In ITCS, pp. 3:1–3:22. Cited by: 2nd item.
- [AQR+16b] (2016) Even faster accelerated coordinate descent using non-uniform sampling. In ICML, pp. 1110–1119. Cited by: 2nd item.
- [AA01] (2001) An inertial proximal method for maximal monotone operators via discretization of a nonlinear oscillator with damping. Set-Valued Anal. 9 (1-2), pp. 3–11. Note: Wellposedness in optimization and related topics (Gargnano, 1999) External Links: ISSN 0927-6947,1572-932X, Document, MathReview Entry Cited by: §1.
- [ACP+18] (2018) Fast convergence of inertial dynamics and algorithms with asymptotic vanishing viscosity. Math. Program. 168 (1), pp. 123–175. Cited by: 1st item.
- [BT09] (2009) A fast iterative shrinkage-thresholding algorithm for linear inverse problems. SIAM J. Imaging Sci. 2 (1), pp. 183–202. External Links: Document, MathReview Entry Cited by: §1, §1.
- [CLW25] (2025) Adaptive acceleration without strong convexity priors or restarts. arXiv preprint arXiv:2506.13033. Cited by: item (ii).
- [CL11] (2011) LIBSVM: a library for support vector machines. ACM Trans. Intell. Syst. Technol. 2 (3), pp. 27. External Links: ISSN 2157-6904, Link, Document Cited by: §5.2.
- [CL19] (2019) First order optimization methods based on hessian-driven nesterov accelerated gradient flow. arXiv preprint arXiv:1912.09276. Cited by: §1.
- [CL21] (2021) A unified convergence analysis of first order convex optimization methods via strong lyapunov functions. arXiv preprint arXiv:2108.00132. Cited by: 1st item.
- [DT22] (2022) On the oracle complexity of smooth strongly convex minimization. Journal of Complexity 68, pp. 101590. Cited by: §1.
- [DT14] (2014) Performance of first-order methods for smooth convex minimization: a novel approach. Math. Program. 145 (1), pp. 451–482. Cited by: §1.
- [FRM+18] (2018) Analysis of optimization algorithms via integral quadratic constraints: nonstrongly convex problems. SIAM J. Optim. 28 (3), pp. 2654–2689. Cited by: 3rd item.
- [GHP+19] (2019) Negative momentum for improved game dynamics. In AISTATS, pp. 1802–1811. Cited by: item (iii).
- [GTD25] (2025) Provable non-accelerations of the heavy-ball method: b. goujaud et al.. Math. Program., pp. 1–59. Cited by: §1, §2.
- [KF16] (2016) Optimized first-order methods for smooth convex minimization. Math. Program. 159 (1), pp. 81–107. Cited by: §1.1, §1.
- [KF18] (2018) Generalizing the optimized gradient method for smooth convex minimization. SIAM J. Optim. 28 (2), pp. 1920–1950. Cited by: §1.1.
- [LRP16] (2016) Analysis and design of optimization algorithms via integral quadratic constraints. SIAM J. Optim. 26 (1), pp. 57–95. Cited by: 3rd item, §1, §2.
- [LC22] (2022) From differential equation solvers to accelerated first-order methods for convex optimization. Math. Program. 195 (1), pp. 735–781. Cited by: 1st item.
- [MM20] (2020) Adaptive gradient descent without descent. In ICML, pp. 6702–6712. Cited by: §1.
- [MJ19] (2019) A dynamical systems perspective on nesterov acceleration. In ICML, pp. 4656–4662. Cited by: 1st item.
- [NEM95] (1995) Information-based complexity of convex programming. Lecture notes 834. Cited by: §1.
- [NES83] (1983) A method for solving the convex programming problem with convergence rate . Dokl. Akad. Nauk SSSR 269 (3), pp. 543–547. External Links: ISSN 0002-3264, MathReview (R. Şerban) Cited by: §1, item (i).
- [NES04] (2004) Introductory lectures on convex optimization. Applied Optimization, Vol. 87, Kluwer Academic Publishers, Boston, MA. Note: A basic course External Links: MathReview Entry Cited by: §1.2, Table 1, §1, item 3, item (i), §5.1, item (ii), footnote 2.
- [NES13] (2013) Gradient methods for minimizing composite functions. Math. Program. 140 (1), pp. 125–161. External Links: ISSN 0025-5610,1436-4646, Document, MathReview (Dana Simian) Cited by: §1.
- [PPR23] (2023) Factor- acceleration of accelerated gradient methods. Applied Mathematics & Optimization 88 (3), pp. 77. Cited by: Table 1, §1, item 5, item (ii), §3.
- [POL64] (1964) Some methods of speeding up the convergence of iteration methods. Ussr computational mathematics and mathematical physics 4 (5), pp. 1–17. Cited by: §1.
- [SDJ+22] (2022) Understanding the acceleration phenomenon via high-resolution differential equations. Math. Program. 195 (1), pp. 79–148. Cited by: 1st item.
- [SWA26] (2026) Negative momentum for convex-concave optimization. arXiv preprint arXiv:2604.17145. Cited by: item (iii).
- [SBC16] (2016) A differential equation for modeling nesterov’s accelerated gradient method: theory and insights. JMLR 17 (153), pp. 1–43. Cited by: 1st item.
- [TD23] (2023) An optimal gradient method for smooth strongly convex minimization. Math. Program. 199 (1), pp. 557–594. Cited by: §1.1, Table 1, §1, §1, Remark 3.2.
- [VFL17] (2017) The fastest known globally convergent first-order method for minimizing strongly convex functions. IEEE Control Systems Letters 2 (1), pp. 49–54. Cited by: Table 1, §1, §1, item 4, item (iii), item (iii).
- [WRM15] (2015) Unified acceleration method for packing and covering problems via diameter reduction. arXiv preprint arXiv:1508.02439. Cited by: 2nd item.
- [WC25] (2025) Accelerated over-relaxation heavy-ball method: achieving global accelerated convergence with broad generalization. In ICLR, Cited by: 1st item, Table 1, §1, item 6, item (iv), item (iv), Remark 3.2.
- [XC26] (2026) Adaptive accelerated gradient descent methods for convex optimization. arXiv preprint arXiv:2601.19013. Cited by: item (ii).
Appendix A Proof on the equivalence between (HB) and (9)
Appendix B Reformulation of several algorithms
In this section, we present equivalent formulations of several well‑known algorithms within the UGM framework (4). Note that all cumbersome identities used in the proofs of Lemmas B.4, B.5, B.6 and B.7 have been validated by hand-written derivation and symbolic computation, where check_proof.ipynb is used exclusively for the symbolic-computation verification.
Lemma B.1.
Proof.
First, (4a) is the same as (49a), from which we obtain . Substituting this relation into (4b) yields
| (50) |
Additionally, from (4d), we have for all . Substituting this relation with and into (4c) and using (50), we obtain
By rearranging the above equality to explicitly solve for , we obtain (49b), which establishes the desired equivalence. ∎
B.1 Gradient descent methods
Lemma B.2.
Proof.
Let be the sequence generated by UGM (4) with initialization and parameters and arbitrary . We prove by induction that for all . Assume that for some . Then, from (4c), (4b) and (4a), we obtain
where the second “” follows from . By plugging into (4d), we obtain . Hence, . The proof is completed by induction. ∎
Lemma B.3.
Proof.
Since and , (4d) yields for all . Hence, for every , we obtain
where the final equality follows from , which holds with . This completes the proof. ∎
B.2 Nesterov’s FGM
Lemma B.4.
Proof.
To prove the equivalence between (4) and (10), we only need to verify that the update in (4d) reduces to that given in (10). Utilizing the given values of and , the following identities hold:
| (51) |
We have validated these equalities through manual derivation, while the symbolic‑computation check is also implemented in the script check_proof.ipynb. It then follows from (49b) and (51) that
This exactly matches the update of in (10), which completes the proof. ∎
B.3 TM method
Lemma B.5.
Proof.
To begin with, we note that all cumbersome identities used in this proof have been checked by manual derivation, while symbolic‑computation verification is carried out in check_proof.ipynb.
First, we establish the equivalence between (11b) and (4d). To prove the equivalence, it suffices to show that from both (11b) and (4d) satisfies
| (52) |
where .
- (i)
Consider generated by the TM method (11). From (11b), the left-hand side of (52) is equal to
(53) On the other hand, from the parameter settings of (11), we obtain and (11b) yields . We then derive
where the third “” follows from . Thus, we obtain . Using this relation together with , we compute the two terms on the right-hand side of (52) as follows:
(54a) (54b) Utilizing again, the following identities can be verified:
(55) Then, substituting (54) into the right-hand side of (52) and using the identities in (55), we derive
(56) By comparing (53) and (56), we confirm that (52) is satisfied by (11).
- (ii)
This completes the proof of the equivalence between (11b) and (4d).
We next establish the equivalence between (11c) and (4c) by verifying that from both (11c) and (4c) satisfies
| (57) |
- (i)
Consider generated by the TM method (11). Subtracting each‑hand‑side of (54b) from the corresponding side of (53) and utilizing alongside elementary algebraic manipulations, we arrive at . Using (11b) and (11c), together with this relation, we further derive
where the last equality follows from the parameter settings of and in (11) and elementary computations. This confirms (57).
- (ii)
This completes the proof of the equivalence between (11c) and (4c). ∎
B.4 SC-OGM
Lemma B.6.
Proof.
To begin with, we note that all cumbersome identities used in this proof have been checked by manual derivation, while symbolic‑computation verification is carried out in check_proof.ipynb.
To establish this equivalence, it suffices to verify that the update for given in (4d) satisfies
| (58) |
which exactly matches the iteration rule for in (12). With the given parameters , , , and noting that , we can verify the following identities:
It then follows from (49b) that , which implies (58) immediately. ∎
B.5 AOR-HB
Lemma B.7.
Proof.
To begin with, we note that all cumbersome identities used in this proof have been checked by manual derivation, while symbolic‑computation verification is carried out in check_proof.ipynb. To prove the equivalence, we only need to verify that the update for from both (4d) and (13) satisfies
| (59) |
where . Using the given values for and , it can be verified that
Hence, for the UGM framework (4), (59) is implied by (49b), together with the above identities. On the other hand, for the AOR-HB scheme (13), noting that , we obtain
Here, elementary computations are involved in deriving the last equality. This confirms (59) and completes the proof of the equivalence.
To complete the proof for the remaining claims of this lemma, we set , which yields . Direct computation yields . Given , we obtain . Hence, holds for every . Accordingly, is strictly-increasing over , and . Finally, holds trivially since . ∎
Appendix C Supplementary proofs for two inequalities
In the analysis for the TM-method in Section 3, the following inequality (60) is used, and we now provide its proof.
Lemma C.1.
Let as introduced right below (27). Then, for any , we have
| (60) |
Proof.
First, we show that for any , there holds
| (61) |
The left-hand-side inequality in (61) holds trivially since for . To verify the right-hand-side inequality, it is sufficient to show that for , where . By noting that for , one can readily show that is equivalent to , and the latter inequality obviously holds since . This confirms that (61) holds for any .
The following inequality is employed in the proof of Lemma 4.2, and we now present its proof.
Lemma C.2.
If satisfies , then .
Proof.
Since , we have . By squaring both sides, it is straightforward to show that the latter inequality is equivalent to . Subtracting from both-hand sides yields , which establishes the desired inequality. ∎