我想使用本征来计算稀疏矩阵的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
最佳答案
别忘了SimplicialLLT
不会分解A = L * L^T
,而是P * A * P^T = L * T^T
和P
一个置换矩阵。如果您需要P
作为标识,请使用NaturalOrdering
:
Eigen::SimplicialLLT<Eigen::SparseMatrix<double>, Eigen::Lower, Eigen::NaturalOrdering<int> > cholesky;