Skip to content

Commit

Permalink
Add tutorial about 1D interpolation, shape parameter, and differentia…
Browse files Browse the repository at this point in the history
…tion (#66)

* add tutorial about 1D interpolation, shape parameter, and differentiation

* more precise intro

* linebreak

* fix at-example

* some small fixes and additions

* fix rendering

* fix rendering
  • Loading branch information
JoshuaLampert authored Jul 20, 2024
1 parent ea69054 commit 4fd33da
Show file tree
Hide file tree
Showing 4 changed files with 136 additions and 2 deletions.
1 change: 1 addition & 0 deletions docs/Project.toml
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
[deps]
Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
Meshes = "eacbb407-ea5a-433e-ab97-5258b1ca43fa"
Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
QuasiMonteCarlo = "8a4e6c94-4038-4cdc-81c3-7e6ffdb2a71b"
Expand Down
2 changes: 2 additions & 0 deletions docs/make.jl
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ makedocs(;
"Sets of nodes" => "nodesets.md",
"Interpolation" => "interpolation.md",
"Solving PDEs by collocation" => "pdes.md"],
"Tutorials" => [
"1D interpolation and differentiation" => "tutorial_differentiating_interpolation.md"],
"Development" => "development.md",
"Reference" => "ref.md",
"License" => "license.md"])
Expand Down
131 changes: 131 additions & 0 deletions docs/src/tutorial_differentiating_interpolation.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,131 @@
# One-dimensional interpolation and differentiation

In this tutorial, we will create a simple one-dimensional interpolation, investigate how to tune the interpolation method,
and show how to apply differential operators on the resulting [`Interpolation`](@ref) object.

## Define problem setup and perform interpolation

We start by defining a simple one-dimensional interpolation problem. We will interpolate the oscillatory function

```math
f(x) = \exp(\sin(2x^2)) + 0.1(x - \pi/2)^2
```

between ``x = -3`` and ``x = 3``. For simplicity, we take 25 equidistant points in the interval ``[-3, 3]``
as interpolation points.

```@example diff-itp
using KernelInterpolation
f(x) = exp(sin(2*x[1]^2)) + 0.1*(x[1] - pi/2)^2
x_min = -3
x_max = 3
N = 25
nodeset = NodeSet(LinRange(x_min, x_max, N))
values = f.(nodeset)
```

Next, we choose the kernel (radial basis function) for the interpolation. We use the Gaussian kernel with a fixed
shape parameter of 0.5 and interpolate the function values.

```@example diff-itp
kernel = GaussKernel{1}(shape_parameter = 0.5)
itp = interpolate(nodeset, values, kernel)
```

Let's plot the interpolated function and the original function on a finer grid to see how well the interpolation works.

```@example diff-itp
using Plots
many_nodes = NodeSet(LinRange(x_min, x_max, 200))
plot(many_nodes, f, label = "Original function")
plot!(many_nodes, itp)
savefig("interpolation_oscillatory.png") # hide
nothing # hide
```

![Interpolation of one-dimensional oscillatory function](interpolation_oscillatory.png)

Uhh, that doesn't look too good. What happened?

## Finding a well-suited interpolation method

We used the [`GaussKernel`](@ref) with a rather small shape parameter of 0.5, which leads to an ill-conditioned
linear system of equations. We can inspect the condition number of the interpolation matrix to confirm this.

```@example diff-itp
using LinearAlgebra
A = system_matrix(itp)
cond(A)
```

Here, we used the [`system_matrix`](@ref) function to obtain the interpolation matrix `A` and calculated the condition number
of the matrix. For this specific example the system matrix simply is the [`kernel_matrix`](@ref), but for more sophisticated
interpolations the system matrix contains additional parts like the polynomial augmentation. The condition number is a measure
of how well-conditioned the matrix is. A large condition number indicates that the matrix is ill-conditioned, which usually
leads to high numerical errors. To avoid this, we have different options. We can either increase the shape parameter of the
kernel or we can use a different kernel. The [`GaussKernel`](@ref) is known to be rather ill-conditioned and other kernels like
the [`WendlandKernel`](@ref) usually lead to better condition numbers.
Here, we choose to increase the shape parameter of the Gaussian kernel to 1.8, which makes the interpolation more localized.
Note, however, that you might need to choose another kernel if you increase the number of interpolation points.

```@example diff-itp
kernel = GaussKernel{1}(shape_parameter = 1.8)
itp = interpolate(nodeset, values, kernel)
plot(many_nodes, f, label = "Original function")
plot!(many_nodes, itp)
savefig("interpolation_oscillatory_1_5.png") # hide
nothing # hide
```

![Interpolation of one-dimensional oscillatory function with higher shape parameter](interpolation_oscillatory_1_5.png)

We can see a much better agreement between the original function and the interpolated function. We still observe some
undershoots, but this is expected due to the oscillatory nature of the function and the limited number of interpolation
points. Let's confirm that increasing the shape parameter improved the condition number of the interpolation matrix.

```@example diff-itp
A = system_matrix(itp)
cond(A)
```

Indeed, the condition number is much smaller than before!

## Applying differential operators

Sometimes, we are not only interested in interpolating a function, but also in computing its derivatives. Remember that
in the simplest case, where no polynomial augmentation is used, the interpolation `itp` represents a linear combination

```math
s(x) = \sum_{j = 1}^N c_j\phi(\|x - x_j\|_2)
```

with ``\phi`` given by the radial basis function, in this case the Gaussian. Because we know ``\phi`` and its derivatives,
we can compute the derivatives of ``s`` by differentiating the kernel function. For a general dimension ``d``, the partial
derivative in the ``i``-th direction, ``i\in\{1,\ldots,d\}``, of the interpolation is then given by

```math
\frac{\partial s}{\partial x_i}(x) = \sum_{j = 1}^N c_j\frac{\partial \phi}{\partial x_i}(\|x - x_j\|_2).
```

!!! note
Although the derivatives of the kernel functions could be computed analytically, KernelInterpolation.jl uses automatic
differentiation (AD) by using [ForwardDiff.jl](https://github.com/JuliaDiff/ForwardDiff.jl). This allows for flexibility,
simplicity, and easier extension, but it might be slower than computing the derivatives analytically.

KernelInterpolation.jl already provides some common differential operators. For example, we can compute the first derivative
of the interpolation `itp` at a specific point `x` by using the [`PartialDerivative`](@ref) operator.

```@example diff-itp
d1 = PartialDerivative(1)
x = 0.0
itp_dx = d1(itp, x)
```

Let's plot the first derivative of the interpolated function and compare it to the analytical first derivative.

```@example diff-itp
itp_dx_many_nodes = d1.(Ref(itp), many_nodes)
f_dx(x) = 4*exp(sin(2*x[1]^2))*x[1]*cos(2*x[1]^2) + 0.2*x[1] - pi/10
plot(many_nodes, f_dx, label = "Derivative of original function")
plot!(many_nodes, itp_dx_many_nodes, label = "Derivative of interpolated function")
```
4 changes: 2 additions & 2 deletions src/differential_operators.jl
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,8 @@ Partial derivative operator with respect to the `i`-th component.
The operator can be called with a [`RadialSymmetricKernel`](@ref) and points
`x` and `y` to evaluate the derivative of the `kernel` at `x - y`.
It can also be called with an [`Interpolation`](@ref) object and a point `x` to evaluate
the `i`-th partial derivative of the interpolation at `x`. Note that this is only supported
for the kernel part of the interpolation, i.e. the polynomial part, if existent, is ignored.
the first partial derivative of the interpolation at `x` in the `i`-th direction. Note that this
is only supported for the kernel part of the interpolation, i.e. the polynomial part, if existent, is ignored.
"""
struct PartialDerivative <: AbstractDifferentialOperator
i::Int
Expand Down

0 comments on commit 4fd33da

Please sign in to comment.