122 lines
3.4 KiB
C++
122 lines
3.4 KiB
C++
/*
|
|
* iterative.h
|
|
* @brief Iterative methods, implementation
|
|
* @author Frank Dellaert
|
|
* Created on: Dec 28, 2009
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <gtsam/base/Matrix.h>
|
|
|
|
namespace gtsam {
|
|
|
|
/**
|
|
* Method of conjugate gradients (CG) template
|
|
* "System" class S needs gradient(S,v), e=S*v, v=S^e
|
|
* "Vector" class V needs dot(v,v), -v, v+v, s*v
|
|
* "Vector" class E needs dot(v,v)
|
|
* @param Ab, the "system" that needs to be solved, examples below
|
|
* @param x is the initial estimate
|
|
* @param epsilon determines the convergence criterion: norm(g)<epsilon*norm(g0)
|
|
* @param maxIterations, if 0 will be set to |x|
|
|
* @param steepest flag, if true does steepest descent, not CG
|
|
* */
|
|
template<class S, class V, class E>
|
|
V conjugateGradients(const S& Ab, V x, bool verbose, double epsilon,
|
|
size_t maxIterations, bool steepest = false);
|
|
|
|
/**
|
|
* Helper class encapsulating the combined system |Ax-b_|^2
|
|
* Needed to run Conjugate Gradients on matrices
|
|
* */
|
|
class System {
|
|
private:
|
|
const Matrix& A_;
|
|
const Vector& b_;
|
|
|
|
public:
|
|
|
|
System(const Matrix& A, const Vector& b) :
|
|
A_(A), b_(b) {
|
|
}
|
|
|
|
/** gradient of objective function 0.5*|Ax-b_|^2 at x = A_'*(Ax-b_) */
|
|
Vector gradient(const Vector& x) const {
|
|
return A_ ^ (A_ * x - b_);
|
|
}
|
|
|
|
/** Apply operator A */
|
|
inline Vector operator*(const Vector& x) const {
|
|
return A_ * x;
|
|
}
|
|
|
|
/** Apply operator A in place */
|
|
inline void multiplyInPlace(const Vector& x, Vector& e) const {
|
|
e = A_ * x;
|
|
}
|
|
|
|
/** Apply operator A'*e */
|
|
inline Vector operator^(const Vector& e) const {
|
|
return A_ ^ e;
|
|
}
|
|
|
|
/** x += alpha* A'*e */
|
|
inline void transposeMultiplyAdd(double alpha, const Vector& e, Vector& x) const {
|
|
gtsam::transposeMultiplyAdd(alpha,A_,e,x);
|
|
}
|
|
|
|
/**
|
|
* Print with optional string
|
|
*/
|
|
void print (const std::string& s = "System") const;
|
|
};
|
|
|
|
/**
|
|
* Method of steepest gradients, System version
|
|
*/
|
|
Vector steepestDescent(const System& Ab, const Vector& x, bool verbose =
|
|
false, double epsilon = 1e-3, double epsilon_abs = 1e-5, size_t maxIterations = 0);
|
|
|
|
/**
|
|
* Method of conjugate gradients (CG), System version
|
|
*/
|
|
Vector conjugateGradientDescent(const System& Ab, const Vector& x,
|
|
bool verbose = false, double epsilon = 1e-3, double epsilon_abs = 1e-5,
|
|
size_t maxIterations = 0);
|
|
|
|
/** convenience calls using matrices, will create System class internally: */
|
|
|
|
/**
|
|
* Method of steepest gradients, Matrix version
|
|
*/
|
|
Vector steepestDescent(const Matrix& A, const Vector& b, const Vector& x,
|
|
bool verbose = false, double epsilon = 1e-3, double epsilon_abs = 1e-5,
|
|
size_t maxIterations = 0);
|
|
|
|
/**
|
|
* Method of conjugate gradients (CG), Matrix version
|
|
*/
|
|
Vector conjugateGradientDescent(const Matrix& A, const Vector& b,
|
|
const Vector& x, bool verbose = false, double epsilon = 1e-3,
|
|
double epsilon_abs = 1e-5, size_t maxIterations = 0);
|
|
|
|
class GaussianFactorGraph;
|
|
class VectorConfig;
|
|
|
|
/**
|
|
* Method of steepest gradients, Gaussian Factor Graph version
|
|
* */
|
|
VectorConfig steepestDescent(const GaussianFactorGraph& fg,
|
|
const VectorConfig& x, bool verbose = false, double epsilon = 1e-3,
|
|
double epsilon_abs = 1e-5, size_t maxIterations = 0);
|
|
|
|
/**
|
|
* Method of conjugate gradients (CG), Gaussian Factor Graph version
|
|
* */
|
|
VectorConfig conjugateGradientDescent(const GaussianFactorGraph& fg,
|
|
const VectorConfig& x, bool verbose = false, double epsilon = 1e-3,
|
|
double epsilon_abs = 1e-5, size_t maxIterations = 0);
|
|
|
|
} // namespace gtsam
|