Polarizability by linear response

We compute the polarizability of a Helium atom. The polarizability is defined as the change in dipole moment

\[\mu = \int r ρ(r) dr\]

with respect to a small uniform electric field $E = -x$.

We compute this in two ways: first by finite differences (applying a finite electric field), then by linear response. Note that DFTK is not really adapted to isolated atoms because it uses periodic boundary conditions. Nevertheless we can simply embed the Helium atom in a large enough box (although this is computationally wasteful).

As in other tests, this is not fully converged, convergence parameters were simply selected for fast execution on CI,

using DFTK
using LinearAlgebra

a = 10.
lattice = a * I(3)  # cube of ``a`` bohrs
He = ElementPsp(:He, psp=load_psp("hgh/lda/He-q2"))
atoms = [He => [[1/2; 1/2; 1/2]]]  # Helium at the center of the box

kgrid = [1, 1, 1]  # no k-point sampling for an isolated system
Ecut = 30
tol = 1e-8

# dipole moment of a given density (assuming the current geometry)
function dipole(basis, ρ)
    rr = [a * (r[1] - 1/2) for r in r_vectors(basis)]
    sum(rr .* ρ) * basis.dvol
end;

Using finite differences

We first compute the polarizability by finite differences. First compute the dipole moment at rest:

model = model_LDA(lattice, atoms; symmetries=false)
basis = PlaneWaveBasis(model; Ecut, kgrid)
res   = self_consistent_field(basis, tol=tol)
μref  = dipole(basis, res.ρ)
-0.00013496489795900243

Then in a small uniform field:

ε = .01
model_ε = model_LDA(lattice, atoms; extra_terms=[ExternalFromReal(r -> -ε * (r[1] - a/2))],
                    symmetries=false)
basis_ε = PlaneWaveBasis(model_ε; Ecut, kgrid)
res_ε = self_consistent_field(basis_ε, tol=tol)
με = dipole(basis_ε, res_ε.ρ)
0.017617868955543712
polarizability = (με - μref) / ε

println("Reference dipole:  $μref")
println("Displaced dipole:  $με")
println("Polarizability :   $polarizability")
Reference dipole:  -0.00013496489795900243
Displaced dipole:  0.017617868955543712
Polarizability :   1.7752833853502714

The result on more converged grids is very close to published results. For example DOI 10.1039/C8CP03569E quotes 1.65 with LSDA and 1.38 with CCSD(T).

Using linear response

Now we use linear response to compute this analytically; we refer to standard textbooks for the formalism. In the following, $\chi_0$ is the independent-particle polarizability, and $K$ the Hartree-exchange-correlation kernel. We denote with $\delta V_{\rm ext}$ an external perturbing potential (like in this case the uniform electric field). Then:

\[\delta\rho = \chi_0 \delta V = \chi_0 (\delta V_{\rm ext} + K \delta\rho),\]

which implies

\[\delta\rho = (1-\chi_0 K)^-1 \chi_0 \delta V_{\rm ext}.\]

From this we identify the polarizability operator to be $\chi = (1-\chi_0 K)^{-1} \chi_0$. Numerically, we apply $\chi$ to $\delta V = -x$ by solving a linear equation (the Dyson equation) iteratively.

using KrylovKit

# Apply (1- χ0 K)
function dielectric_operator(δρ)
    δV = apply_kernel(basis, δρ; ρ=res.ρ)
    χ0δV = apply_χ0(res.ham, res.ψ, res.εF, res.eigenvalues, δV)
    δρ - χ0δV
end

# δVext is the potential from a uniform field interacting with the dielectric dipole
# of the density.
δVext = [-a * (r[1] - 1/2) for r in r_vectors(basis)]
δVext = cat(δVext; dims=4)

# Apply χ0 once to get non-interacting dipole
δρ_nointeract = apply_χ0(res.ham, res.ψ, res.εF, res.eigenvalues, δVext)

# Solve Dyson equation to get interacting dipole
δρ = linsolve(dielectric_operator, δρ_nointeract, verbosity=3)[1]

println("Non-interacting polarizability: $(dipole(basis, δρ_nointeract))")
println("Interacting polarizability:     $(dipole(basis, δρ))")
WARNING: using KrylovKit.basis in module ex-polarizability conflicts with an existing identifier.
[ Info: GMRES linsolve in iter 1; step 1: normres = 2.494336696913e-01
[ Info: GMRES linsolve in iter 1; step 2: normres = 3.767805969009e-03
[ Info: GMRES linsolve in iter 1; step 3: normres = 3.044287255901e-04
[ Info: GMRES linsolve in iter 1; step 4: normres = 4.882082463449e-06
[ Info: GMRES linsolve in iter 1; step 5: normres = 2.817752745021e-07
[ Info: GMRES linsolve in iter 1; step 6: normres = 2.023508907929e-09
[ Info: GMRES linsolve in iter 1; step 7: normres = 2.288256695107e-11
[ Info: GMRES linsolve in iter 1; step 8: normres = 8.411037746206e-13
[ Info: GMRES linsolve in iter 1; finished at step 8: normres = 8.411037746206e-13
[ Info: GMRES linsolve in iter 2; step 1: normres = 7.058715434233e-07
[ Info: GMRES linsolve in iter 2; step 2: normres = 8.488592196039e-08
[ Info: GMRES linsolve in iter 2; step 3: normres = 1.620287526952e-09
[ Info: GMRES linsolve in iter 2; step 4: normres = 3.038868490078e-11
[ Info: GMRES linsolve in iter 2; step 5: normres = 4.321830615590e-13
[ Info: GMRES linsolve in iter 2; finished at step 5: normres = 4.321830615590e-13
[ Info: GMRES linsolve in iter 3; step 1: normres = 4.934656618738e-11
[ Info: GMRES linsolve in iter 3; step 2: normres = 6.138693868253e-12
[ Info: GMRES linsolve in iter 3; step 3: normres = 5.286673689063e-14
[ Info: GMRES linsolve in iter 3; finished at step 3: normres = 5.286673689063e-14
┌ Info: GMRES linsolve converged at iteration 3, step 3:
│ *  norm of residual = 5.356934151567308e-14
└ *  number of operations = 18
Non-interacting polarizability: 1.9263338894385105
Interacting polarizability:     1.7742250062484572

As expected, the interacting polarizability matches the finite difference result. The non-interacting polarizability is higher.