SimplicialLLT返回错误的cholesky因子

时间:2018-11-09 12:02:03

标签: c++ eigen scientific-computing

我想使用本征来计算稀疏矩阵的cholesky分解。但是,结果不正确,我无法找到原因。如何获得正确答案?

在Eigen中是否实现了利用稀疏矩阵的结构以提高性能的特殊例程(例如,对于下面示例中的带状矩阵或三角矩阵)?

#include <iostream>
#include <Eigen/Sparse>
#include <Eigen/Dense>

int main() 
{

    // create sparse Matrix
    int n = 5; 
    std::vector<Eigen::Triplet<double> > ijv; 
    for(int i = 0; i < n; i++)
    {
        ijv.push_back(Eigen::Triplet<double>(i,i,1));
        if(i < n-1)
        {
            ijv.push_back(Eigen::Triplet<double>(i+1,i,-0.9));
        }
    }
    Eigen::SparseMatrix<double> X(n,n);
    X.setFromTriplets(ijv.begin(), ijv.end());
    Eigen::SparseMatrix<double> XX = X * X.transpose();

    // Cholesky decomposition 
    Eigen::SimplicialLLT <Eigen::SparseMatrix<double> > cholesky;
    cholesky.analyzePattern(XX);
    cholesky.factorize(XX);

    std::cout << Eigen::MatrixXd(XX) << std::endl;
    std::cout << Eigen::MatrixXd(cholesky.matrixL()) << std::endl;

}

矩阵如下:

输入XX

   1 -0.9    0    0    0
-0.9 1.81 -0.9    0    0
   0 -0.9 1.81 -0.9    0
   0    0 -0.9 1.81 -0.9
   0    0    0 -0.9 1.81

输出(cholesky.matrixL()):

  1.34536         0         0         0         0
-0.668965   1.16726         0         0         0
        0 -0.771039    1.1025         0         0
        0         0         0         1         0
        0         0 -0.816329      -0.9  0.577587

它应该是什么样子(X):

   1    0    0    0   0
-0.9    1    0    0   0
   0 -0.9    1    0   0
   0    0 -0.9    1   0
   0    0    0 -0.9   1

2 个答案:

答案 0 :(得分:3)

SimplicialLLT计算Cholesky分解直至一个排列。使用Eigen::SimplicialLLT <Eigen::SparseMatrix<double>, Eigen::Lower, Eigen::NaturalOrdering<int>> cholesky; cholesky.compute(XX); 使用身份置换,您将获得预期的结果:

metafields

答案 1 :(得分:2)

请不要忘记SimplicialLLT不会分解A = L * L^T而是P * A * P^T = L * T^TP一起排列矩阵。如果您需要P作为身份,请使用NaturalOrdering

Eigen::SimplicialLLT<Eigen::SparseMatrix<double>, Eigen::Lower, Eigen::NaturalOrdering<int> > cholesky;