Equivalent extensions of Hamilton-Jacobi-Bellman equations on hypersurfaces
Lindsay Martin, Richard Tsai

TL;DR
This paper introduces a new method to solve Hamilton-Jacobi-Bellman equations on surfaces by extending the problem to a narrow band around the surface, enabling the use of Cartesian grid methods for efficient computation.
Contribution
The authors develop an equivalent formulation of HJB equations on surfaces by extending the control problem to a narrow band, facilitating the use of existing numerical methods.
Findings
Unique viscosity solutions are obtained in the narrow band.
The method allows efficient computation using Cartesian grid techniques.
Applicable to unstructured point clouds sampled from surfaces.
Abstract
We present a new formulation for the computation of solutions of a class of Hamilton Jacobi Bellman (HJB) equations on closed smooth surfaces of co-dimension one. For the class of equations considered in this paper, the viscosity solution of the HJB equation is equivalent to the value function of a corresponding optimal control problem. In this work, we extend the optimal control problem given on the surface to an equivalent one defined in a sufficiently thin narrow band of the co-dimensional one surface. The extension is done appropriately so that the corresponding HJB equation, in the narrow band, has a unique viscosity solution which is identical to the constant normal extension of the value function of the original optimal control problem. With this framework, one can easily use existing (high order) numerical methods developed on Cartesian grids to solve HJB equations on surfaces,…
| number of points in | error | order | error | order | iterations | |
|---|---|---|---|---|---|---|
| 101 | 63302 | 0.02319 | 0.03741 | 32 | ||
| 201 | 252310 | 0.01138 | 1.0345 | 0.02213 | 0.76291 | 49 |
| 301 | 566458 | 7.5443e-3 | 1.0158 | 0.01611 | 0.78624 | 65 |
| 401 | 1571974 | 5.6562e-3 | 1.0041 | 0.01280 | 0.79978 | 80 |
| number of points in | error | order | error | order | iterations | |
|---|---|---|---|---|---|---|
| 101 | 183810 | 5.1129e-4 | 0.02779 | 65 | ||
| 201 | 702626 | 6.6194e-5 | 2.9706 | 0.01137 | 1.2987 | 92 |
| 301 | 1566014 | 2.0373-5 | 2.9182 | 7.1199e-3 | 1.1587 | 100 |
| 401 | 2776370 | 8.5992e-6 | 3.0069 | 5.1806e-3 | 1.1085 | 119 |
| errors using | ||||
|---|---|---|---|---|
| N | ||||
| 0 | 101 | 0.02332 | 0.02317 | 0.02318 |
| 201 | 0.01172 | 0.01144 | 0.01139 | |
| 301 | 8.0413e-3 | 7.6406e-3 | 7.5649e-3 | |
| 0.001 | 101 | 0.02349 | 0.02321 | 0.02313 |
| 201 | 0.01192 | 0.01153 | 0.01148 | |
| 301 | 8.2866e-3 | 7.7936e-3 | 7.75321e-3 | |
| 0.005 | 101 | 0.02464 | 0.02364 | 0.02404 |
| 201 | 0.01389 | 0.01339 | 0.01531 | |
| 301 | 0.01082 | 0.01265 | 0.01279 | |
| error | 0.02298 | 0.02303 | 0.02320 | 0.02349 |
|---|
| Method | (1) | (2) | (3) |
|---|---|---|---|
| error | 0.03226 | 0.03334 | 0.17072 |
| error | 0.07787 | 0.07667 |
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Equivalent extensions of Hamilton-Jacobi-Bellman equations on hypersurfaces
Lindsay Martin
and
Richard Tsai
Abstract.
We present a new formulation for the computation of solutions of a class of Hamilton Jacobi Bellman (HJB) equations on closed smooth surfaces of co-dimension one. For the class of equations considered in this paper, the viscosity solution of the HJB equation is equivalent to the value function of a corresponding optimal control problem. In this work, we extend the optimal control problem given on the surface to an equivalent one defined in a sufficiently thin narrow band of the co-dimensional one surface. The extension is done appropriately so that the corresponding HJB equation, in the narrow band, has a unique viscosity solution which is identical to the constant normal extension of the value function of the original optimal control problem. With this framework, one can easily use existing (high order) numerical methods developed on Cartesian grids to solve HJB equations on surfaces, with a computational cost that scales with the dimension of the surfaces. This framework also provides a systematic way for solving HJB equations on the unstructured point clouds that are sampled from the surface.
2010 Mathematics Subject Classification:
Primary 70H20,49J20, 35R01 58E10, 65N06
Department of Mathematics, The University of Texas at Austin, TX 78712, USA. E-mail: [email protected]
Department of Mathematics and Oden Institute for Computational Engineering and Sciences, The University of Texas at Austin, TX 78712, USA. E-mail: [email protected]
1. Introduction
Hamilton-Jacobi-Bellman equations have many applications in optimal control, seismology, geometrical optics, etc. From the solutions of HJB equations on surfaces, the corresponding characteristics curves can be extracted and used in many applications. Some examples include mesh generation [15], path planning [19, 37], and brain mapping [4, 21]. The equations are fully nonlinear and classical solutions typically do not exist. The unique viscosity solution [10] is often sought after. Sophisticated algorithms have been developed for computing the viscosity solution of HJB equations defined in Euclidean space.
Let where or be a bounded and connected open set with smooth closed boundary . Our goal is to compute solutions of the following HJB equation defined on smooth surfaces, with given Dirichlet boundary conditions:
[TABLE]
The set is a compact set where is the -dimensional unit sphere in and is the tangent space of at the point . We define to be the unit tangent bundle of where is the tangent bundle of . Here , is a closed subset of , and is the surface gradient on [11].
For , define the narrow band of by
[TABLE]
In this paper, we shall derive a Hamiltonian, , and extensions and of and , respectively, such that the viscosity solution to
[TABLE]
is the constant normal extension of the solution to (1)-(2), for any positive and sufficiently small .
Our contribution includes a theory for how optional control problems on surfaces can be extended into “equivalent” ones defined in a narrow band of the surface. Depending on whether the problem is isotropic or anisotropic, we must take careful consideration of extending the control space. We show for the anisotropic or most general case, the same control space used in the surface problem must be used in defining the extended problem. However, in the isotropic case the extension is equivalent even when we appropriately extend the control space. After defining the extension of optimal control problems, we then extend the corresponding HJB equation defined on the narrow band of the surface. The main advantage of this approach is that to compute solutions of HJB equations on surfaces, we are able to use Cartesian grids and existing methods (including high order methods) which solve HJB equations in Euclidean space. This allows us to avoid unnecessary patching and triangulation for solving the surface HJB equation. We show that in fact the narrow band can be very thin, i.e., it’s thickness is of order , where is the grid size.
Before proceeding, we will first define the closest point mapping that is used in the extensions. Define the closest point mapping, , by
[TABLE]
Here, must be chosen so that the closest point mapping is unique. Then we define the signed distance function to , , by
[TABLE]
For , we will define the parallel surface, , by
[TABLE]
Closest point mappings are easily derived in the context of level set methods [30], and there are a variety of fast and high order methods available to compute them from the distance function to [37, 33, 8, 36, 39], i.e.,
[TABLE]
Finally, we define the constant normal extension of function on , , to be given by
[TABLE]
Again, our goal is to define an extended HJB equation on so that the solution is the constant normal extension to the solution of the HJB equation on the surface.
The rest of the paper is organized as follows: In Section 2, we present the optimal control problem which models controlled motion on a surface and define the related class of HJB equations. We then extend the control problem and HJB equations on the narrow band, . In Section 3, we present a special case of the problem when the HJB equation reduces to the Eikonal equation and the distance function on the surface is desired. Finally, we give some numerical simulations in Section 4.
1.1. Previous work: solving HJB equations in Euclidean space
There is extensive work on developing numerical methods to compute solutions to (1) where the domain is a bounded open set in . In the view of a optimal control problem, one can obtain a semi-Lagrangian discretization for HJB equations which gives a large system of coupled nonlinear equations. Extensive studies of semi-Lagrangian techniques are found in [12, 3, 13, 14]. Solving the discretized system using fixed-point iterations is expensive, thus fast marching methods (FMMs) and fast sweeping methods (FSMs) were developed.
FMMs are a variant of Djikstra’s method and use a heapsort algorithm to determine the order in which the grid nodes are updated. The algorithm was first developed on Cartesian grids [37, 33] to solve Eikonal equations which are a special case of (1) when the functions and are isotropic, i.e., and . The solution to the Eikonal equation is well known to be the distance function to the set . FMMs have complexity , where is the total number of unknowns [31, 37]. Based on the Godunov upwind numerical scheme developed to solve Eikonal equations on triangulated domains in [5], FMMs were extended to acute triangulated meshes in [17] . However, FMMs do not directly apply to the anisotropic case. FMMs were applied to compute solutions of a class of “axis-aligned” anisotropic HJ equations in [1] on orthogonal grids, and ordered upwind methods (OUMs) were developed to solve a class of general HJB equations on any triangulated surface [34]. In [6, 7], related PDEs are solved using similar fast methods.
FSMs were developed in [36, 40] to compute solutions of a class of strictly convex HJ equations, including Eikonal equation, on Cartesian grids. The FSM is an iterative method that relies on an upwind discretization of the PDE and Gauss-Seidel style updates with different orderings of grid nodes. The idea is to avoid using heapsort and, instead, update the grid nodes in different orderings (sweeps). In each sweep, characteristics that go in similar directions are propagated automatically. In [22], a FSM was developed for Eikonal equations using a discontinuous Galerkin (DG) local solver for computing the distance function. For more general Hamilton-Jacobi equations, one may use a Lax-Friedrichs type numerical Hamiltonian [16] on Cartesian grids. The resulting Lax-Friedrichs sweeping scheme is typically more diffusive and requires more iterations. FSMs have complexity with the caveat that the constant hidden in the notation may be large depending on the characteristics of the PDE.
1.2. Previous work: solving HJB equations on surfaces
Several versions of FMMs and FSMs can be applied to computing the distance function on surfaces. As already mentioned, the FMM in [17] only works for solving Eikonal equations on acute triangulated surfaces. In [35], FMMs equation were extended to solve Eikonal equations on parametric surfaces where the discretization is done on a Cartesian grid in the parametric plane. The FSM in [36] and the FMM in [29] can compute the distance function on surfaces that are defined as the graph of a smooth function. Another approach to computing the distance function on surfaces is to solve a corresponding evolutionary HJ equation where the distance function is the steady state solution [8]. However, with the addition of the time variable this method is not computationally optimal.
OUMs are applicable to more general anisotropic HJB equations on triangulated surfaces. However, if the surface is given in implicit form, triangulation may be unnecessary. We want to avoid triangulation and have the ability to handle general surface representations. Our motivation is to derive equivalent extended HJB equations on a narrow band of the surface such that a variety of meshes and methods can be used to solve the equivalent equations in the narrow band. In particular, the methods developed for Cartesian grids (surveyed in Section 1.1) can then be used.
In [27], the idea of extending the surface by a small offset, , to a narrow band is used to compute distance functions on surface. On the narrow band, an extended distance function is computed, but it is not the constant normal extension of the distance function on the surface. Therefore, the formulation introduces an analytical error. It is proven that the error between the two distance functions is controlled by , the width of the narrow band. In order for the method to converge as the grid size goes to zero, it is required that where is the grid size and The advantage of our framework is that there no analytical error introduced. Our formulation is convergent with respect to the grid size, , and for narrow bands of order . We are also able to perform high order computations for the distance function on surfaces.
The idea of extending functions or differential operators defined on surfaces, via closest point mapping, to the embedding Euclidean space, is used in [32, 24, 9] to compute solutions to parabolic and elliptic PDEs. Our approach is inspired by the work in [9] where variational integrals defined originally on surfaces are extended to ones defined in the narrow band, and the Euler-Lagrange equations of the extended problem are solved by standard numerical methods. Due to the way the extensions are defined, the resulting solution to each equation is automatically the constant normal extension of the solution to the variational problem on the surface.
1.3. Computing PDEs “on point clouds”
Point clouds arise in many applications including facial recognition, manufacturing, medicine, and geosciences and can be acquired easily through modern sensing devices such as laser scanners and cell phones. A typical approach, particularly predominant in the computer graphics community, is to create triangular meshes from the data points and seek to define partial derivatives in certain suitable ways. Among myriad of papers using triangular surface meshes, we mention [5], an early paper which is the most related to the topic of this paper. Another school of approaches aim at solving differential equations on surfaces, using only a finite set of points sampled from the surfaces, without globally reconstructing the surface.
In [28], the algorithm developed in [27] is extended to compute distance functions on surfaces represented as point clouds. In [23], a framework for computing solutions to PDEs on point clouds based on a local approximation of the manifold was presented. A local mesh algorithm was developed in [20] that solves PDEs on point clouds where any of the existing methods valid on triangular meshes discussed in Section 1.2 can be used to compute the solutions to HJB equations once the local mesh is determined.
Our new proposed non-parametric formulation also provides a convenient way to solve equations “on point clouds.” That is we can solve HJB equations which are defined on a smooth closed surface, but the only available information about the surface is a finite set of points that are reasonably distributed over the surface. The discussion of the implementation of our method applied to point clouds is in Section 4.1.4. There, it is assumed that there is no noise perturbing the point cloud in the surface normal directions, and that the point clouds are evenly distributed over the surfaces. The generalization of the proposed algorithm to more general point clouds is the subject of another paper.
2. Static Hamilton-Jacobi-Bellman equations on surfaces
2.1. Modeling controlled motion on a surface
First, we define the optimal control problem from which the HJB equations can be derived. For each , let be set of all unit tangent vectors of at , i.e.,
[TABLE]
Note each is a compact set in for each . Define , then where is the tangent bundle of . The set of admissible controls is given by:
[TABLE]
We are interested in the trajectories governed by the dynamical system:
[TABLE]
where represents the velocity and , which ensures that for all . Denote the solution to (5)-(6) (which exists under the assumption (A1) below) by . For each define the following subset of the admissible controls:
[TABLE]
Then the requirement that for all in the dynamical system (5)-(6) is equivalent to Refer to that satisfy (5)-(6) where as an admissible path on . For the rest of the paper, we will suppress writing the dependence of on when there is no confusion in the context.
Let denote a running cost per unit time and be an exit time penalty when the state reaches a closed target set . The total cost associated with initial state and control to reach is given by
[TABLE]
where
We have the following assumptions on and :
[TABLE]
The value function is defined to be the minimal total cost to starting at and is given by
[TABLE]
Under the assumption (A1), an optimal control does not necessarily exist. If for the given and , we have that the set is strictly convex for each , then an optimal control is guaranteed [2]. This property is trivially satisfied in the case of isotropic and . Regardless of whether an optimal control exists, we can still derive the corresponding HJB equation.
The dynamic programming principle [2] for the value function states that for sufficiently small ,
[TABLE]
The corresponding Hamilton-Jacobi-Bellman equation on is
[TABLE]
The boundary condition is
[TABLE]
Note that in (10) we can take the minimum since is compact.
In general, classical solutions of (10)-(11) do not exist, and the unique viscosity solution is sought after [10]. A detailed discussion of viscosity solutions of Hamilton-Jacobi equations on manifolds can be found in [25]. Under the assumption (A1), it is a classic result that the value function coincides with the unique viscosity solution of (10)-(11) [2]. In the case of isotropic running cost and speed, then (10) reduces to the Eikonal equation on the surface:
[TABLE]
We will expand more on the Eikonal equation in the next section.
2.2. Extension to
We derive a Hamiltonian, , on such that if is the unique viscosity solution of
[TABLE]
with boundary conditions defined appropriately, then
[TABLE]
where is the viscosity solution to (10)-(11).
First, we extend the optimal control problem on the surface to one in the narrow band, . We start by defining a tensor that will be essential in formulating an equivalent optimal control problem on .
Definition 2.1**.**
Let and be two orthonormal tangent vectors corresponding to the directions that yield the principle curvatures of at a point . Let be the unit normal vector to the tangent plane at . Let and be the singular values of , the Jacobian matrix of at . Then for any real number , define the tensor by
[TABLE]
where
[TABLE]
In the above definition, corresponds to a weighted orthogonal projection onto the tangent plane of at , and is the projection along the normal of passing through . In [18], it is proven that
[TABLE]
where and are the principal curvatures of the parallel surface . Note that when , For the sake of notation, we will suppress the dependence of the tangent vectors, normal vectors, and singular values on the point .
Assuming small enough, we have that the tangent planes coincide for and if . Thus, for the extended optimal control problem, the set of compact control values at each is Then and are as in Section 2.1. Now, consider the following extended dynamical system in :
[TABLE]
Here is given by and we have that for all This last requirement ensures that the trajectory remains in for all where . The inclusion of the tensor in the dynamical system above adjusts the velocity according to the curvature of the surface. This has the implication that equivalent paths will have the same total cost.
Denote the solution to (14)-(15) (which exists under the assumption (B1) below) by . For each with , we have the following subset of admissible controls:
[TABLE]
Then the requirement that for all in the dynamical system (14)-(15) is equivalent to Since , we have . Refer to that satisfy (14)-(15) where as an admissible path on . Again, for the rest of the paper we will write suppress writing the dependence of on . There is no confusion between the notation for admissible paths on and . The initial point will indicate which dynamical system the path solves.
Note that if the initial point , and
[TABLE]
since and Thus, (14)-(15) reduces to the dynamical system on given in (5)-(6), and an admissible path on with is also an admissible path on with the same .
Now, let . Let the running cost and exit time penalty on be given by where and where , respectively. Then the total cost associated with initial state and control to reach is given by
[TABLE]
where . Note that if , then The value function is the minimal total cost to starting at and is given by
[TABLE]
We have that the assumption A1 implies:
[TABLE]
Again, while (B1) does not guarantee the existence of the optimal control, the assumption does ensure that the value function and corresponding viscosity solution of the HJB equation coincide. If we also assume the set is strictly convex, then is strictly convex for each . Thus, if the optimal control exists in the problem on , it also exists for extended problem on .
Before deriving the corresponding HJB equation, we prove that the value function, , is the constant normal extension of the value function of the optimal control problem defined on . The following theorem states that two admissible paths, and , on with equivalent initial points in and equivalent controls stay equivalent for all time.
Theorem 2.1**.**
Suppose that such that . Let and such that . Suppose solves (14) with and solves (14) with . Then for all .
Proof.
Let and be given by
[TABLE]
and
[TABLE]
for all , respectively. We need to show that for all . We compute . We have the following singular value decomposition for : where
[TABLE]
Since , and are orthonormal, for ,
[TABLE]
and for
[TABLE]
Using the above, we have
[TABLE]
The last equality is true since is the orthogonal projection of onto the tangent space at and . Thus is an admissible path on where and By the same reasoning also an admissible path with and We assume (A1). Thus, the solution to (5)-(6) is unique. Since , for all . ∎
Now, it easily follows that the value function defined on is the constant normal extension of the value function defined on
Corollary 2.2**.**
If is the value function on given by (8), then
[TABLE]
where is the value function on given by (16).
Proof.
Let and . Then since . Then we have
[TABLE]
and
[TABLE]
where and are admissible paths on with , , respectively. Recall, that is also an admissible path on so that .
Theorem 2.1 implies that for all . Since
[TABLE]
we have
[TABLE]
Therefore, , and
[TABLE]
Finally, we have
[TABLE]
∎
Now that we have showed that the value function on the narrow band is equivalent to the value function on the surface, we define the Hamilton-Jacobi-Bellman equation on associated the value function . Again, the dynamic programming principle states that for sufficiently small , we have
[TABLE]
The corresponding Hamilton-Jacobi-Bellman equation on is
[TABLE]
The boundary condition is
[TABLE]
The assumption (B1) implies that the value function coincides with the unique viscosity solution [2]. Corollary 2.2 implies that the viscosity solution of (18)-(19) is given by
[TABLE]
where is the viscosity solution of (10)-(11) on . An interesting result is that we can prove (20) without Corollary 2.2, i.e., using only the derived HJB equations and properties of viscosity solutions.
We begin with the following lemma that relates the surface gradients for parallel surfaces at equivalent points:
Lemma 2.3**.**
Assume is small enough so that is differentiable, and let . Given , define by . Then for ,
[TABLE]
Proof.
Let be the constant normal extension of Then and {\left.\kern-1.2pt\overline{\phi}_{0}\vphantom{\big{|}}\right|_{\Gamma_{\eta}}}=\phi_{\eta}. From Theorem A.1 in [18], we have that . Since is the constant normal extension of , . Thus, if , , and (21) holds. ∎
The following theorem proves (20) directly using the definition of viscosity solutions.
Theorem 2.4**.**
If is the viscosity solution to (10)-(11), then the constant normal extension of , is the viscosity solution to (18)-(19).
Proof.
We have for and if . Then for ,
[TABLE]
Thus, satisfies the boundary conditions (19). Next, we will show that is a viscosity subsolution of (18).
Assume and such that has a strict local maximum at and We need to show that
[TABLE]
Let and be the restriction to of the normal extension of {\left.\kern-1.2pt\phi\vphantom{\big{|}}\right|_{\Gamma_{\eta}}}. Then we have that and \phi_{\eta}:={\left.\kern-1.2pt\phi\vphantom{\big{|}}\right|_{\Gamma_{\eta}}}\in C^{1}(\Gamma_{\eta}) such that for .
First, we will prove
[TABLE]
Since , we have and
[TABLE]
Now, . Therefore, (23) holds.
Lemma 2.3 implies
[TABLE]
Let Then using the fact that is symmetric, (23), and (24), we have
[TABLE]
Now, such that has a strict local maximum at and
[TABLE]
Since is a viscosity subsolution of (10),
[TABLE]
Thus, (22) holds. The same argument can be applied to show that is a viscosity supersolution of (10). ∎
In the next section, we will present a special case of the above formulations for when the speed and running cost functions are isotropic. The HJB equation on reduces to the Eikonal equation. A noteworthy revelation is that if we ensure that the extended speed and cost functions are isotropic as well, the control values in the extended optimal control problem do not have to be restricted to be in the tangent bundle of , i.e., the extended set of controls is . In this setup, the value function is the constant normal extension of the value function on Before we proceeding, we present a counter example to show that is not necessarily the value function of the extended control problem if the extended speed function is anisotropic and .
Example
Consider the following control problems on and : Let
[TABLE]
and
[TABLE]
Let , then
[TABLE]
and
[TABLE]
For the control problem on , the set of controls is and is the usual set of admissible controls. Define by and suppose that we have unit running cost, i.e., . Then it is clear that which is the minimum travel time from to while traveling at speed .
Now for the extended control problem, let the control values be the set . Then the set of admissible controls is given by
[TABLE]
Define such that and the set is the ellipse given by rotated at the origin clockwise by the angle . Then if so that for . Again, assume unit running cost.
Let be the value function for this extended control problem. If , then we should have . However, note that and the Euclidean distance between and is . Thus,
[TABLE]
when . By the definition of the value function we must have
[TABLE]
Therefore, cannot be the value function for the extended control problem.
3. The Isotropic Case
We derive the optimal control problems and HJB equations corresponding to the case when the speed and cost functions are isotropic. As mentioned at the end of Section 2.2, we will show that value function on is the normal extension of the value function on even when we do not restrict the controls to the tangent space in the extended optimal control problem.
3.1. The controlled dynamics on for isotropic speed and cost functions
The optimal control problem formulation is exactly the same as in Section 2.1, and (10) reduces to the Eikonal equation on the surface:
[TABLE]
We have the HJB equation
[TABLE]
and the boundary condition is
[TABLE]
Assuming (A1), the value function coincides with the unique viscosity solution of (26)-(27). We also have that is strictly convex for all . Thus, an optimal control is guaranteed.
3.2. Extension to in the isotropic case
We derive the extended Hamiltonian, , on from the extended isotropic optimal control problem such that if is the unique viscosity solution of
[TABLE]
with boundary conditions defined appropriately, then
[TABLE]
where is the solution to (26)-(27).
The notations , , , , and are as in Section 2.2. The extended set of compact control values is , and the set of admissible controls is given by:
[TABLE]
This is where the formulation differs from the previous section. We do not restrict the controls to belong to the tangent bundle of . We extend the speed, cost, and exit time penalty functions as above: , , and . The dynamical system is given by
[TABLE]
where . Admissible paths on with extended controls are solutions to (28)-(29) denoted by . Note that the admissible paths are not restricted to the parallel surface, , to which the initial point belongs.
The total cost function associated to the initial state and control is the same as in Section 2.2. The value function is the minimal total cost to starting at and is given by
[TABLE]
We have that (A1) implies the following is true:
[TABLE]
We have that (C1) implies the value function, , coincides with the unique viscosity solution of (26)-(27) [2]. We also have that is strictly convex for all . Thus, an optimal control is guaranteed.
We now prove that the value function, , is a constant extension of a function on . However, unlike in the previous section, it is not immediate from the following proofs that , where is the value function on .
Theorem 3.1**.**
Given and such that , let
[TABLE]
and
[TABLE]
solve (28) with and , respectively. Then for all .
The proof of Theorem 3.1 is analogous to the proof of Theorem 2.1 except the control belongs to the extended control space, . A consequence of the control belonging to the extended space is that now the projected paths may not solve (5). This is due to the fact that if where and are the basis tangent vectors of We now prove that the extended value function in the isotropic case is constant along the normals of .
Corollary 3.2**.**
If for , then .
Proof.
Let . We have
[TABLE]
and
[TABLE]
where and are admissible paths on with extended control, , and , where .
Theorem 3.1 implies that for all . Since
[TABLE]
we have
[TABLE]
Therefore, and for . Now,
[TABLE]
∎
In the above proof, it is not immediate that we can take the infimum over which would imply that is the normal extension of , the value function on . We will use the corresponding HJB equations and Corollary 3.2 to show that .
We define the HJB equation on associated the value function . Again, the dynamic programming principle states that for sufficiently small we have
[TABLE]
We get the Hamilton-Jacobi-Bellman equation on :
[TABLE]
Since the speed and cost functions are isotropic, (32) reduces to an anisotropic Eikonal equation:
[TABLE]
The boundary condition is
[TABLE]
Since the value function coincides with the unique viscosity solution, Corollary 3.2 implies that the viscosity solution of (33)-(34) is a constant along the normals of , i.e., there is a function such that for ,
[TABLE]
We now prove that where is the viscosity solution of (26)-(27).
Theorem 3.3**.**
If is the viscosity solution to (26)-(27), then the constant normal extension of , is the viscosity solution to (33)-(34).
Proof.
Just as in Theorem 2.4, satisfies the boundary conditions (34). Assume for contradiction that is not the viscosity solution to (33). But from Corollary 3.2 we have that viscosity solution of (33)-(34), for some Therefore, if is not the unique viscosity solution then we have . For contradiction, we will show that is the viscosity solution of (26) and thus, .
Let and such that has a local maximum at . Consider the normal extensions of and , then has a local max at . Since is a viscosity subsolution to (33) and , we have that
[TABLE]
Since , and . Therefore,
[TABLE]
and we have that . Thus,
[TABLE]
and is a viscosity subsolution. The same argument can be applied to show that is a viscosity supersolution. Therefore, is the viscosity solution to (26) and hence . ∎
To summarize, we have shown that we may appropriately extend the control space in the case of isotropic speed and running costs. This has the implication that even though the admissible paths on have the “choice” to leave the tangent space of , the optimal paths in the extended control problem remain on or parallel to
4. Numerical implementation and simulations
4.1. Numerical setup
In the implementation of our new framework, we use a uniform Cartesian grids and denote the grid step size by . denotes the part of the grid in the narrow band of radius around the given surface.
We use the Lax-Friedrichs fast sweeping methods in [16] and a high order version in [39] for our simulations. It is also possible to an upwind scheme generalized from [36]. Unless we mention otherwise, we use the standard first order finite differencing in the approximation of the partial derivatives for the Lax-Friedrichs numerical Hamiltonians.111 All code used to produce the numerical simulations can be found at https://github.com/lindsmart/MartinTsaiExtHJB.
We present several examples of our new formulation performed on surfaces of co-dimension one in three dimensions. For the first three examples, we compute the solution to the Eikonal equation on the surface. As shown in Section 3, the equivalent equation on is
[TABLE]
Again,
[TABLE]
Since the desired solution, , has been proven to be constant along the normals of the surface, , . Therefore, can be any real number. We let in all of our computations.
Approximating the solution to (37)-(38) requires the computation of the singular values and vectors of the derivative of the closest point mapping, . We defer the discussion of these approximations to Section 4.1.4. In the last example, we apply our framework to solve an HJB equation with an anisotropic speed function.
4.1.1. Boundary closure
We note that the analytical formulation of the HJB equations does not require boundary conditions on . However, since we are using Cartesian grids we must take careful consideration of the discretization near the boundary, . When approximating the partial derivatives of , a neighboring grid node may lie outside of . We will call these ghost nodes. In our implementation, we provide a boundary closure procedure to enforce the fact that the solution is a constant along normal function. For each ghost node, we perform the following procedure:
- (1)
Project the node into the narrow band. 2. (2)
The value at the ghost node is then calculated by interpolating grid values surrounding the projected point. Formally, we must use an interpolation scheme of order higher to the discretization of the PDE on .
Next, we describe how the boundary closure procedure affects the numerical accuracy. Let be a ghost node, and be the projection of the ghost node into . Denote the solution at by . Suppose that we use a first order scheme to discretize the HJB equation on . Then the interpolation used in the approximation of should yield at least second order in accuracy, in order to maintain formally the first order accuracy. This is due to the amplification by a factor of in the finite difference scheme errors of the approximation of the values at the neighboring grid nodes of .
4.1.2. Depth of projected points and bounding the thickness of the narrow band
When choosing for the the projected ghost node discussed above, a necessary condition is that the nearby surrounding nodes used to interpolate the value of must be in . In our numerical simulations, we use cubic interpolation to approximate . Thus, it requires nearby grid nodes for three dimensional cases. It can be shown that if , where is the dimension. Then the inner nodes used to interpolate the value at each ghost node are in . We suggest to use a projected node as close to the boundary node as possible. Therefore in our numerical simulations, we choose . We discuss the effect of the choice of in Section 4.2.
The maximal value of is restricted by the curvatures of . We must have smaller than the reach of the surface, which depends on the reciprocal of the maximal curvatures. The minimal value of is determined by the finite difference stencils in approximating the partial derivatives of the PDE and the boundary closure procedure. Both stencils require that the narrow band is sufficiently “thick” relative to the mesh size with .
The resulting numerical discretization of the HBJ equation on is convergent for any in the interval described above; i.e. it is convergent for very thin narrow bands as well as for thicker one whose widths are independent of . We remark that this convergent regime is very different that requiring , in the related work of [18], dealing with singular integrals.
4.1.3. Approximation errors
The error computed by the proposed algorithm can be written as the sum of errors corresponding to different approximations:
[TABLE]
where
- •
relates to how the surface PDE is approximated in the tubular neighborhood by an extended PDE and the boundary conditions. Our main contribution is in deriving the extension for which for any that is smaller than the reach of the surface;
- •
corresponds to the numerical error for approximating the extended PDE problem. Our extension allows for higher order (in ) methods to be applied;
- •
relates to errors in approximating the surface and its geometric properties. This error corresponds to approximation of the closest point mapping to the surface and its curvatures. Since these quantities are computed on based on finite differencing using the grid, the error depends on ;
- •
is the numerical error in approximating the norm of the computed errors. Our formulation allows for a very accurate approximation to this error term.
4.1.4. Application to solving HJB on point clouds
We now describe how we compute the closest point mapping when the surface is represented as a point cloud. (See Figure 1.) Denote the point cloud by , with indicating the total number of points in the set. Here, we shall assume that the points uniformly distribute on the surface. The purpose of the exposition here is to point out a promising application of the method to point cloud data. A full analysis of the algorithm in this area warrants a separate paper.
We approximate using the following strategy:
- (1)
For each , estimate locally on the grid nodes. Call the local surface associated to the point . 2. (2)
Compute the closest point of on the local surface,
In our implementation, we use biquadratic interpolation at each to compute . Another option to locally estimate the surface is least squares as in [23]. We then use Newton’s method to obtain the closest point of each on Denote the closest point mappings by . Then on each grid node , the equation is discretized as usual, with being approximated by finite differences of We use a fourth-order finite difference scheme to estimate and singular value decomposition to obtain the singular values and singular vectors of . We show the effect of this approximation procedure in Section 4.2.
In any case, the term, , in (39) now depends on , which corresponds to the uniform spacing of the points. Therefore, with a fixed point cloud, refinement of Cartesian grids eventually will not further improve the accuracy in the numerical solutions when compared to the solution on the idealized smooth surface sampled by the point cloud. In Section 4.3, we present some results revealing the effect of different surface sampling density.
If noise is introduced to the point cloud, and if the amount of perturbation in the original ”surface normals” is significant, more sophisticated surface fitting is required; e.g. adopting a similar strategy proposed in [23]. We will investigate this important issue in a future manuscript. Nevertheless, in Section 4.3, we present some results revealing the effect of noise in the point cloud.
4.2. Example 1.1
First, we present a numerical convergence study. We approximate the solution to (37)-(38), where is a sphere centered at with radius, , and is a point on the sphere. The exact singular values and vectors in are used. In the case of a sphere
[TABLE]
where is the radius of the sphere going though centered at . Figure 2 shows the distance function to a point on the sphere for two different view points.
Since the solution is constant along the normals of , we can easily estimate the error. Using the formulation in [18], the estimation of the error is given by the following formula:
[TABLE]
where is the exact solution and is the approximated solution on using the approximate solution, , of the extended equation with grid size . Here,
[TABLE]
where and
[TABLE]
where are given in (40).
Because all the characteristics of the solution emanate from the source point on the sphere, the errors at the source point will propagate through the solution in the entire computational domain. Therefore, we initialize the solution near with the exact solution if The and errors are reported in Table 1. We see that first order convergence is achieved in the error. However, because of the singularity of the solution at the point opposite the source point on the sphere, the order in the error is less than one. The results verify that the boundary closure procedure does not influence the overall order of the scheme.
We report the number of iterations it takes for the norm of the difference in the solution between successive iterations is less than . One iteration includes one sweep of the grid in each of the eight sweeping directions. The timing for the numerical experiments scale linearly to the number of grid nodes in the narrow band and the number of iterations used in the Lax-Friedrichs scheme. The computations were performed on a MacBook laptop with a 1.6 GHz Intel CPU on a single core using Julia. One Lax-Friedrichs iteration took on average 0.34260 seconds for the case in Table 1. We note that our code was not optimized for efficiency.
Another advantage of our setup is that we can use existing high order methods to compute the solution to HJB equations on surfaces. In Table 2, we present the and errors and orders for the third order method given in [39]. We initialize the solution near to the exact solution if We see that third order is achieved in the error. Again, because of the singularity in the solution at the point opposite the source point on the sphere, the order in the error is first order.
4.3. Example 1.2
In this example, we compute the distance function on the sphere when the sphere is represented as a point cloud. We show the effect of our method when using different uniform samplings of the sphere.222The uniform samplings of the sphere were computed using the code provided at https://github.com/AntonSemechko/S2-Sampling-Toolbox. The results are tabulated in Table 3 when the sampling of the sphere has and points. We again initialize the solution by exact solutions around the same neighborhood of as in Section 4.2. By comparing the error from Table 1 for and the error in Table 3 for and no noise, it appears that the dominating source of error is the numerical discretization of the PDE. As the grid is refined, we see that it appears the error related to approximating the surface begins to dominate. This is apparent by comparing the error from Table 1 for and the error from Table 3 for with no noise. Notice as the point cloud becomes more dense and the accuracy of the surface approximation increases, it appears the dominating error is again due to the numerical discretization of the PDE.
In Table 3, we also consider when noise is introduced to the point clouds of different sampling densities. The noise is applied by adding to each point in the point clouds, where and rand() is a uniformly distributed random number in . We can see if the noise is too large the error begins to deteriorate and we need a more sophisticated surface fitting algorithm in our method.
4.4. Example 1.3
Next, we study the affect that the depth of projected ghost nodes has on the overall error for the distance function on the sphere. The set up is the same as in the previous section. Here, we choose a sized grid with where . We estimate the error for when the boundary closure procedure is carried out at four different depths. Recall that a ghost node, , is outside of and is projected into along the normal of the surface at that point, i.e.,
[TABLE]
Table 4 shows the error for the depths:
[TABLE]
We can see that the error the solution is not very sensitive to the choice of depth. In our simulations, we chose .
4.5. Example 1.4
Finally, we compare the following: (1) our method on the sphere, using exact singular values and vectors as in the set up for Table 1; (2) our method with the point cloud procedure described in 4.1.4 and finite differences to estimate ; (3) the method in [27]. Since method (3) requires where and , to fairly compare the error of the methods, we increase the width of the narrow band to which is used in the convergence analysis of (3) in [27]. We initialize the experiments using methods (1), (2), and (3) with exact values given in a box of length around the source point/points in the initial sets. We also are not able to compute the error for method (3) since the solution is not constant along the normals of . Therefore, we can only report the error, and we use trilinear interpolation to approximate the solutions on for (3). The errors are reported in Table 5. We can also see that estimating from a point cloud and from finite differences does not greatly influence the error of the method.
4.6. Example 2
In Figures 3, 4, and 5, the distance function to a source point is shown on various surfaces. The contours shown are equally spaced and parallel. The solutions for the torus and bunny are computed on a grid, and a grid is used for the elephant, mask, human skull, and dinosaur skull. All computations use a narrow band width of and point cloud representations for the surfaces. The number of points in the point clouds of the torus333The point cloud for the torus was generated using the standard parametrization of a torus. and bunny444The point cloud for the Stanford bunny is generated from a refinement of the triangulated Stanford bunny from https://casual-effects.com/data/ [26]. are 178,350 and 228,096, respectively. For the elephant, mask, human skull, and dinosaur skull555The point clouds for the surfaces in Figure 5 were generated from the triangulations downloaded at https://www.myminifactory.com/scantheworld/., we used point clouds with 65,292, 1,199,988, 234,618, and 139,491 points, respectively. In Figure 3, a cross section of the solution is displayed to show that the solution is indeed constant along the normals of the surface. Our new framework also allows us to sort point clouds. Figure 3 displays level “belts” of the point cloud, i.e., points in the point cloud whose distance from the source point lies in given interval.
Another advantage of our formulation is that when we compute the characteristics, known as geodesics, of the Eikonal equation via the extended Eikonal equation on the narrowband, the paths remain on the surface if the initial point lies on . Since the solution to (37)-(38) is constant along the normals of , the gradient always belongs to the tangent spaces of or its parallel surfaces, . We have that for and since . Thus, if the initial point is , the geodesic on the surface can be obtained by solving the dynamical system
[TABLE]
In Figure 6, we consider the case when we have two source points and display the solution and some geodesics. Recall that in Theorem 2.1 we showed that paths with equivalent initial points remain equivalent for all time. We show a visualization of this property in Figure 7. In the next section we will compare geodesics of the HJB equation given different anisotropic speed functions.
4.7. Example 3
Finally, we test implement the framework on an anisotropic speed function. We solve
[TABLE]
where is a curvature based speed function on the surface. We use the speed function proposed in [38]. The speeds are fast on low curvature areas of the surface and slow on high curvature areas. This means that the corresponding “shortest” paths traverse areas of the lowest curvatures. These paths are typically longer than the geodesics of the surface.
The normal curvature of at in the direction is given by
[TABLE]
where and are the principal curvatures of at . Then the curvature-minimizing speed function is given by
[TABLE]
where is a positive constant. When , (43) reduces Eikonal equation on the surface. A larger value of corresponds to a greater difference in speeds between areas of low and high curvature. We display the solutions for varying values to a source point on the bunny in Figure 8.
In this example, the set is not neccesarily convex for all in Therefore, an optimal control may not exist. However, we can still extract suboptimal paths, called anisotropic geodesics, whose total cost is arbitrarily close to the value function at the starting point. Just as in the isotropic case, the anisotropic geodesics computed from the extended HJB equation on the narrow band will lie on the surface since for . The paths can easily be extracted because we compute the minimizing control at each grid node when solving (43)-(44). Once we have the optimal (or suboptimal) control values, , we then solve dynamical system
[TABLE]
We plot three anisotropic geodesics for each value in Figure 8. We can see as increases the Euclidean distances of the paths are longer, and the paths start to seek out the narrow valleys of the bunny.
5. Summary and conclusion
In this paper, we presented a new formulation to compute solutions of a class of HJB equations on smooth hypersurfaces. We extend the HJB equation’s associated optimal control problem from the surface to an equivalent problem defined in a sufficiently ”thin” narrow band around the surface, in the embedding Euclidean space. The extension was done so that the resulting value function is the constant normal extension of the value function defined in the optimal control problem on the surface. We presented the formulations for the general anisotropic equation and showed that the viscosity solution of the HJB equation on the narrow band is the constant normal extension of the viscosity solution on the surface, independently of the optimal control problems. We also presented the isotropic case and showed there is no need to restrict the control space in order to have an equivalent formulation. The proposed approach is independent of surface representation and can be used to compute and define optimal control problems on uniformly distributed point clouds sampled from some smooth surface. It is also clear that our proposed extension approach can be applied in to time dependent equations arising from the finite horizon control problems. Together with [9], our extension approach provides a good framework for solving to mean field games to high order accuracy on complicated and non-parametrized surfaces.
With this new framework, we are able to use Cartesian grids and the existing methods for computing solutions to HJB equations in Euclidean space on a very thin narrow band to solve HJB equations on surfaces coupled with a simple boundary closure procedure. Our numerical examples verify that the boundary closure procedure does not influence the overall order of the method. We also show that our formulation allows one to easily solve the surface HJB equations to high order accuracy.
Acknowledgment
The authors are supported partially by National Science Foundation Grant DMS-1720171. Tsai also thanks National Center for Theoretical Study, Taipei for hosting his visits, in which some of the ideas presented in this paper originated.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] K. Alton and I. M. Mitchell. Fast marching methods for stationary Hamilton-Jacobi equations with axis-aligned anisotropy. SIAM J. Numer. Anal. , 47(1):363–385, 2008/09.
- 2[2] M. Bardi and I. Capuzzo-Dolcetta. Optimal control and viscosity solutions of Hamilton-Jacobi-Bellman equations . Systems & Control: Foundations & Applications. Birkhäuser Boston, Inc., Boston, MA, 1997. With appendices by Maurizio Falcone and Pierpaolo Soravia.
- 3[3] M. Bardi and M. Falcone. An approximation scheme for the minimum time function. SIAM J. Control Optim. , 28(4):950–965, 1990.
- 4[4] A. Bartesaghi and G. Sapiro. A system for the generation of curves on 3d brain images. Human Brain Mapping , 14(1):1–15, 2001.
- 5[5] T. J. Barth and J. A. Sethian. Numerical schemes for the Hamilton-Jacobi and level set equations on triangulated domains. J. Comput. Phys. , 145(1):1–40, 1998.
- 6[6] E. Carlini, M. Falcone, and R. Ferretti. A time-adaptive semi-Lagrangian approximation to mean curvature motion. In Numerical mathematics and advanced applications , pages 732–739. Springer, Berlin, 2006.
- 7[7] E. Carlini, M. Falcone, N. Forcadel, and R. Monneau. Convergence of a generalized fast-marching method for an eikonal equation with a velocity-changing sign. SIAM J. Numer. Anal. , 46(6):2920–2952, 2008.
- 8[8] L.-T. Cheng, P. Burchard, B. Merriman, and S. Osher. Motion of curves constrained on surfaces using a level-set approach. J. Comput. Phys. , 175(2):604–644, 2002.
