Skip to main content

LBFGS

Struct LBFGS 

Source
pub struct LBFGS<A: Float + ScalarOperand + Debug> { /* private fields */ }
Expand description

L-BFGS optimizer

Implements the Limited-memory Broyden-Fletcher-Goldfarb-Shanno (L-BFGS) algorithm. This is a quasi-Newton method that approximates the Hessian inverse using a limited amount of memory by storing only a few vectors from previous iterations.

§Curvature pairs

The optimizer stores the previous parameters and the previous gradient, so the curvature pair is the true s = x_k - x_{k-1}, y = g_k - g_{k-1} even when the caller post-processes the returned parameters (projection, clipping, weight decay, a different step size, …). Pairs with y·s <= 0 are skipped so the implicit inverse-Hessian stays positive definite.

Optimizer::step applies a fixed step size (the learning rate) along the two-loop direction: it has no access to the objective, so it cannot run a line search. Use LBFGS::step_with_loss to get a backtracking Armijo line search that uses the configured c1 and max_ls parameters.

§Examples

use scirs2_core::ndarray::Array1;
use optirs_core::optimizers::{LBFGS, Optimizer};

// Initialize parameters and gradients
let params = Array1::zeros(5);
let gradients = Array1::from_vec(vec![0.1, 0.2, -0.3, 0.0, 0.5]);

// Create an L-BFGS optimizer
let mut optimizer = LBFGS::new(1.0);

// Update parameters
let new_params = optimizer.step(&params, &gradients).expect("optimizer.step succeeds");

Implementations§

Source§

impl<A: Float + ScalarOperand + Debug + Send + Sync> LBFGS<A>

Source

pub fn new(learning_rate: A) -> Self

Creates a new L-BFGS optimizer with the given learning rate

§Arguments
  • learning_rate - The learning rate for parameter updates
Source

pub fn new_with_config( learning_rate: A, history_size: usize, tolerance_grad: A, c1: A, c2: A, max_ls: usize, ) -> Self

Creates a new L-BFGS optimizer with full configuration

§Arguments
  • learning_rate - The learning rate for parameter updates
  • history_size - Number of past gradients/steps to store (default: 100)
  • tolerance_grad - Gradient norm tolerance for convergence (default: 1e-7)
  • c1 - Wolfe line search parameter for Armijo condition (default: 1e-4)
  • c2 - Wolfe line search parameter for curvature condition (default: 0.9)
  • max_ls - Maximum line search iterations (default: 25)
Source

pub fn learning_rate(&self) -> A

Gets the current learning rate

Source

pub fn set_lr(&mut self, lr: A)

Sets the learning rate

Source

pub fn c1(&self) -> A

Armijo (sufficient-decrease) line search parameter c1

Source

pub fn c2(&self) -> A

Wolfe curvature line search parameter c2

Source

pub fn max_ls(&self) -> usize

Maximum number of line search iterations

Source

pub fn history_len(&self) -> usize

Number of curvature pairs currently stored

Source

pub fn last_curvature_pair(&self) -> Option<(&Array1<A>, &Array1<A>)>

The most recently stored curvature pair (s, y), if any.

s is the true parameter difference x_k - x_{k-1} (as observed across two consecutive calls) and y is the corresponding gradient difference g_k - g_{k-1}.

Source

pub fn initial_hessian_scale(&self) -> A

The current initial inverse-Hessian scaling gamma_k = (s·y) / (y·y).

Source

pub fn set_line_search_contraction(&mut self, rho: A) -> Result<()>

Sets the backtracking contraction factor used by LBFGS::step_with_loss.

Must lie strictly between 0 and 1; other values are rejected.

Source

pub fn reset(&mut self)

Resets the internal state of the optimizer

Source

pub fn step_with_loss<D, F>( &mut self, params: &Array<A, D>, gradients: &Array<A, D>, loss_fn: F, ) -> Result<Array<A, D>>
where D: Dimension, F: FnMut(&Array<A, D>) -> A,

Performs an L-BFGS step with a backtracking Armijo line search.

Unlike Optimizer::step, which has no access to the objective and therefore applies a fixed step size, this method evaluates loss_fn at trial points and accepts the first step size satisfying the Armijo sufficient-decrease condition

f(x + alpha * d) <= f(x) + c1 * alpha * g^T d

starting from alpha = learning_rate and contracting by the line search contraction factor (default 0.5) for at most max_ls iterations. If no trial step satisfies the condition, the trial with the lowest objective value is used when it improves on f(x); otherwise the parameters are returned unchanged.

If the two-loop direction is not a descent direction (which can only happen through numerical error, since non-positive curvature pairs are never stored), the search falls back to steepest descent for this step.

§Errors

Returns OptimError::DimensionMismatch if params and gradients have different shapes, and OptimError::InvalidConfig if the objective is not finite at the current parameters.

§Examples
use scirs2_core::ndarray::Array1;
use optirs_core::optimizers::LBFGS;

let mut optimizer = LBFGS::new(1.0);
let mut params = Array1::from_vec(vec![2.0_f64, -3.0]);
let loss = |x: &Array1<f64>| x.iter().map(|v| v * v).sum::<f64>();

for _ in 0..30 {
    let grads = params.mapv(|v| 2.0 * v);
    params = optimizer
        .step_with_loss(&params, &grads, loss)
        .expect("step succeeds");
}
assert!(params.iter().all(|v| v.abs() < 1e-6));

Trait Implementations§

Source§

impl<A: Clone + Float + ScalarOperand + Debug> Clone for LBFGS<A>

Source§

fn clone(&self) -> LBFGS<A>

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl<A: Debug + Float + ScalarOperand + Debug> Debug for LBFGS<A>

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl<A, D> Optimizer<A, D> for LBFGS<A>
where A: Float + ScalarOperand + Debug + Send + Sync, D: Dimension,

Source§

fn step( &mut self, params: &Array<A, D>, gradients: &Array<A, D>, ) -> Result<Array<A, D>>

Performs an L-BFGS step with a fixed step size.

This trait method has no access to the objective, so no line search is possible; the two-loop direction is scaled by the learning rate (reduced by 1 / (1 + ||g||) on the very first step, before any curvature information exists). Use LBFGS::step_with_loss for the backtracking Armijo line search that uses the configured c1 and max_ls.

Source§

fn get_learning_rate(&self) -> A

Gets the current learning rate
Source§

fn set_learning_rate(&mut self, learning_rate: A)

Sets a new learning rate
Source§

fn step_list( &mut self, params_list: &[&Array<A, D>], gradients_list: &[&Array<A, D>], ) -> Result<Vec<Array<A, D>>>

Updates multiple parameter arrays at once Read more

Auto Trait Implementations§

§

impl<A> Freeze for LBFGS<A>
where A: Freeze,

§

impl<A> RefUnwindSafe for LBFGS<A>
where A: RefUnwindSafe,

§

impl<A> Send for LBFGS<A>
where A: Send,

§

impl<A> Sync for LBFGS<A>
where A: Sync,

§

impl<A> Unpin for LBFGS<A>
where A: Unpin,

§

impl<A> UnsafeUnpin for LBFGS<A>
where A: UnsafeUnpin,

§

impl<A> UnwindSafe for LBFGS<A>

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> DynClone for T
where T: Clone,

Source§

fn __clone_box(&self, _: Private) -> *mut ()

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<SS, SP> SupersetOf<SS> for SP
where SS: SubsetOf<SP>,

Source§

fn to_subset(&self) -> Option<SS>

The inverse inclusion map: attempts to construct self from the equivalent element of its superset. Read more
Source§

fn is_in_subset(&self) -> bool

Checks if self is actually part of its subset T (and can be converted to it).
Source§

fn to_subset_unchecked(&self) -> SS

Use with care! Same as self.to_subset but without any property checks. Always succeeds.
Source§

fn from_subset(element: &SS) -> SP

The inclusion map: converts self to the equivalent element of its superset.
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V