Contracting Symmetric Tensors Using Fewer Multiplications

We present more computationally-efficient algorithms for contracting symmetric tensors. Tensor contractions are reducible to matrix multiplication, but permutational symmetries of the data, which are expressed by the tensor representation, provide an opportunity for more efficient algorithms. Previously known methods have exploited only tensor symmetries that yield identical computations that are directly evident in the contraction expression. We present a new ‘symmetry preserving’ algorithm that uses an algebraic reorganization in order to exploit considerably more symmetry in the computation of the contraction than the conventional approach. The new algorithm requires fewer multiplications but more additions per multiplication than previous approaches. The applications of this result include the capability to multiply a symmetric matrix by a vector, as well as compute the rank-2 symmetric vector outer product in half the number of scalar multiplications, albeit with more additions. The symmetry preserving algorithm can also be adapted to perform the complex versions of these operations, namely the product of a Hermitian matrix and a vector and the rank-2 Hermitian vector outer product, in 3/4 of the overall operations. Consequently, the number of operations needed for the direct algorithm to compute the eigenvalues of a Hermitian matrix is reduced by the same factor. Our symmetry preserving tensor contraction algorithm can also be adapted to the antisymmetric case and is therefore applicable to the tensor-contraction computations employed in quantum chemistry. For these applications, notably the coupled-cluster method, our algorithm yields the highest potential speed-ups, since in many higher-order contractions the reduction in the number of multiplications achieved by our algorithm enables an equivalent reduction in overall contraction cost. We highlight that for three typical coupled-cluster contractions taken from methods of three different orders, our algorithm achieves 2X, 4X, and 9X improvements in arithmetic cost over the standard approach.

[1]  Josef Paldus,et al.  Correlation problems in atomic and molecular systems III. Rederivation of the coupled-pair many-electron theory using the traditional quantum chemical methodst†‡§ , 1971 .

[2]  R. Bartlett,et al.  Coupled-cluster methods that include connected quadruple excitations, T4: CCSDTQ-1 and Q(CCSDT) , 1989 .

[3]  Charles L. Lawson,et al.  Basic Linear Algebra Subprograms for Fortran Usage , 1979, TOMS.

[4]  James Demmel,et al.  Communication-optimal parallel algorithm for strassen's matrix multiplication , 2012, SPAA '12.

[5]  Edgar Solomonik Provably Efficient Algorithms for Numerical Tensor Algebra , 2014 .

[6]  R. Bartlett Many-Body Perturbation Theory and Coupled Cluster Theory for Electron Correlation in Molecules , 1981 .

[7]  V. Pan How can we speed up matrix multiplication , 1984 .

[8]  Tamara G. Kolda,et al.  Tensor Decompositions and Applications , 2009, SIAM Rev..

[9]  R. Bartlett,et al.  The full CCSDT model for molecular electronic structure , 1987 .

[10]  J. Cizek On the Correlation Problem in Atomic and Molecular Systems. Calculation of Wavefunction Components in Ursell-Type Expansion Using Quantum-Field Theoretical Methods , 1966 .

[11]  V. Strassen Gaussian elimination is not optimal , 1969 .

[12]  Michael J. Frisch,et al.  MP2 energy evaluation by direct methods , 1988 .

[13]  John F. Stanton,et al.  A massively parallel tensor contraction framework for coupled-cluster computations , 2014, J. Parallel Distributed Comput..

[14]  Acknowledgements , 1992, Experimental Gerontology.

[15]  S. Hirata Tensor Contraction Engine: Abstraction and Automated Parallel Implementation of Configuration-Interaction, Coupled-Cluster, and Many-Body Perturbation Theories , 2003 .

[16]  John D. Watts,et al.  Non-iterative fifth-order triple and quadruple excitation energy corrections in correlated methods , 1990 .

[17]  M. Head‐Gordon,et al.  A fifth-order perturbation comparison of electron correlation theories , 1989 .

[18]  R. D. Schafer An Introduction to Nonassociative Algebras , 1966 .

[19]  A. Adrian Albert On Jordan algebras of linear transformations , 1946 .