MatrixBasedPLS

class probnum.linalg.solvers.MatrixBasedPLS(policy=<probnum.linalg.solvers.policies.ConjugateGradientPolicy object>, stopping_criterion=<probnum.LambdaStoppingCriterion object>)

Bases: probnum.linalg.solvers.ProbabilisticLinearSolver

Matrix-based probabilistic linear solver.

Probabilistic linear solver updating beliefs over the system matrix and its inverse. The solver makes use of prior information and iteratively infers the matrix and its inverse by matrix-vector multiplication.

This code implements the method described in Wenger et al. 1.

Parameters

References

1

Wenger, J. and Hennig, P., Probabilistic Linear Solvers for Machine Learning, Advances in Neural Information Processing Systems (NeurIPS), 2020

Methods Summary

solve(prior, problem[, rng])

Solve the linear system.

solve_iterator(prior, problem[, rng])

Generator implementing the solver iteration.

Methods Documentation

solve(prior, problem, rng=None)

Solve the linear system.

Parameters
Return type

Tuple[LinearSystemBelief, LinearSolverState]

Returns

  • belief – Posterior belief \((\mathsf{x}, \mathsf{A}, \mathsf{H}, \mathsf{b})\) over the solution \(x\), the system matrix \(A\), its (pseudo-)inverse \(H=A^\dagger\) and the right hand side \(b\).

  • solver_state – Final state of the solver.

solve_iterator(prior, problem, rng=None)

Generator implementing the solver iteration.

This function allows stepping through the solver iteration one step at a time and exposes the internal solver state.

Parameters
Yields

solver_state – State of the probabilistic linear solver.

Return type

Generator[LinearSolverState, None, None]