Standard 5 stops to get here · leads to 3

Energy-Based Model

A model that scores every possible configuration with a single number, its energy, where low energy means compatible or likely and high energy means not.

Your route here

5 stops · basics first
  1. Machine Learning ✓ understood

    Building systems that learn patterns from data instead of following hand-written rules, getting better at a task as they see more examples.

  2. Neural Network ✓ understood

    A model built from layers of simple units, each taking a weighted sum of its inputs and applying a nonlinearity; training adjusts the weights so the whole stack maps inputs to useful outputs.

  3. Dataset ✓ understood

    A collection of data examples used for training, validating, or testing machine learning models.

  4. Training ✓ understood

    The process of fitting a model to data by repeatedly measuring how wrong its outputs are and adjusting its parameters to reduce that error.

  5. Loss Function ✓ understood

    A function that scores how wrong a model's prediction is as a single number, which training then works to make as small as possible.

  6. Energy-Based Model · you are here ✓ understood

An energy-based model (EBM) learns a function that assigns one number, the energy, to every configuration of the variables it cares about. Configurations that fit the data, such as a realistic image or a correct answer paired with its question, get low energy; everything else gets high energy. Making a prediction means searching for the lowest-energy answer given what you already know.

The appeal is freedom. A normal probabilistic model must make its probabilities add up to one, which requires summing over every possible configuration, a quantity called the partition function that is often impossible to compute. An EBM only has to rank configurations, so it can use any neural network as its energy function. When probabilities are needed, energies can be turned into them, with lower energy meaning exponentially more likely.

Training

Learning has two halves: push the energy down on real data and push it up on wrong or made-up configurations. The hard part is finding good “wrong” examples to push up. Classic approaches draw them with MCMC sampling; others use margin-based loss functions against the most tempting wrong answer. LeCun and colleagues framed many familiar models, including some discriminative classifiers, as special cases of this recipe.

Family tree

Hopfield networks and Boltzmann machines are early EBMs: they define an energy over the states of their units and settle into low-energy states. Modern EBMs use deep networks as the energy function; Du and Mordatch scaled MCMC-based training to generate ImageNet images.

The catch

Generating a sample means running a long search or sampling chain rather than a single forward pass, and training can be unstable. That cost is the main reason EBMs remain a research topic more than a production tool.

Where it sits

Explore nearby

Sources

  1. Yann LeCun, Sumit Chopra, Raia Hadsell, Marc'Aurelio Ranzato and Fu Jie Huang, "A Tutorial on Energy-Based Learning" . In Predicting Structured Data, MIT Press, 2006
  2. Yilun Du and Igor Mordatch, "Implicit Generation and Generalization in Energy-Based Models" . NeurIPS 2019