mbi.estimation.LBFGS

class mbi.estimation.LBFGS(marginal_oracle: MarginalOracle | None = None)[source]

Bases: Estimator

L-BFGS estimator for graphical models.

Optimizes the potentials (theta) directly via L-BFGS, back-propagating through the marginal inference oracle. The loss is convex w.r.t. marginals but typically non-convex w.r.t. potentials; in practice, L-BFGS still converges well.

See “Learning Graphical Model Parameters with Approximate Marginal Inference”.

marginal_oracle

The function to compute marginals from potentials. If None (default), uses default_oracle() to auto-select.

Type:

mbi.marginal_oracles.MarginalOracle | None

Methods

__init__

estimate

Estimate a Model from noisy marginal measurements.

precompile

Warm up the JIT cache for estimate asynchronously.

Attributes

marginal_oracle

marginal_oracle: MarginalOracle | None = None