iterativesolvers.h
//
// Copyright 2018 The Simons Foundation, Inc. - All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#ifndef __ITENSOR_ITERATIVESOLVERS_H
#define __ITENSOR_ITERATIVESOLVERS_H
#include "itensor/util/iterate.h"
#include "itensor/itensor.h"
#include "itensor/tensor/algs.h"
namespace itensor {
//
// Use the Davidson algorithm to find the
// eigenvector of the Hermitian matrix A with minimal eigenvalue.
// (BigMatrixT objects must implement the methods product, size and diag.)
// Returns the minimal eigenvalue lambda such that
// A phi = lambda phi.
//
template <class BigMatrixT>
Real
davidson(BigMatrixT const& A,
ITensor& phi,
Args const& args = Args::global());
//
// Use Davidson to find the N eigenvectors with smallest
// eigenvalues of the Hermitian matrix A, given a vector of N
// initial guesses (zero indexed).
// (BigMatrixT objects must implement the methods product, size and diag.)
// Returns a vector of the N smallest eigenvalues corresponding
// to the set of eigenvectors phi.
//
template <class BigMatrixT>
std::vector<Real>
davidson(BigMatrixT const& A,
std::vector<ITensor>& phi,
Args const& args = Args::global());
//
// Use GMRES to iteratively solve A x = b for x.
// (BigMatrixT objects must implement the methods product and size.)
// Initial guess x is overwritten with the output.
//
template<typename BigMatrixT>
void
gmres(BigMatrixT const& A,
ITensor const& b,
ITensor& x,
Args const& args = Args::global());
//
//
// Implementations
//
//
template <class BigMatrixT>
Real
davidson(BigMatrixT const& A,
ITensor& phi,
Args const& args)
{
auto v = std::vector<ITensor>(1);
v.front() = phi;
auto eigs = davidson(A,v,args);
phi = v.front();
return eigs.front();
}
template <class BigMatrixT>
std::vector<Real>
davidson(BigMatrixT const& A,
std::vector<ITensor>& phi,
Args const& args)
{
auto maxiter_ = args.getSizeT("MaxIter",2);
auto errgoal_ = args.getReal("ErrGoal",1E-14);
auto debug_level_ = args.getInt("DebugLevel",-1);
auto miniter_ = args.getSizeT("MinIter",1);
Real Approx0 = 1E-12;
auto nget = phi.size();
if(nget == 0) Error("No initial vectors passed to davidson.");
for(auto j : range(nget))
{
auto nrm = norm(phi[j]);
while(nrm == 0.0)
{
phi[j].randomize();
nrm = norm(phi[j]);
}
phi[j] *= 1./nrm;
}
auto maxsize = A.size();
auto actual_maxiter = std::min(maxiter_,maxsize-1);
if(debug_level_ >= 2)
{
printfln("maxsize-1 = %d, maxiter = %d, actual_maxiter = %d",
(maxsize-1), maxiter_, actual_maxiter);
}
if(dim(inds(phi.front())) != maxsize)
{
println("dim(inds(phi.front())) = ",dim(inds(phi.front())));
println("A.size() = ",A.size());
Error("davidson: size of initial vector should match linear matrix size");
}
auto V = std::vector<ITensor>(actual_maxiter+2);
auto AV = std::vector<ITensor>(actual_maxiter+2);
//Storage for Matrix that gets diagonalized
//set to NAN to ensure failure if we use uninitialized elements
auto M = CMatrix(actual_maxiter+2,actual_maxiter+2);
for(auto& el : M) el = Cplx(NAN,NAN);
auto NC = CVector(actual_maxiter+2);
//Mref holds current projection of A into V's
auto Mref = subMatrix(M,0,1,0,1);
//Get diagonal of A to use later
//auto Adiag = A.diag();
Real qnorm = NAN;
Vector D;
CMatrix U;
Real last_lambda = 1000.;
auto eigs = std::vector<Real>(nget,NAN);
V[0] = phi.front();
A.product(V[0],AV[0]);
auto initEn = ((dag(V[0])*AV[0]).eltC()).real();
if(debug_level_ > 2)
printfln("Initial Davidson energy = %.10f",initEn);
auto t = size_t(0); //which eigenvector we are currently targeting
auto iter = size_t(0);
for(auto ii : range(actual_maxiter+1))
{
//Diagonalize dag(V)*A*V
//and compute the residual q
auto ni = ii+1;
auto& q = V.at(ni);
auto& phi_t = phi.at(t);
auto& lambda = eigs.at(t);
//Step A (or I) of Davidson (1975)
if(ii == 0)
{
lambda = initEn;
stdx::fill(Mref,lambda);
//Calculate residual q
q = AV[0] - lambda*V[0];
//printfln("ii=%d, q = \n%f",ii,q);
}
else // ii != 0
{
Mref *= -1;
if(debug_level_ > 3)
{
println("Mref = \n",Mref);
}
diagHermitian(Mref,U,D);
Mref *= -1;
D *= -1;
lambda = D(t);
phi_t = U(0,t)*V[0];
q = U(0,t)*AV[0];
for(auto k : range(ii+1))
{
phi_t += U(k,t)*V[k];
q += U(k,t)*AV[k];
}
//Step B of Davidson (1975)
//Calculate residual q
q += (-lambda)*phi_t;
//Fix sign
if(U(0,t).real() < 0)
{
phi_t *= -1;
q *= -1;
}
if(debug_level_ >= 3)
{
println("D = ",D);
printfln("lambda = %.10f",lambda);
}
//printfln("ii=%d, full q = \n%f",ii,q);
}
//Step C of Davidson (1975)
//Check convergence
qnorm = norm(q);
bool converged = (qnorm < errgoal_ && std::abs(lambda-last_lambda) < errgoal_)
|| qnorm < std::max(Approx0,errgoal_ * 1E-3);
last_lambda = lambda;
if((qnorm < 1E-20) || (converged && ii >= miniter_) || (ii == actual_maxiter))
{
if(t < (nget-1) && ii < actual_maxiter)
{
++t;
last_lambda = 1000.;
}
else
{
if(debug_level_ >= 3) //Explain why breaking out of Davidson loop early
{
if((qnorm < errgoal_ && std::fabs(lambda-last_lambda) < errgoal_))
printfln("Exiting Davidson because errgoal=%.0E reached",errgoal_);
else if(ii < miniter_ || qnorm < std::max(Approx0,errgoal_ * 1.0e-3))
printfln("Exiting Davidson because small residual=%.0E obtained",qnorm);
else if(ii == actual_maxiter)
println("Exiting Davidson because ii == actual_maxiter");
}
goto done;
}
}
if(debug_level_ >= 2 || (ii == 0 && debug_level_ >= 1))
{
printf("I %d q %.0E E",iter,qnorm);
for(auto eig : eigs)
{
if(std::isnan(eig)) break;
printf(" %.10f",eig);
}
println();
}
//Compute next trial vector by
//first applying Davidson preconditioner
//formula then orthogonalizing against
//other vectors
//Step D of Davidson (1975)
//Apply Davidson preconditioner
//
//TODO add preconditioner step (may require
//non-contracting product to do efficiently)
//
//if(Adiag)
// {
// //Function which applies the mapping
// // f(x,theta) = 1/(theta - x)
// auto precond = [theta=lambda.real()](Real val)
// {
// return (theta==val) ? 0 : 1./(theta-val);
// };
// auto cond= Adiag;
// cond.apply(precond);
// q /= cond;
// }
//Step E and F of Davidson (1975)
//Do Gram-Schmidt on d (Npass times)
//to include it in the subbasis
int Npass = 1;
auto Vq = std::vector<Cplx>(ni);
int pass = 1;
int tot_pass = 0;
while(pass <= Npass)
{
if(debug_level_ >= 3) println("Doing orthog pass");
++tot_pass;
for(auto k : range(ni))
{
Vq[k] = (dag(V[k])*q).eltC();
//printfln("pass=%d Vq[%d] = %s",pass,k,Vq[k]);
}
for(auto k : range(ni))
{
q += (-Vq[k])*V[k];
}
auto qnrm = norm(q);
//printfln("pass=%d qnrm=%s",pass,qnrm);
if(qnrm < 1E-10)
{
//Orthogonalization failure,
//try randomizing
if(debug_level_ >= 2) println("Vector not independent, randomizing");
q = V.at(ni-1);
q.randomize();
qnrm = norm(q);
//Do another orthog pass
--pass;
if(debug_level_ >= 3) printfln("Now pass = %d",pass);
if(ni >= maxsize)
{
//Not be possible to orthogonalize if
//max size of q (vecSize after randomize)
//is size of current basis
if(debug_level_ >= 3)
println("Breaking out of Davidson: max Hilbert space size reached");
goto done;
}
if(tot_pass > Npass * 3)
{
// Maybe the size of the matrix is only 1?
if(debug_level_ >= 3)
println("Breaking out of Davidson: orthog step too big");
goto done;
}
}
q *= 1./qnrm;
q.scaleTo(1.);
++pass;
}
if(debug_level_ >= 3) println("Done with orthog step, tot_pass=",tot_pass);
//Check V's are orthonormal
//Mat Vo(ni+1,ni+1,NAN);
//for(int r = 1; r <= ni+1; ++r)
//for(int c = r; c <= ni+1; ++c)
// {
// z = (dag(V[r-1])*V[c-1]).eltC();
// Vo(r,c) = abs(z);
// Vo(c,r) = Vo(r,c);
// }
//println("Vo = \n",Vo);
if(debug_level_ >= 3)
{
if(std::fabs(norm(q)-1.0) > 1E-10)
{
println("norm(q) = ",norm(q));
Error("q not normalized after Gram Schmidt.");
}
}
//Step G of Davidson (1975)
//Expand AV and M
//for next step
A.product(V[ni],AV[ni]);
//Step H of Davidson (1975)
//Add new row and column to M
Mref = subMatrix(M,0,ni+1,0,ni+1);
auto newCol = subVector(NC,0,1+ni);
for(auto k : range(ni+1))
{
newCol(k) = (dag(V.at(k))*AV.at(ni)).eltC();
}
column(Mref,ni) &= newCol;
row(Mref,ni) &= conj(newCol);
++iter;
} //for(ii)
done:
//TODO: put this back?
//for(auto& T : phi)
// {
// if(T.scale().logNum() > 2) T.scaleTo(1.);
// }
//TODO: previously, it seems like the
//eigenvectors were being normalized
//automatically, and this was expected
//by DMRG (when calculating the proper
//entanglement entropy). At some point,
//normalization had to be done explicitly, why?
for(auto& phi_j : phi)
phi_j /= norm(phi_j);
//Compute any remaining eigenvalues and eigenvectors requested
//(zero indexed) value of t indicates how many have been "targeted" so far
if(debug_level_ >= 2 && t+1 < nget) printfln("Max iter. reached, computing remaining %d evecs",nget-t-1);
for(auto j : range(t+1,nget))
{
eigs.at(j) = D(j);
auto& phi_j = phi.at(j);
auto Nr = size_t(nrows(U));
phi_j = U(0,j)*V[0];
for(auto k : range1(std::min(V.size(),Nr)-1))
{
phi_j += U(k,j)*V[k];
}
}
if(debug_level_ >= 4)
{
//Check V's are orthonormal
auto Vo_final = CMatrix(iter+1,iter+1);
for(auto r : range(iter+1))
for(auto c : range(r,iter+1))
{
auto z = (dag(V[r])*V[c]).eltC();
Vo_final(r,c) = std::abs(z);
Vo_final(c,r) = Vo_final(r,c);
}
println("Vo_final = \n",Vo_final);
}
if(debug_level_ > 0)
{
printf("I %d q %.0E E",iter,qnorm);
for(auto eig : eigs)
{
if(std::isnan(eig)) break;
printf(" %.10f",eig);
}
println();
}
return eigs;
}
namespace gmres_details {
template<class Matrix, class T>
void
update(ITensor &x, int const k, Matrix const& h, std::vector<T>& s, std::vector<ITensor> const& v)
{
std::vector<T> y(s);
// Backsolve:
for (int i = k; i >= 0; i--)
{
y[i] /= h(i,i);
for (int j = i - 1; j >= 0; j--)
y[j] -= h(j,i) * y[i];
}
for (int j = 0; j <= k; j++)
x += v[j] * y[j];
}
template<typename T>
void
generatePlaneRotation(T const& dx, T const& dy, T& cs, T& sn)
{
if(dy == 0.0)
{
cs = 1.0;
sn = 0.0;
}
else if(std::abs(dy) > std::abs(dx))
{
auto temp = dx / dy;
sn = 1.0 / std::sqrt( 1.0 + temp*temp );
cs = temp * sn;
}
else
{
auto temp = dy / dx;
cs = 1.0 / std::sqrt( 1.0 + temp*temp );
sn = temp * cs;
}
}
void inline
applyPlaneRotation(Real& dx, Real& dy, Real const& cs, Real const& sn)
{
auto temp = cs * dx + sn * dy;
dy = -sn * dx + cs * dy;
dx = temp;
}
void inline
applyPlaneRotation(Cplx& dx, Cplx& dy, Cplx const& cs, Cplx const& sn)
{
auto temp = std::conj(cs) * dx + std::conj(sn) * dy;
dy = -sn * dx + cs * dy;
dx = temp;
}
void inline
dot(ITensor const& A, ITensor const& B, Real& res)
{
res = elt(dag(A)*B);
}
void inline
dot(ITensor const& A, ITensor const& B, Cplx& res)
{
res = eltC(dag(A)*B);
}
}//namespace gmres_details
template<typename T, typename BigMatrixT>
void
gmresImpl(BigMatrixT const& A,
ITensor const& b,
ITensor& x,
ITensor& Ax,
Args const& args)
{
auto n = A.size();
auto max_iter = args.getInt("MaxIter",n);
auto m = args.getInt("RestartIter",max_iter);
auto tol = args.getReal("ErrGoal",1E-14);
auto debug_level_ = args.getInt("DebugLevel",-1);
auto H = Mat<T>(m+1,m+1);
int i;
int j = 1;
int k;
std::vector<T> s(m+1);
std::vector<T> cs(m+1);
std::vector<T> sn(m+1);
ITensor w;
auto normb = norm(b);
auto r = b - Ax;
auto beta = norm(r);
if(normb == 0.0)
normb = 1.0;
auto resid = norm(r)/normb;
if(resid <= tol)
{
tol = resid;
max_iter = 0;
}
std::vector<ITensor> v(m+1);
while(j <= max_iter)
{
v[0] = r/beta;
v[0].scaleTo(1.0);
std::fill(s.begin(), s.end(), 0.0);
s[0] = beta;
for(i = 0; i < m && j <= max_iter; i++, j++)
{
ITensor w;
A.product(v[i],w);
// Begin Arnoldi iteration
// TODO: turn into a function?
for(k = 0; k<=i; ++k)
{
gmres_details::dot(w, v[k], H(k,i));
w -= H(k,i)*v[k];
}
auto normw = norm(w);
if(debug_level_ > 0)
println("norm(w) = ", normw);
H(i+1,i) = normw;
if(normw != 0)
{
v[i+1] = w/H(i+1,i);
v[i+1].scaleTo(1.0);
}
else
{
// Maybe this should be a warning?
// Also, maybe check if it is very close to zero?
// GMRES generally is converged at this point anyway
error("Norm of new Krylov vector is zero. Try raising 'ErrGoal'.");
}
for(k = 0; k<i; ++k)
gmres_details::applyPlaneRotation(H(k,i), H(k+1,i), cs[k], sn[k]);
gmres_details::generatePlaneRotation(H(i,i), H(i+1,i), cs[i], sn[i]);
gmres_details::applyPlaneRotation(H(i,i), H(i+1,i), cs[i], sn[i]);
gmres_details::applyPlaneRotation(s[i], s[i+1], cs[i], sn[i]);
resid = std::abs(s[i+1])/normb;
if(resid < tol)
{
gmres_details::update(x, i, H, s, v);
return;
}
} // end for loop
gmres_details::update(x, i-1, H, s, v);
A.product(x, Ax);
r = b - Ax;
beta = norm(r);
resid = beta/normb;
if(resid < tol)
return;
} // end while loop
}
template<typename BigMatrixT>
void
gmres(BigMatrixT const& A,
ITensor const& b,
ITensor& x,
Args const& args)
{
auto debug_level_ = args.getInt("DebugLevel",-1);
// Precompute Ax to figure out whether A or x is
// complex, maybe there is a cleaner code design
// to avoid this?
// Otherwise we would need to require that BigMatrixT
// has a function isComplex()
ITensor Ax;
A.product(x, Ax);
if(isComplex(b) || isComplex(Ax))
{
if(debug_level_ > 0)
println("Calling complex version of gmresImpl()");
gmresImpl<Cplx>(A,b,x,Ax,args);
}
else
{
if(debug_level_ > 0)
println("Calling real version of gmresImpl()");
gmresImpl<Real>(A,b,x,Ax,args);
}
}
} //namespace itensor
#endif