SearcharxivSearch

arXiv subjects

Meiyue Shao

Publications and source records attributed to Meiyue Shao.

At least 19 recordsLinked to original sources

A mixed precision algorithm for the matrix square root

Mixed precision algorithms can significantly enhance the performance of linear algebra solvers by leveraging increasingly powerful low precision hardware while recovering working precision accuracy through, for example, iterative refinement. In this paper, we propose a novel mixed precision algorithm for computing matrix square roots. Our algorithm combines a Schur decomposition approach in low precision with iterative refinement performed through an approximate Newton method. We perform a detailed convergence analysis of the approximate Newton method. For the special case of symmetric positive definite matrices, this analysis implies that one can recover full working precision accuracy under mild conditions. Numerical experiments on x86-64 architectures indicate that our algorithm frequently reduces execution time compared with a fixed working-precision Schur algorithm.

math.NA

Linear convergence of iterative contour integral-based eigensolvers for nonlinear eigenvalue problems

Solving nonlinear eigenvalue problems is an important and challenging task in scientific computing. Contour integral-based approaches are attractive for such eigenvalue problems because they reliably target all eigenvalues in a prescribed domain. However, unlike in the linear case, many traditional methods of this type, such as Beyn's method, lack an inherent iterative refinement mechanism. Consequently, achieving high accuracy requires high-quality quadrature rules for approximating the contour integral, which often leads to prohibitive computational costs. A notable exception is the so-called NLFEAST algorithm, which combines contour integral techniques with a nonlinear Rayleigh--Ritz extraction step. In this work, we propose a general framework of iterative contour integral-based methods for nonlinear eigenvalue problems that includes NLFEAST. This allows us to prove linear convergence of NLFEAST under mild assumptions and also explains why certain nonlinear eigensolvers do not combine well with iterative methods. Numerical experiments confirm our theoretical findings; in particular that NLFEAST can achieve high accuracy even with a limited number of quadrature nodes, significantly outperforming Beyn's method on challenging problems.

math.NA

On Two-Stage Householder Orthogonalization

Two-stage orthogonalization is essential in numerical algorithms such as Krylov subspace methods. For this task we need to orthogonalize a matrix $A$ against another matrix $V$ with orthonormal columns. A common approach is to employ the block Gram--Schmidt algorithm. However, its stability largely depends on the condition number of $[V,A]$. While performing a Householder orthogonalization on $[V,A]$ is unconditionally stable, it does not utilize the knowledge that $V$ has orthonormal columns. To address these issues, we propose a two-stage Householder orthogonalization algorithm based on the generalized Householder transformation. Instead of explicitly orthogonalizing the entire $V$, our algorithm only needs to orthogonalizes a square submatrix of $V$. Theoretical analysis and numerical experiments demonstrate that our method is also unconditionally stable.

math.NA

Mixed precision thin SVD algorithms based on the Gram matrix

In this work, we present a mixed precision algorithm that leverages the Gram matrix and Jacobi methods to compute the singular value decomposition (SVD) of tall-and-skinny matrices. By constructing the Gram matrix in higher precision and coupling it with a Jacobi algorithm, our theoretical analysis and numerical experiments both indicate that the singular values computed by this mixed precision thin SVD algorithm attain high relative accuracy. In practice, our mixed precision thin SVD algorithm yields speedups of over 10x on a single CPU and about 2x on distributed memory systems when compared with traditional thin SVD methods.

math.NA

A Contour Integral-Based Algorithm for Computing Generalized Singular Values

We propose a contour integral-based algorithm for computing a few singular values of a matrix or a few generalized singular values of a matrix pair. Mathematically, the generalized singular values of a matrix pair are the eigenvalues of an equivalent Hermitian-definite matrix pencil, known as the Jordan-Wielandt matrix pencil. However, direct application of the FEAST algorithm does not fully exploit the structure of this problem. We analyze several projection strategies on the Jordan-Wielandt matrix pencil, and propose an effective and robust scheme tailored to GSVD. Both theoretical analysis and numerical experiments demonstrate that our algorithm achieves rapid convergence and satisfactory accuracy.

math.NA

A Structure-Preserving LOBPCG Algorithm for the Bethe-Salpeter Eigenvalue Problem

The Bethe-Salpeter eigenvalue problem is a structured eigenvalue problem arising in many-body physics. In practice, a few of the smallest positive eigenvalues and the corresponding eigenvectors need to be computed. In principle, the LOBPCG algorithm can be applied to solve this eigenvalue problem. However, direct application of the existing LOBPCG algorithm does not utilize the inherent structure of the problem. We design a structure-preserving eigensolver based on the indefinite LOBPCG algorithm to efficiently solve the Bethe-Salpeter eigenvalue problem. We propose an improved Hetmaniuk-Lehoucq trick for the indefinite inner product, as well as an adaptive, multi-level orthogonalization strategy to ensure the numerical stability of our algorithm. Numerical experiments demonstrate that the proposed algorithm can efficiently and accurately compute the desired eigenpairs. Since the symplectic eigenvalue problem for symmetric positive definite matrices can be transformed to the Bethe-Salpeter eigenvalue problem, our algorithm can naturally be adopted as a symplectic eigensolver.

math.NA

On Eigenvector Computation and Eigenvalue Reordering for the Non-Hermitian Quaternion Eigenvalue Problem

In this paper we present several additions to the quaternion QR algorithm, including algorithms for eigenvector computation and eigenvalue reordering. A key outcome of the eigenvalue reordering algorithm is that the aggressive early deflation (AED) technique, which significantly enhances the convergence of the QR algorithm, is successfully applied to the quaternion eigenvalue problem. We conduct numerical experiments to demonstrate the efficiency and effectiveness of the proposed algorithms.

math.NA

Mixed precision iterative refinement for least squares with linear equality constraints and generalized least squares problems

Recent development on mixed precision techniques has largely enhanced the performance of various linear algebra solvers, one of which being the solver for the least squares problem $\min_{x}\lVert b-Ax\rVert_{2}$. By transforming least squares problems into augmented linear systems, mixed precision techniques are capable of refining the lower precision solution to the working precision. In this paper, we propose mixed precision iterative refinement algorithms for two variants of least squares problems -- the least squares problem with linear equality constraints (LSE) and the generalized least squares problem (GLS). Both classical and GMRES-based iterative refinement can be applied to augmented systems of these two problems to improve the accuracy of the solution. For reasonably well-conditioned problems, our algorithms reduce the execution time by a factor of 40% on average compared to the fixed precision ones from LAPACK on the x86-64 architecture.

math.NA

On a shrink-and-expand technique for symmetric block eigensolvers

In symmetric block eigenvalue algorithms, such as the subspace iteration algorithm and the locally optimal block preconditioned conjugate gradient (LOBPCG) algorithm, a large block size is often employed to achieve robustness and rapid convergence. However, using a large block size also increases the computational cost. Traditionally, the block size is typically reduced after convergence of some eigenpairs, known as deflation. In this work, we propose a non-deflation-based, more aggressive technique, where the block size is adjusted dynamically during the algorithm. This technique can be applied to a wide range of block eigensolvers, reducing computational cost without compromising convergence speed. We present three adaptive strategies for adjusting the block size, and apply them to four well-known eigensolvers as examples. Detailed theoretical analysis and numerical experiments are provided to illustrate the efficiency of the proposed technique. In practice, an overall acceleration of 20% to 30% is observed.

math.NA

Solving nonlinear eigenvalue problems via contour integration and region partitioning

In this work, we combine Beyn's method and the recently developed recursive integral method (RIM) to propose a contour integral-based, region partitioning eigensolver for nonlinear eigenvalue problems. A new partitioning criterion is employed to eliminate the need for a problem-dependent parameter, making our algorithm much more robust compared to the original RIM. Moreover, our algorithm can be directly applied to regions containing singularities or accumulation points, which are typically challenging for existing nonlinear eigensolvers to handle. Comprehensive numerical experiments are provided to demonstrate that the proposed algorithm is particularly well suited for dealing with regions including many eigenvalues.

math.NA

Improving performance of contour integral-based nonlinear eigensolvers with infinite GMRES

In this work, the infinite GMRES algorithm, recently proposed by Correnty et al., is employed in contour integral-based nonlinear eigensolvers, avoiding the computation of costly factorizations at each quadrature node to solve the linear systems efficiently. Several techniques are applied to make the infinite GMRES memory-friendly, computationally efficient, and numerically stable in practice. More specifically, we analyze the relationship between polynomial eigenvalue problems and their scaled linearizations, and provide a novel weighting strategy which can significantly accelerate the convergence of infinite GMRES in this particular context. We also adopt the technique of TOAR to infinite GMRES to reduce the memory footprint. Theoretical analysis and numerical experiments are provided to illustrate the efficiency of the proposed algorithm.

math.NA

A mixed precision Jacobi SVD algorithm

We propose a mixed precision Jacobi algorithm for computing the singular value decomposition (SVD) of a dense matrix. After appropriate preconditioning, the proposed algorithm computes the SVD in a lower precision as an initial guess, and then performs one-sided Jacobi rotations in the working precision as iterative refinement. By carefully transforming a lower precision solution to a higher precision one, our algorithm achieves about 2 times speedup on the x86-64 architecture compared to the usual one-sided Jacobi SVD algorithm in LAPACK, without sacrificing the accuracy.

math.NA

A fast low-rank inversion algorithm of dielectric matrix in GW approximation

The dielectric response function and its inverse are crucial physical quantities in materials science. We propose an accurate and efficient strategy to invert the dielectric function matrix. The GW approximation, a powerful approach to accurately describe many-body excited states, is taken as an application to demonstrate accuracy and efficiency. We incorporate the interpolative separable density fitting (ISDF) algorithm with Sherman--Morrison--Woodbury (SMW) formula to accelerate the inversion process by exploiting low-rank properties of dielectric function in plane-wave GW calculations. Our ISDF--SMW strategy produces accurate quasiparticle energies with $O(N_{\mathrm{r}}N_{\mathrm{e}}^2)$ computational cost $(N_{\mathrm{e}}$ is the number of electrons and $N_{\mathrm{r}}=100$--$1000N_{\mathrm{e}}$ is the number of grid points) with negligible small error of $0.03$ eV for both complex molecules and solids. This new strategy for inverting the dielectric matrix can be \(50\times\) faster than the current state-of-the-art implementation in BerkeleyGW, resulting in two orders of magnitude speedup for total GW calculations.

math.NA

A mixed precision LOBPCG algorithm

The locally optimal block preconditioned conjugate gradient (LOBPCG) algorithm is a popular approach for computing a few smallest eigenvalues and the corresponding eigenvectors of a large Hermitian positive definite matrix A. In this work, we propose a mixed precision variant of LOBPCG that uses a (sparse) Cholesky factorization of A computed in reduced precision as the preconditioner. To further enhance performance, a mixed precision orthogonalization strategy is proposed. To analyze the impact of reducing precision in the preconditioner on performance, we carry out a rounding error and convergence analysis of PINVIT, a simplified variant of LOBPCG. Our theoretical results predict and our numerical experiments confirm that the impact on convergence remains marginal. In practice, our mixed precision LOBPCG algorithm typically reduces the computation time by a factor of 1.4--2.0 on both CPUs and GPUs.

math.NA

Householder orthogonalization with a non-standard inner product

Householder orthogonalization plays an important role in numerical linear algebra. It attains perfect orthogonality regardless of the conditioning of the input. However, in the context of a non-standard inner product, it becomes difficult to apply Householder orthogonalization due to the lack of an initial orthonormal basis. We propose strategies to overcome this obstacle and discuss algorithms and variants of Householder orthogonalization with a non-standard inner product. Theoretical analysis and numerical experiments demonstrate that our approach is numerically stable under mild assumptions.

math.NA

A robust and efficient implementation of LOBPCG

Locally Optimal Block Preconditioned Conjugate Gradient (LOBPCG) is widely used to compute eigenvalues of large sparse symmetric matrices. The algorithm can suffer from numerical instability if it is not implemented with care. This is especially problematic when the number of eigenpairs to be computed is relatively large. In this paper we propose an improved basis selection strategy based on earlier work by Hetmaniuk and Lehoucq as well as a robust convergence criterion which is backward stable to enhance the robustness. We also suggest several algorithmic optimizations that improve performance of practical LOBPCG implementations. Numerical examples confirm that our approach consistently and significantly outperforms previous competing approaches in both stability and speed.

math.NA

Ab Initio No Core Shell Model with Leadership-Class Supercomputers

Nuclear structure and reaction theory is undergoing a major renaissance with advances in many-body methods, strong interactions with greatly improved links to Quantum Chromodynamics (QCD), the advent of high performance computing, and improved computational algorithms. Predictive power, with well-quantified uncertainty, is emerging from non-perturbative approaches along with the potential for guiding experiments to new discoveries. We present an overview of some of our recent developments and discuss challenges that lie ahead. Our foci include: (1) strong interactions derived from chiral effective field theory; (2) advances in solving the large sparse matrix eigenvalue problem on leadership-class supercomputers; (3) selected observables in light nuclei with the JISP16 interaction; (4) effective electroweak operators consistent with the Hamiltonian; and, (5) discussion of A=48 system as an opportunity for the no-core approach with the reintroduction of the core.

nucl-th

Accelerating Optical Absorption Spectra and Exciton Energy Computation for Nanosystems via Interpolative Separable Density Fitting

We present an efficient way to solve the Bethe-Salpeter equation (BSE), a model for the computation of absorption spectra in molecules and solids that includes electron-hole excitations. Standard approaches to construct and diagonalize the Bethe-Salpeter Hamiltonian require at least $Ø(N_e^5)$ operations, where $N_e$ is proportional to the number of electrons in the system, limiting its application to small systems. Our approach is based on the interpolative separable density fitting (ISDF) technique to construct low rank approximations to the bare and screened exchange operators associated with the BSE Hamiltonian. This approach reduces the complexity of the Hamiltonian construction to $Ø(N_e^3)$ with a much smaller pre-constant. Here, we implement the ISDF method for the BSE calculations within the Tamm-Dancoff approximation (TDA) in the BerkeleyGW software package. We show that ISDF-based BSE calculations in molecules and solids reproduce accurate exciton energies and optical absorption spectra with significantly reduced computational cost.

physics.comp-ph