简体   繁体   中英

Compute eigenvalues of complex-hermitian sparsematrix in Julia

Im working with some roughly 100000x100000 hermitian complex sparse-matrices , with roughly 5% of entries populated, and want to calculate the eigenvalues/eigenvectors.

Sofar ive been using Arpack.jl eigs(A) . But this is not working well as soon as i crank the size to higher then 5000.

For the benchmarks ive been using the following code to generate some TestMatrices:

using Arpack
using SparseArrays
using ProgressMeter

pop = 0.05
n = 3000 # for example

A = spzeros(Complex{Float64}, n, n)

@showprogress for _ in 1:round(Int,pop * (n^2))
  A[rand(1:n), rand(1:n)] = rand(Complex{Float64})
end


# make A hermite
A = A + conj(A)

t = @elapsed eigs(A,maxiter=1500) # ends up being ~ 13 seconds

For n ~ 3000 the eigs() call already takes 13 seconds on my machine, and for bigger n it doesn't finish in any 'reasonable' time or outright quits.

Is there a specialized package/method for this?

Any help is appreciated

https://github.com/JuliaLinearAlgebra/ArnoldiMethod.jl seems to be what you want:

julia> let pop=0.05, n=3000
           A = sprand(Complex{Float64},n,n, 0.05)
           A = A + conj(A)
           @time eigs(A; maxiter=1500)
           @time decomp, history = partialschur(A, nev=10, tol=1e-6, which=LM());
       end;
 10.521786 seconds (50.73 k allocations: 3.458 MiB, 0.04% gc time)
  2.244129 seconds (19 allocations: 1.892 MiB)

sanity check:

julia> a,(b,c) = let pop=0.05, n=300
           A = sprand(Complex{Float64},n,n, 0.05)
           A = A + conj(A)
           eigs(A; maxiter=2500), partialschur(A, nev=6, tol=1e-6, which=LM());
       end;

julia> a[1]
6-element Vector{ComplexF64}:
    14.5707071003175 + 8.218901803015509e-16im
   4.493079744504954 - 0.8390429567118733im
   4.493079744504933 + 0.8390429567118641im
 -0.3415176925293196 + 4.254184281244591im
 -0.3415176925293088 - 4.25418428124452im
 0.49406553681541177 - 4.229680489599233im

julia> b
PartialSchur decomposition (ComplexF64) of dimension 6
eigenvalues:
6-element Vector{ComplexF64}:
   14.570707100307926 + 7.10698633463049e-12im
    4.493079906269516 + 0.8390429076809746im
    4.493079701528448 - 0.8390430155670777im
  -0.3415174262177961 + 4.254183175902487im
 -0.34151626930774975 - 4.25418321627979im
     0.49406543866702 + 4.229680079205066im

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM