mbi.estimation.InteriorGradient

class mbi.estimation.InteriorGradient(marginal_oracle: MarginalOracle | None = None, mesh: Mesh | None = None)[source]

Bases: Estimator

Interior Gradient estimator for graphical models.

Interior Gradient is an accelerated proximal algorithm for solving a smooth convex optimization problem over the marginal polytope. This algorithm requires knowledge of the Lipschitz constant of the gradient of the loss function. Based on the paper “Interior Gradient and Proximal Methods for Convex and Conic Optimization”.

marginal_oracle

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

Type:: mbi.marginal_oracles.MarginalOracle | None

mesh

JAX sharding mesh.

Type:: jax._src.mesh.Mesh | None

Methods

`__init__`
`estimate`	Estimate a Model from noisy marginal measurements.
`precompile`	Warm up the JIT cache for `estimate` asynchronously.

Attributes

`marginal_oracle`
`mesh`

marginal_oracle: MarginalOracle | None = None

mesh: Mesh | None = None