š¤ 2d generalized BratuāGelfand problem
We consider the problem of Mittelmann [Farrell] [Wouters]:
\[\Delta u + NL(\lambda,u) = 0\]
with Neumann boundary condition on $\Omega = (0,1)^2$ and where $NL(\lambda,u)\equiv-10(u-\lambda e^u)$. This is a good example to show how automatic branch switching works and also nonlinear deflation.
We start with some imports:
using Revise, ForwardDiff
using BifurcationKit, LinearAlgebra, Plots, SparseArrays
const BK = BifurcationKit
# define the sup norm and a L2 norm
normbratu(x) = norm(x .* w) / sqrt(length(x)) # the weight w is defined below
# some plotting functions to simplify our life
plotsol!(x, nx = Nx, ny = Ny; kwargs...) = heatmap!(reshape(x, nx, ny); color = :viridis, kwargs...)
plotsol(x, nx = Nx, ny = Ny; kwargs...) = (plot();plotsol!(x, nx, ny; kwargs...))
and with the discretization of the problem
function Laplacian2D(Nx, Ny, lx, ly)
hx = 2lx/Nx
hy = 2ly/Ny
D2x = spdiagm(0 => -2ones(Nx), 1 => ones(Nx-1), -1 => ones(Nx-1) ) / hx^2
D2y = spdiagm(0 => -2ones(Ny), 1 => ones(Ny-1), -1 => ones(Ny-1) ) / hy^2
D2x[1,1] = -1/hx^2
D2x[end,end] = -1/hx^2
D2y[1,1] = -1/hy^2
D2y[end,end] = -1/hy^2
D2xsp = sparse(D2x)
D2ysp = sparse(D2y)
A = kron(sparse(I, Ny, Ny), D2xsp) + kron(D2ysp, sparse(I, Nx, Nx))
return A, D2x
end
Ļ(u, Ī») = -10(u-Ī»*exp(u))
dĻ(u, Ī») = -10(1-Ī»*exp(u))
function NL!(dest, u, p)
(;Ī») = p
dest .= Ļ.(u, Ī»)
return dest
end
NL(u, p) = NL!(similar(u), u, p)
function Fmit!(f, u, p)
mul!(f, p.Ī, u)
f .= f .+ NL(u, p)
return f
end
Fmit(u, p) = Fmit!(similar(u), u, p)
It will also prove useful to have the derivatives of our functional:
function JFmit(x,p)
J = p.Ī
dg = dĻ.(x, p.Ī»)
return J + spdiagm(0 => dg)
end
We need to define the parameters associated to this problem:
Nx = 30; Ny = 30
lx = 0.5; ly = 0.5
# weight for the weighted norm
const w = (lx .+ LinRange(-lx,lx,Nx)) * (LinRange(-ly,ly,Ny))' |> vec
Ī, = Laplacian2D(Nx, Ny, lx, ly)
par_mit = (Ī» = .05, Ī = Ī)
# initial guess f for newton
sol0 = zeros(Nx, Ny) |> vec
# Bifurcation Problem
prob = BifurcationProblem(Fmit, sol0, par_mit, (@optic _.Ī»),; J = JFmit,
record_from_solution = (x, p; k...) -> (x = normbratu(x), n2 = norm(x), nā = norminf(x)),
plot_solution = (x, p; k...) -> plotsol!(x ; k...))
To compute the eigenvalues, we opt for the shift-invert strategy with shift =0.5
# eigensolver
eigls = EigKrylovKit(dim = 70)
# options for Newton solver, we pass the eigen solver
opt_newton = BK.NewtonPar(tol = 1e-8, eigsolver = eigls, max_iterations = 20)
# options for continuation
opts_br = ContinuationPar(p_max = 3.5, p_min = 0.025,
# for a good looking curve
dsmin = 0.001, dsmax = 0.05, ds = 0.01,
# number of eigenvalues to compute
nev = 30,
newton_options = (@set opt_newton.verbose = false),
tol_stability = 1e-6,
# detect codim 1 bifurcations
detect_bifurcation = 3,
# Optional: bisection options for locating bifurcations
n_inversion = 4, dsmin_bisection = 1e-7, max_bisection_steps = 25)
Note that we put the option detect_bifurcation = 3
to detect bifurcations precisely with a bisection method. Indeed, we need to locate these branch points precisely to be able to call automatic branch switching.
Branch of homogeneous solutions
At this stage, we note that the problem has a curve of homogeneous (constant in space) solutions $u_h$ solving $N(\lambda, u_h)=0$. We shall compute this branch now.
Given that we will use these arguments for continuation
many times, it is wise to collect them:
# optional arguments for continuation
kwargsC = (verbosity = 0, plot = true, normC = norminf)
We call continuation
with the initial guess sol0
which is homogeneous, thereby generating homogeneous solutions:
br = continuation(prob, PALC(), opts_br; kwargsC...)
show(br)
GKS: Possible loss of precision in routine SET_WINDOW
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
GKS: Rectangle definition is invalid in routine CELLARRAY
āā Curve type: EquilibriumCont
āā Number of points: 84
āā Type of vectors: Vector{Float64}
āā Parameter Ī» starts at 0.05, ends at 0.025
āā Algo: PALC
āā Special points:
- # 1, bp at Ī» ā +0.36787944 ā (+0.36787944, +0.36787944), |Ī“p|=2e-10, [converged], Ī“ = ( 1, 0), step = 18
- # 2, nd at Ī» ā +0.27255474 ā (+0.27255474, +0.27255937), |Ī“p|=5e-06, [converged], Ī“ = ( 2, 0), step = 33
- # 3, bp at Ī» ā +0.15215124 ā (+0.15215124, +0.15215818), |Ī“p|=7e-06, [converged], Ī“ = ( 1, 0), step = 48
- # 4, nd at Ī» ā +0.03551852 ā (+0.03551852, +0.03554981), |Ī“p|=3e-05, [converged], Ī“ = ( 2, 0), step = 76
- # 5, endpoint at Ī» ā +0.02500000, step = 83
You should see the following result:
title!("")
We note several simple bifurcation points for which the dimension of the kernel of the jacobian is one dimensional. In the above box, Ī“ = ( 1, 0)
gives the change in the stability. In this case, there is one vector in the kernel which is real. The bifurcation point 2 has a 2d kernel and is thus not amenable to automatic branch switching.
Automatic branch switching at simple branch points
We can compute the branch off the third bifurcation point:
br1 = continuation(br, 3, setproperties(opts_br;ds = 0.001, max_steps = 40); kwargsC...)
title!("")
You can also plot the two branches together plot(br, br1, plotfold=false)
and get
scene = plot(br,br1,plotfold=false)
We continue our journey and compute the branch bifurcating of the first bifurcation point from the last branch we computed:
br2 = continuation(br1, 1, setproperties(opts_br;ds = 0.001, max_steps = 40); kwargsC...)
scene = plot(br, br1, br2)
Automatic branch switching at the 2d-branch points
We now show how to perform automatic branch switching at the nonsimple branch points. However, we think it is important that the user is able to use the previous tools in case automatic branch switching fails. This is explained in the next sections.
The call for automatic branch switching is the same as in the case of simple branch points (see above) except that many branches are returned.
branches = continuation(br, 2,
setproperties(opts_br; detect_bifurcation = 3, ds = 0.001, p_min = 0.01, max_steps = 30 ) ;
alg = PALC(tangent = Bordered()),
kwargsC...,
nev = 30,
)
You can plot the branches using
scene = plot(br, branches...)
Analysis at the 2d-branch points (manual)
The second bifurcation point on the branch br
of homogeneous solutions has a 2d kernel. we provide two methods to deal with such case
- automatic local bifurcation diagram (see below)
- branch switching with deflation (see next section)
We provide a generic way to study branch points of arbitrary dimensions by computing a reduced equation. The general method is based on a Lyapunov-Schmidt reduction. We can compute the information about the branch point using the generic function (valid for simple branch points, Hopf bifurcation points,...)
bp2d = get_normal_form(br, 2; verbose = true, nev = 50)
Non simple bifurcation point at Ī» ā 0.2725547358342202.
Kernel dimension = 2
Normal form:
+ -73.8978 * x1 ā
p + -0.0084 ā
x1Ā³ + 0.0273 ā
x1Ā² ā
x2 + 0.0201 ā
x1 ā
x2Ā² + -0.009 ā
x2Ā³
+ -73.8978 * x2 ā
p + 0.0092 ā
x1Ā³ + 0.0195 ā
x1Ā² ā
x2 + -0.0273 ā
x1 ā
x2Ā² + -0.0086 ā
x2Ā³
Note that this is a multivariate polynomials. For more information, see Non-simple branch point.
You can evaluate this polynomial as follows bp2d(Val(:reducedForm),[0.1,0.2], 0.01)
which returns a 2d vector or bp2d([0.1,0.2], 0.01)
. This last expression actually returns a vector corresponding to the PDE problem.
You need to solve these equations to compute the bifurcation diagram in the neighborhood of the bifurcation point. In the present case, we do it using brute force. We suggest to use IntervalConstraintProgramming.jl
for a more precise way.
using ProgressMeter
Nd = 200; L = 0.9
# sampling grid
X = LinRange(-L,L, Nd); Y = LinRange(-L,L, Nd); P = LinRange(-0.0001,0.0001, Nd+1)
# sample reduced equation on the grid for the first component
V1a = @showprogress [bp2d(Val(:reducedForm),[x1,y1], p1)[1] for p1 in P, x1 in X, y1 in Y]
Ind1 = findall( abs.(V1a) .<= 9e-4 * maximum(abs.(V1a)))
# intersect with second component
V2a = @showprogress [bp2d(Val(:reducedForm),[X[ii[2]],Y[ii[3]]], P[ii[1]])[2] for ii in Ind1]
Ind2 = findall( abs.(V2a) .<= 3e-3 * maximum(abs.(V2a)))
# get solutions
resp = Float64[]; resx = Vector{Float64}[]; resnrm = Float64[]
@showprogress for k in Ind2
ii = Ind1[k]
push!(resp, P[ii[1]])
push!(resnrm, sqrt(X[ii[2]]^2+Y[ii[3]]^2))
push!(resx, [X[ii[2]], Y[ii[3]]])
end
We can now plot the local bifurcation diagram as follows
using LaTeXStrings
plot(
scatter(1e4resp, map(x->x[1], resx), map(x->x[2], resx); label = "", markerstrokewidth=0, xlabel = L"10^4 \cdot \lambda", ylabel = L"x_1", zlabel = L"x_2", zcolor = resnrm, color = :viridis,colorbar=false),
scatter(1e4resp, resnrm; label = "", markersize =2, markerstrokewidth=0, xlabel = L"10^4 \cdot \lambda", ylabel = L"\|x\|"))
This looks like a Pitchfork bifurcation with D4 symmetry
We can see that there are two types of solutions. After the bifurcation point, the solutions are of the form $(x_1,x_2) = (\pm x,\pm x)$ for some real $x$. Before the bifurcation point, the solutions are of the form $(x_1,x_2) = (\pm x,0), (0, \pm x)$ for some real $x$. Here is an example plotsol(bp2d(resx[10], resp[10]))
We could use the solutions saved in resp, resx
as initial guesses for a call to continuation
but we turn to a different method.
The brute force method provided all solutions in a neighborhood of the bifurcation point.
Instead of using brute force and computing the vector field on a grid. One can rely on IntervalConstraintProgramming.jl
to do better using bisection. See also this discourse post where the same example is treated by D. P. Sanders.
Branch switching with deflated newton (manual)
At this stage, we know what happens at the 2d bifurcation point of the curve of homogeneous solutions. We chose another method based on Deflated problems. We want to find all nearby solutions of the problem close to this bifurcation point. This is readily done by trying several initial guesses in a brute force manner:
out = zeros(Nx*Ny)
# deflation operator to
deflationOp = DeflationOperator(2, 1.0, [zeros(Nx*Ny)])
# options for the newton solver
optdef = setproperties(opt_newton; tol = 1e-8, max_iterations = 100)
# eigen-elements close to the second bifurcation point on the branch
# of homogeneous solutions
vp, ve, _, _= eigls(JFmit(out, @set par_mit.Ī» = br.specialpoint[2].param), 5)
for ii=1:length(ve)
outdef1 = newton(
re_make(prob,
# initial guess for newton
u0 = br.specialpoint[2].x .+ 0.01 .* real.(ve[ii]) .* (1 .+ 0.01 .* rand(Nx*Ny)),
params = (@set par_mit.Ī» = br.specialpoint[2].param + 0.005)),
deflationOp,
optdef)
BK.converged(outdef1) && push!(deflationOp, outdef1.u)
end
This provides length(deflationOp) = 5
solutions as there are some symmetries in the problem. For example plotsol(deflationOp[5])
gives
We can continue this solution as follows in one direction
brdef1 = continuation(
re_make(prob,
u0 = deflationOp[3],
params = (@set par_mit.Ī» = br.specialpoint[2].param + 0.005)),
PALC(),
setproperties(opts_br;ds = -0.001, detect_bifurcation = 3, dsmax = 0.01, max_steps = 500);
kwargsC...)
If we repeat the above loop but before the branch point by using @set par_mit.Ī» = br.specialpoint[2].param + 0.005
, we get 3 new solutions that we can continue
brdef2 = continuation(
re_make(prob,
u0 = deflationOp[5],
params = (@set par_mit.Ī» = br.specialpoint[2].param + 0.005)),
PALC(),
setproperties(opts_br;ds = 0.001, detect_bifurcation = 3, dsmax = 0.01);
kwargsC...)
thereby providing the following bifurcation diagram with plot(br,br1,br2,brdef1, brdef2,plotfold=false, putbifptlegend = false)