decl/specialized-lbfgs.hpp

Go to the documentation of this file.

#pragma once
 
#include <panoc-alm/inner/directions/decl/lbfgs.hpp>
 
namespace pa {
 
class SpecializedLBFGS {
  public:
    using Params = LBFGSParams;
 
    SpecializedLBFGS(Params params) : params(params) {}
    SpecializedLBFGS(Params params, size_t n, size_t history) : params(params) {
        resize(n, history);
    }
 
    bool standard_update(crvec xₖ, crvec xₖ₊₁, crvec pₖ,
                         crvec pₖ₊₁, crvec gradₖ₊₁);
    bool full_update(crvec xₖ, crvec xₖ₊₁, crvec pₖ_old_γ,
                     crvec pₖ₊₁, crvec gradₖ₊₁, const Box &C,
                     real_t γ);
    bool update(crvec xₖ, crvec xₖ₊₁, crvec pₖ, crvec pₖ₊₁,
                crvec gradₖ₊₁, const Box &C, real_t γ);
 
    template <class Vec>
    void apply(Vec &&q);
 
    void initialize(crvec x₀, crvec grad₀);
 
    void reset();
    void resize(size_t n, size_t history);
 
    std::string get_name() const { return "SpecializedLBFGS"; }
 
    size_t n() const { return sto.rows() - 1; }
    size_t history() const { return (sto.cols() - 2) / 4; }
    size_t succ(size_t i) const { return i + 1 < history() ? i + 1 : 0; }
    size_t pred(size_t i) const { return i == 0 ? history() - 1 : i - 1; }
 
    auto s(size_t i) { return sto.col(2 * i).topRows(n()); }
    auto s(size_t i) const { return sto.col(2 * i).topRows(n()); }
    auto y(size_t i) { return sto.col(2 * i + 1).topRows(n()); }
    auto y(size_t i) const { return sto.col(2 * i + 1).topRows(n()); }
    auto x(size_t i) { return sto.col(2 * history() + 2 * i).topRows(n()); }
    auto x(size_t i) const {
        return sto.col(2 * history() + 2 * i).topRows(n());
    }
    auto g(size_t i) { return sto.col(2 * history() + 2 * i + 1).topRows(n()); }
    auto g(size_t i) const {
        return sto.col(2 * history() + 2 * i + 1).topRows(n());
    }
    auto p() { return sto.col(4 * history()).topRows(n()); }
    auto p() const { return sto.col(4 * history()).topRows(n()); }
    auto w() { return sto.col(4 * history() + 1).topRows(n()); }
    auto w() const { return sto.col(4 * history() + 1).topRows(n()); }
    real_t &ρ(size_t i) { return sto.coeffRef(n(), 2 * i); }
    const real_t &ρ(size_t i) const { return sto.coeff(n(), 2 * i); }
    real_t &α(size_t i) { return sto.coeffRef(n(), 2 * i + 1); }
    const real_t &α(size_t i) const { return sto.coeff(n(), 2 * i + 1); }
 
  private:
    using storage_t = Eigen::Matrix<real_t, Eigen::Dynamic, Eigen::Dynamic>;
 
    storage_t sto;
    size_t idx   = 0;
    bool full    = false;
    real_t old_γ = 0;
    Params params;
};
 
} // namespace pa