How to define the cost in the embedding
Ronny Bergmann
Specifying a cost function $f\colon \mathcal M \to \mathbb R$ on a manifold is usually the model one starts with. Specifying its gradient $\operatorname{grad} f\colon\mathcal M \to T\mathcal M$, or more precisely $\operatorname{grad}f(p) \in T_p\mathcal M$, and eventually a Hessian $\operatorname{Hess} f\colon T_p\mathcal M \to T_p\mathcal M$ are then necessary to perform optimization. Since these might be challenging to compute, especially when manifolds and differential geometry are not the main area of a user – easier to use methods might be welcome.
This tutorial discusses how to specify $f$ in the embedding as $\tilde f$, maybe only locally around the manifold, and use the Euclidean gradient $∇ \tilde f$ and Hessian $∇^2 \tilde f$ within Manopt.jl
.
For the theoretical background see convert an Euclidean to an Riemannian Gradient, or Section 4.7 of [Bou23] for the gradient part or Section 5.11 as well as [Ngu23] for the background on converting Hessians.
Here we use the Examples 9.40 and 9.49 of [Bou23] and compare the different methods, one can call the solver, depending on which gradient and/or Hessian one provides.
using Manifolds, Manopt, ManifoldDiff
using LinearAlgebra, Random, Colors, Plots
Random.seed!(123)
We consider the cost function on the Grassmann
manifold given by
n = 5
k = 2
M = Grassmann(5,2)
A = Symmetric(rand(n,n));
f(M, p) = 1 / 2 * tr(p' * A * p)
Note that this implementation is already also a valid implementation / continuation of $f$ into the (lifted) embedding of the Grassmann manifold. In the implementation we can use f
for both the Euclidean $\tilde f$ and the Grassmann case $f$.
Its Euclidean gradient $\nabla f$ and Hessian $\nabla^2f$ are easy to compute as
∇f(M, p) = A * p
∇²f(M,p,X) = A*X
On the other hand, from the aforementioned Example 9.49 we can also state the Riemannian gradient and Hessian for comparison as
grad_f(M, p) = A * p  p * (p' * A * p)
Hess_f(M, p, X) = A * X  p * p' * A * X  X * p' * A * p
We can check that these are the correct at least numerically by calling the check_gradient
check_gradient(M, f, grad_f; plot=true)
and the check_Hessian
, which requires a bit more tolerance in its linearity check
check_Hessian(M, f, grad_f, Hess_f; plot=true, throw_error=true, atol=1e15)
While they look reasonable here and were already derived – for the general case this derivation might be more complicated.
Luckily there exist two functions in ManifoldDiff.jl
that are implemented for several manifolds from Manifolds.jl
, namely riemannian_gradient
(M, p, eG)
that converts a Riemannian gradient eG=
$\nabla \tilde f(p)$ into a the Riemannain one $\operatorname{grad} f(p)$ and riemannian_Hessian
(M, p, eG, eH, X)
which converts the Euclidean Hessian eH=
$\nabla^2 \tilde f(p)[X]$ into $\operatorname{Hess} f(p)[X]$, where we also require the Euclidean gradient eG=
$\nabla \tilde f(p)$.
So we can define
grad2_f(M, p) = riemannian_gradient(M, p, ∇f(get_embedding(M), embed(M, p)))
where only formally we here call embed(M,p)
before passing p
to the Euclidean gradient, though here (for the Grassmann manifold with Stiefel representation) the embedding function is the identity.
Similarly for the Hessian, where in our example the embeddings of both the points and tangent vectors are the identity.
function Hess2_f(M, p, X)
return riemannian_Hessian(
M,
p,
∇f(get_embedding(M), embed(M, p)),
∇²f(get_embedding(M), embed(M, p), embed(M, p, X)),
X
)
end
And we can again check these numerically,
check_gradient(M, f, grad2_f; plot=true)
and
check_Hessian(M, f, grad2_f, Hess2_f; plot=true, throw_error=true, atol=1e14)
which yields the same result, but we see that the Euclidean conversion might be a bit less stable.
Now if we want to use these in optimization we would require these two functions to call e.g.
p0 = [1.0 0.0; 0.0 1.0; 0.0 0.0; 0.0 0.0; 0.0 0.0]
r1 = adaptive_regularization_with_cubics(
M,
f,
grad_f,
Hess_f,
p0;
debug=[:Iteration, :Cost, "\n"],
return_objective=true,
return_state=true,
)
q1 = get_solver_result(r1)
r1
Initial f(x): 0.666814
# 1 f(x): 0.333500
# 2 f(x): 0.233243
# 3 f(x): 0.440486
# 4 f(x): 0.607487
# 5 f(x): 0.608797
# 6 f(x): 0.608797
# 7 f(x): 0.608797
# Solver state for `Manopt.jl`s Adaptive Regularization with Cubics (ARC)
After 7 iterations
## Parameters
* η1  η2 : 0.1  0.9
* γ1  γ2 : 0.1  2.0
* σ (σmin) : 0.0004082482904638632 (1.0e10)
* ρ (ρ_regularization) : 0.9998886221507552 (1000.0)
* retraction method : PolarRetraction()
* sub solver state :
 # Solver state for `Manopt.jl`s Lanczos Iteration
 After 6 iterations

 ## Parameters
 * σ : 0.0040824829046386315
 * # of Lanczos vectors used : 6

 ## Stopping Criteria
 (a) For the Lanczos Iteration
 Stop When _one_ of the following are fulfilled:
 Max Iteration 6: reached
 First order progress with θ=0.5: not reached
 Overall: reached
 (b) For the Newton sub solver
 Max Iteration 200: not reached
 This indicates convergence: No
## Stopping Criterion
Stop When _one_ of the following are fulfilled:
Max Iteration 40: not reached
grad f < 1.0e9: reached
All Lanczos vectors (5) used: not reached
Overall: reached
This indicates convergence: Yes
## Debug
[ (:Iteration, "# %6d"), (:Cost, "f(x): %f"), "
" ]
but if you choose to go for the conversions, then, thinking of the embedding and defining two new functions might be tedious. There is a shortcut for these, which performs the change internally, when necessary by specifying objective_type=:Euclidean
.
r2 = adaptive_regularization_with_cubics(
M,
f,
∇f,
∇²f,
p0;
# The one line different to specify our grad/Hess are Eucldiean:
objective_type=:Euclidean,
debug=[:Iteration, :Cost, "\n"],
return_objective=true,
return_state=true,
)
q2 = get_solver_result(r2)
r2
Initial f(x): 0.666814
# 1 f(x): 0.333500
# 2 f(x): 0.233243
# 3 f(x): 0.440486
# 4 f(x): 0.607487
# 5 f(x): 0.608797
# 6 f(x): 0.608797
# 7 f(x): 0.608797
# Solver state for `Manopt.jl`s Adaptive Regularization with Cubics (ARC)
After 7 iterations
## Parameters
* η1  η2 : 0.1  0.9
* γ1  γ2 : 0.1  2.0
* σ (σmin) : 0.0004082482904638632 (1.0e10)
* ρ (ρ_regularization) : 0.9998886221248858 (1000.0)
* retraction method : PolarRetraction()
* sub solver state :
 # Solver state for `Manopt.jl`s Lanczos Iteration
 After 6 iterations

 ## Parameters
 * σ : 0.0040824829046386315
 * # of Lanczos vectors used : 6

 ## Stopping Criteria
 (a) For the Lanczos Iteration
 Stop When _one_ of the following are fulfilled:
 Max Iteration 6: reached
 First order progress with θ=0.5: not reached
 Overall: reached
 (b) For the Newton sub solver
 Max Iteration 200: not reached
 This indicates convergence: No
## Stopping Criterion
Stop When _one_ of the following are fulfilled:
Max Iteration 40: not reached
grad f < 1.0e9: reached
All Lanczos vectors (5) used: not reached
Overall: reached
This indicates convergence: Yes
## Debug
[ (:Iteration, "# %6d"), (:Cost, "f(x): %f"), "
" ]
which returns the same result, see
distance(M, q1, q2)
3.2016811410571575e16
This conversion also works for the gradients of constraints, and is passed down to subsolvers by deault when these are created using the Euclidean objective $f$, $\nabla f$ and $\nabla^2 f$.
Summary
If you have the Euclidean gradient (or Hessian) available for a solver call, all you need to provide is objective_type=:Euclidean
to convert the objective to a Riemannian one.
Literature
 [Bou23]

N. Boumal. An Introduction to Optimization on Smooth Manifolds. Cambridge University Press (2023).
 [Ngu23]

D. Nguyen. OperatorValued Formulas for Riemannian Gradient and Hessian and Families of Tractable Metrics in Riemannian Optimization. Journal of Optimization Theory and Applications 198, 135–164 (2023), arXiv:2009.10159.
Technical Details
This notebook was rendered with the following environment
Pkg.status()
Status `~/work/Manopt.jl/Manopt.jl/tutorials/Project.toml`
[6e4b80f9] BenchmarkTools v1.3.2
[5ae59095] Colors v0.12.10
[31c24e10] Distributions v0.25.100
[26cc04aa] FiniteDifferences v0.12.30
[7073ff75] IJulia v1.24.2
[8ac3fa9e] LRUCache v1.4.1
[af67fdf4] ManifoldDiff v0.3.6
[1cead3c2] Manifolds v0.8.75
[3362f125] ManifoldsBase v0.14.11
[0fc0a36d] Manopt v0.4.34 `~/work/Manopt.jl/Manopt.jl`
[91a5bcdd] Plots v1.39.0