Differentiation matrices
Contents
10.3. Differentiation matrices#
In Section 5.4 we used finite differences to turn a discrete collection of function values into an estimate of the derivative of the function at a point. Just as with differentiation in elementary calculus, we can generalize differences at a point into an operation that maps discretized functions to discretized derivatives.
Matrices for finite differences#
We first discretize the interval \(x\in[a,b]\) into equal pieces of length \(h=(b-a)/n\), leading to the nodes
Note again that our node indexing scheme starts at zero. Our goal is to find a vector \(\mathbf{g}\) such that \(g_i \approx f'(x_i)\) for \(i=0,\ldots,n\). Our first try is the forward difference formula (5.4.2),
However, this leaves \(g_n\) undefined, because the formula would refer to the unavailable value \(f_{n+1}\). For \(g_n\) we could resort to the backward difference
We can summarize the entire set of formulas by defining
and then the vector equation
Here as elsewhere, elements of \(\mathbf{D}_x\) that are not shown are zero. We call \(\mathbf{D}_x\) a differentiation matrix. Each row of \(\mathbf{D}_x\) gives the weights of the finite-difference formula being used at one of the nodes.
The differentiation matrix \(\mathbf{D}_x\) in (10.3.1) is not a unique choice. We are free to use whatever finite-difference formulas we like in each row. However, it makes sense to choose rows that are as similar as possible. Using second-order centered differences where possible and second-order one-sided formulas (see Table 5.4.2) at the boundary points leads to
The differentiation matrices so far are banded matrices, i.e., all the nonzero values are along diagonals close to the main diagonal.1
Second derivative#
Similarly, we can define differentiation matrices for second derivatives. For example,
We have multiple choices again for \(\mathbf{D}_{xx}\), and it need not be the square of any particular \(\mathbf{D}_x\). As pointed out in Section 5.4, squaring the first derivative is a valid approach but would place entries in \(\mathbf{D}_{xx}\) farther from the diagonal than is necessary.
Second-order accurate differentiation matrices
1"""
2 diffmat2(n,xspan)
3
4Compute 2nd-order-accurate differentiation matrices on `n`+1 points
5in the interval `xspan`. Returns a vector of nodes and the matrices
6for the first and second derivatives.
7"""
8function diffmat2(n,xspan)
9 a,b = xspan
10 h = (b-a)/n
11 x = [ a + i*h for i in 0:n ] # nodes
12
13 # Define most of Dₓ by its diagonals.
14 dp = fill(0.5/h,n) # superdiagonal
15 dm = fill(-0.5/h,n) # subdiagonal
16 Dₓ = diagm(-1=>dm,1=>dp)
17
18 # Fix first and last rows.
19 Dₓ[1,1:3] = [-1.5,2,-0.5]/h
20 Dₓ[n+1,n-1:n+1] = [0.5,-2,1.5]/h
21
22 # Define most of Dₓₓ by its diagonals.
23 d0 = fill(-2/h^2,n+1) # main diagonal
24 dp = ones(n)/h^2 # super- and subdiagonal
25 Dₓₓ = diagm(-1=>dp,0=>d0,1=>dp)
26
27 # Fix first and last rows.
28 Dₓₓ[1,1:4] = [2,-5,4,-1]/h^2
29 Dₓₓ[n+1,n-2:n+1] = [-1,4,-5,2]/h^2
30
31 return x,Dₓ,Dₓₓ
32end
Together the matrices (10.3.2) and (10.3.3) give second-order approximations of the first and second derivatives at all nodes. These matrices, as well as the nodes \(x_0,\ldots,x_n\), are returned by Function 10.3.1.
We test first-order and second-order differentiation matrices for the function \(x + \exp(\sin 4x)\) over \([-1,1]\).
f = x -> x + exp(sin(4*x));
For reference, here are the exact first and second derivatives.
dfdx = x -> 1 + 4*exp(sin(4*x)) * cos(4*x);
d2fdx2 = x -> 4*exp(sin(4*x)) * (4*cos(4*x)^2-4*sin(4*x));
We discretize on equally spaced nodes and evaluate \(f\) at the nodes.
t,Dₓ,Dₓₓ = FNC.diffmat2(18,[-1,1])
y = f.(t);
Then the first two derivatives of \(f\) each require one matrix-vector multiplication.
yₓ = Dₓ*y;
yₓₓ = Dₓₓ*y;
The results show poor accuracy for this small value of \(n\).
plot(dfdx,-1,1,layout=2,xaxis=(L"x"),yaxis=(L"f'(x)"))
scatter!(t, yₓ,subplot=1)
plot!(d2fdx2,-1,1,subplot=2,xaxis=(L"x"),yaxis=(L"f''(x)"))
scatter!(t, yₓₓ,subplot=2)
An convergence experiment confirms the order of accuracy. Because we expect an algebraic convergence rate, we use a log-log plot of the errors.
n = @. round(Int,2^(4:.5:11) )
err1 = zeros(size(n))
err2 = zeros(size(n))
for (k,n) in enumerate(n)
t,Dₓ,Dₓₓ = FNC.diffmat2(n,[-1,1])
y = f.(t)
err1[k] = norm( dfdx.(t) - Dₓ*y, Inf )
err2[k] = norm( d2fdx2.(t) - Dₓₓ*y, Inf )
end
plot(n,[err1 err2],m=:o,label=[L"f'" L"f''"])
plot!(n,10*10*n.^(-2),l=(:dash,:gray),label="2nd order",
xaxis=(:log10,"n"), yaxis=(:log10,"max error"),
title="Convergence of finite differences")
Spectral differentiation#
Recall that finite-difference formulas are derived in three steps:
Choose a node index set \(S\) near node \(i\).
Interpolate with a polynomial using the nodes in \(S\).
Differentiate the interpolant and evaluate at node \(i\).
We can modify this process by using a global interpolant, either polynomial or trigonometric, as in Chapter 9. Rather than choosing a different index set for each node, we use all of the nodes each time.
In a nonperiodic setting we use Chebyshev second-kind points for stability:
then the resulting Chebyshev differentiation matrix has entries
where \(c_0=c_n=2\) and \(c_i=1\) for \(i=1,\ldots,n-1\). Note that this matrix is dense. The simplest way to compute a second derivative is by squaring \(\mathbf{D}_x\), as there is no longer any concern about the bandwidth of the result.
Function 10.3.3 returns these two matrices. The function uses a change of variable to transplant the standard \([-1,1]\) for Chebyshev nodes to any \([a,b]\). It also takes a different approach to computing the diagonal elements of \(\mathbf{D}_x\) than the formulas in (10.3.4) (see Exercise 5).
Chebyshev differentiation matrices
1"""
2 diffcheb(n,xspan)
3
4Compute Chebyshev differentiation matrices on `n`+1 points in the
5interval `xspan`. Returns a vector of nodes and the matrices for the
6first and second derivatives.
7"""
8function diffcheb(n,xspan)
9 x = [ -cos( k*π/n ) for k in 0:n ] # nodes in [-1,1]
10
11 # Off-diagonal entries.
12 c = [2; ones(n-1); 2]; # endpoint factors
13 dij = (i,j) -> (-1)^(i+j)*c[i+1]/(c[j+1]*(x[i+1]-x[j+1]))
14 Dₓ = [ dij(i,j) for i in 0:n, j in 0:n ]
15
16 # Diagonal entries.
17 Dₓ[isinf.(Dₓ)] .= 0 # fix divisions by zero on diagonal
18 s = sum(Dₓ,dims=2)
19 Dₓ -= diagm(s[:,1]) # "negative sum trick"
20
21 # Transplant to [a,b].
22 a,b = xspan
23 x = @. a + (b-a)*(x+1)/2
24 Dₓ = 2*Dₓ/(b-a) # chain rule
25
26 # Second derivative.
27 Dₓₓ = Dₓ^2
28 return x,Dₓ,Dₓₓ
29end
Here is a \(4\times 4\) Chebyshev differentiation matrix.
t,Dₓ = FNC.diffcheb(3,[-1,1])
Dₓ
4×4 Matrix{Float64}:
-3.16667 4.0 -1.33333 0.5
-1.0 0.333333 1.0 -0.333333
0.333333 -1.0 -0.333333 1.0
-0.5 1.33333 -4.0 3.16667
We again test the convergence rate.
f = x -> x + exp(sin(4*x));
dfdx = x -> 1 + 4*exp(sin(4*x))*cos(4*x);
d2fdx2 = x -> 4*exp(sin(4*x))*(4*cos(4*x)^2-4*sin(4*x));
n = 5:5:70
err1 = zeros(size(n))
err2 = zeros(size(n))
for (k,n) in enumerate(n)
t,Dₓ,Dₓₓ = FNC.diffcheb(n,[-1,1])
y = f.(t)
err1[k] = norm( dfdx.(t) - Dₓ*y, Inf )
err2[k] = norm( d2fdx2.(t) - Dₓₓ*y, Inf )
end
Since we expect a spectral convergence rate, we use a semi-log plot for the error.
plot(n,[err1 err2],m=:o,label=[L"f'" L"f''"],
xaxis=(L"n"), yaxis=(:log10,"max error"),
title="Convergence of Chebyshev derivatives")
According to Theorem 9.3.6, the convergence of polynomial interpolation to \(f\) using Chebyshev nodes is spectral if \(f\) is analytic (at least having infinitely many derivatives) on the interval. The derivatives of \(f\) are also approximated with spectral accuracy.
Exercises#
(a) ✍ Calculate \(\mathbf{D}_x^2\) using (10.3.1) to define \(\mathbf{D}_x\).
(b) ⌨ Repeat the experiment of Demo 10.3.2, but using this version of \(\mathbf{D}_x^2\) to estimate \(f''\). What is the apparent order of accuracy in \(f''\)?
(a) ✍ Find the derivative of \(f(x) =\operatorname{sign}(x)x^2\) on the interval \([-1,1]\). (If this gives you trouble, use an equivalent piecewise definition of \(f\).) What is special about this function at \(x=0\)?
(b) ⌨ Adapt Demo 10.3.2 to operate on the function from part (a), computing only the first derivative. What is the observed order of accuracy?
(c) ✍ Show that for even values of \(n\), there is only one node at which the error for computing \(f'\) in part (b) is nonzero.
⌨ To get a fourth-order accurate version of \(\mathbf{D}_x\), five points per row are needed, including two special rows at each boundary. For a fourth-order \(\mathbf{D}_{xx}\), five symmetric points per row are needed for interior rows and six points are needed for the rows near a boundary.
(a) Modify Function 10.3.1 to a function
diffmat4
, which outputs fourth-order accurate differentiation matrices. You may want to use Function 5.4.7.(b) Repeat the experiment of Demo 10.3.2 using
diffmat4
in place of Function 10.3.1, and compare observed errors to fourth-order accuracy.✍ Explain in detail how lines 23-24 in Function 10.3.3 correctly change the interval from \([-1,1]\) to \([a,b]\).
(a) ✍ What is the derivative of a constant function?
(b) ✍ Explain why for any reasonable differentiation matrix \(\mathbf{D}\), we should find \(\displaystyle \sum_{j=0}^nD_{ij}=0\) for all \(i\).
(c) ✍ What does this have to do with Function 10.3.3? Refer to specific line(s) in the function for your answer.
Define the \((n+1)\times (n+1)\) matrix \(\mathbf{T} = \displaystyle \begin{bmatrix} 1 & & & \\ 1 & 1 & & \\ \vdots & & \ddots & \\ 1 & 1 & \cdots & 1 \end{bmatrix}\).
(a) ✍ Write out \(\mathbf{T}\mathbf{u}\) for a generic vector \(\mathbf{u}\) (start with a zero index). How is this like integration?
(b) ✍ Find the inverse of \(\mathbf{T}\) for any \(n\). (You can use Julia to find the pattern, but show that the result is correct in general.) What does this have to do with the inverse of integration?
- 1
In order to exploit this structure efficiently in Julia, these matrices first need to be constructed as or converted to sparse or tridiagonal form.