Техническо РЪКОВОДСТВО

Енергийно базирани модели

An energy-based model (EBM) is a machine learning model that assigns a single number, called energy, to every possible configuration of its variables, with low energy meaning compatible or likely and high energy meaning unlikely.

Чуйте това ръководствоAI-разказан · ~9 min

На тази страница10 минути четене
  1. Преглед
  2. Основни изводи
  3. Дълбоко гмуркане
  4. Reading a three-class classifier as an energy-based model
  5. Внедряване в реалния свят
  6. Бъдещето на моделите, базирани на енергия
  7. Прилагайки го на практика
  8. Източници и допълнителна литература
  9. Продължете да изследвате
  10. Често задавани въпроси

Преглед

Because the energies never have to be normalized into probabilities that sum to one, EBMs are very flexible, which is why they serve as a common framework for classifiers, Hopfield networks, Boltzmann machines and score-based generative models.

Основни изводи

  1. An energy-based model assigns one scalar energy to each configuration of variables, with lower energy meaning a better or more likely fit.
  2. Energies become probabilities by exponentiating minus the energy and dividing by a normalizing constant Z that is usually impossible to compute exactly.
  3. Training must lower energy on real data and raise it elsewhere, using MCMC sampling, score matching or noise contrastive estimation.
  4. Hopfield networks from 1982 and Boltzmann machines from 1985 are early energy-based models, recognized by the 2024 Nobel Prize in Physics.
  5. Any classifier's logits can be read as negative energies, which is the basis of the 2019 Joint Energy-based Model.

Дълбоко гмуркане

An energy-based model, or EBM, is a way of describing what a model knows using a single scalar score. For any setting of the variables the model cares about, for example an image and a label, or a sentence and its translation, the model outputs a number called the energy. Low energy means the combination fits well together; high energy means it does not. The word comes from physics, where systems tend to settle into low-energy states. Yann LeCun and colleagues' 2006 "A Tutorial on Energy-Based Learning" put it simply: inference means fixing the observed variables and searching for values of the remaining variables that make the energy as small as possible, and learning means shaping the energy function so that correct configurations get lower energy than incorrect ones.

A concrete example makes this clearer. Suppose X is the pixels of a photo and Y is one of five labels: human, animal, airplane, car or truck. An EBM computes E(Y, X) for each of the five labels and predicts the label with the lowest energy. A standard neural network classifier already does something very close to this: its output scores, called logits, can be read as negative energies. What makes the energy view useful is that it does not require those scores to be turned into probabilities. A probabilistic model must divide by a normalizing constant, the sum or integral of the unnormalized scores over every possible configuration, so that probabilities add up to one. For images or sentences, that sum covers an astronomically large space and usually cannot be computed. The LeCun tutorial argues that because EBMs have no requirement for proper normalization, this problem is avoided and model designers gain flexibility in architectures and training criteria.

When a probability is needed, an energy can be converted into one using the Gibbs or Boltzmann distribution: the probability of a configuration is proportional to the exponential of minus its energy, divided by the normalizing constant, often written Z. Lower energy therefore means exponentially higher probability. This link is exact, so every probabilistic model with a positive density can be written as an EBM, and every EBM defines a probability distribution whenever Z is finite. The catch is that Z is still there; EBMs move the difficulty from defining the model to training and sampling it.

Training has to do two things at once: push energy down on real data and push it up on everything else. If a method only pushes energy down on the training examples, the model can cheat by giving every input low energy, a failure often called collapse. Song and Kingma's 2021 tutorial "How to Train Your Energy-Based Models" groups the main solutions into three families. Maximum likelihood with Markov chain Monte Carlo (MCMC) sampling draws "negative" samples from the current model and raises their energy. Score matching avoids sampling by fitting the gradient of the log density with respect to the data, which does not depend on Z. Noise contrastive estimation trains the model to tell real data apart from samples of a known noise distribution. The tutorial shows that these three approaches are theoretically connected.

The history runs through some of the best-known work in the field. The Nobel Committee's popular explanation of the 2024 Physics prize describes how John Hopfield's 1982 associative memory network was defined by an energy borrowed from models of magnetic spins: stored patterns are low-energy states, and the network recovers a pattern by updating its units so the energy falls. Geoffrey Hinton built on this with the Boltzmann machine, published in 1985, which uses tools from statistical physics to learn a probability distribution over patterns and can generate new examples. The two shared the 2024 Nobel Prize in Physics for these foundations. In 2006 LeCun's group published its tutorial, recasting conditional random fields, max-margin Markov networks and graph transformer networks as EBMs. In 2019 Du and Mordatch showed that deep convolutional EBMs trained with MCMC could generate CIFAR-10 and ImageNet images approaching the quality of generative adversarial networks of that time, and Grathwohl and colleagues showed that a standard classifier could be reinterpreted and trained as an EBM. The same year Song and Ermon's score-based model, which learns energy gradients rather than energies, helped set up the diffusion models that now dominate image generation.

Compared with other generative approaches, EBMs make the fewest structural assumptions. Autoregressive models such as GPT-style language models factor the probability into a product of next-token probabilities, which makes exact likelihood easy but fixes an order. Normalizing flows require invertible architectures so the density can be computed exactly. Variational autoencoders train an approximate lower bound. Generative adversarial networks generate samples quickly but do not provide a density at all. An EBM can use any neural network that outputs one number, can combine several energies by adding them, which Du and Mordatch used for compositional generation, and gives a score that can flag unusual inputs. Diffusion and score-based models keep much of this flexibility while sidestepping the hardest part of EBM training by learning gradients at many noise levels.

The weaknesses are real. Sampling from an EBM usually requires many iterative MCMC steps, often Langevin dynamics, which repeatedly moves a sample a small step down the energy gradient and adds noise. That makes both training and generation slow, and short chains can leave the model poorly trained in regions the sampler never visits. Training can be unstable, and because Z is unknown, the exact likelihood of a test example usually cannot be reported, which makes comparison with other models harder. Results are sensitive to step sizes, noise levels and regularization.

In practice, building a small EBM requires no special hardware beyond what a classifier of the same size would need, but training costs multiply with the number of sampling steps per update. For a learner, the most accessible routes are reading a classifier's logits as energies, which costs nothing extra, or training a score-based model, which turns the energy idea into a stable recipe with standard deep learning tools.

04Работен пример

Reading a three-class classifier as an energy-based model

  1. Take a classifier that outputs logits of 2.0 for cat, 0.5 for dog and -1.0 for car on one image.

  2. Define each label's energy as the negative logit: E(cat) = -2.0, E(dog) = -0.5, E(car) = 1.0.

  3. Predict by energy minimization: cat has the lowest energy, so it is the predicted label, matching the classifier.

  4. Compute probabilities with the Gibbs distribution: exp(2.0) = 7.39, exp(0.5) = 1.65, exp(-1.0) = 0.37, and Z = 9.41.

  5. Divide to get p(cat) = 0.79, p(dog) = 0.18 and p(car) = 0.04, which is exactly the softmax of the logits.

  6. Compute the input's own energy as -log Z = -log 9.41 = -2.24, a score that should be higher for unusual inputs.

Какво показва

A label prediction, a probability for each label and a single energy for the input itself. Check that the three probabilities sum to about 1 and match a softmax computed by any library; then compare the input energy across normal and unusual images to see whether it separates them.

Внедряване в реалния свят

A Hopfield network, introduced by John Hopfield in 1982, stores images as low-energy states and recovers a stored image from a distorted or partial copy by updating units until the network's energy stops falling.

Grathwohl and colleagues' 2019 Joint Energy-based Model reinterprets an ordinary image classifier's output scores as energies, which the authors report improves calibration and out-of-distribution detection while also letting the classifier generate images.

Du and Mordatch's 2019 EBM work generated CIFAR-10 and ImageNet images by Markov chain sampling from a learned energy function and used the same model to fill in missing image regions and plan robot hand trajectories.

Score-based generative models, such as Song and Ermon's NeurIPS 2019 method, learn the gradient of the log density, which is the negative gradient of an energy, and reached an Inception Score of 8.87 on CIFAR-10.

Structured prediction systems such as conditional random fields, used for tasks like sequence labeling in speech and handwriting recognition, can be written as EBMs, as LeCun and colleagues' 2006 tutorial shows.

Бъдещето на моделите, базирани на енергия

Research on EBMs now runs along three lines. The first is cheaper training and sampling. Song and Kingma's 2021 tutorial closed with a survey of alternative methods that were still under active research, and the high cost of MCMC sampling remains the main reason EBMs are less common than diffusion or autoregressive models for large-scale generation. The second is the overlap with diffusion and score-based models. Since Song and Ermon's 2019 NeurIPS paper showed that learning energy gradients at several noise levels gives strong image samples, much of the practical progress on energy-like models has happened under the diffusion label. Open questions include when an explicit energy, rather than only its gradient, is worth the extra cost, for example for composing several models by adding their energies or for scoring how unusual an input is. The third is using energy as a scoring function inside larger systems, for out-of-distribution detection, calibration and planning, directions already explored by Grathwohl and colleagues and by Du and Mordatch in 2019. The 2024 Nobel Prize in Physics for Hopfield networks and Boltzmann machines has also renewed attention to the physics roots of these ideas. None of this makes EBMs a replacement for current large models, but the energy view remains a common language for comparing them.

Прилагайки го на практика

Защо има значение

  • Разходи и бюджет. Архитектурните решения стимулират производителността и оперативните разходи в продължение на години.
  • По-ясни решения. Техническото образование помага на екипите да изберат правилния стек, а не само най-новия.
  • Контрол на качеството. По-добрият инженерен избор намалява инцидентите, свързани с надеждността в производството.

Внимавайте за

  • Оптимизирането на един бенчмарк може да скрие по-широки системни слабости.
  • Разходите за инфраструктура и поддръжка често се подценяват.
  • Пропуските в сигурността и видимостта могат да нарастват, когато системите стават по-сложни.

Следващи стъпки

  1. Определете целите за латентност, качество и разходи преди внедряването.
  2. Бенчмарк при реалистични условия на натоварване и данни.
  3. Мониторинг на инструмента за грешки, отклонение и въздействие върху потребителя.
  4. Подгответе пътеките за връщане назад и реакция на инцидент преди мащабиране.

Източници и допълнителна литература

  1. LeCun et al., in Predicting Structured Data, MIT Press, 2006A Tutorial on Energy-Based Learning
  2. arXiv (Song and Kingma, 2021)How to Train Your Energy-Based Models
  3. arXiv (Du and Mordatch, 2019)Implicit Generation and Generalization in Energy-Based Models
  4. arXiv (Grathwohl et al., 2019)Your Classifier is Secretly an Energy Based Model and You Should Treat it Like One
  5. NeurIPS 2019 (arXiv)Generative Modeling by Estimating Gradients of the Data Distribution
  6. The Nobel PrizeThe Nobel Prize in Physics 2024: Popular information

Продължете да изследвате

Безплатен бюлетин

Получавайте дневния AI бюлетин

Три проверени AI истории всяка делнична сутрин, написани на достъпен език. Безплатно завинаги, без реклами.

Един имейл всеки делничен ден. Отписване с едно кликване. Никога не продаваме и не споделяме вашия имейл адрес.

Тествайте себе си

Take the Energy-Based Models quiz

Незабавна обратна връзка за всеки отговор и сертификат за споделяне с подлежащ на проверка идентификационен номер, след като преминете курс.

Стартирай теста

Подкрепете безплатното AI образование. AI Understanding е 501(c)(3) с нестопанска цел — без реклами, без платен достъп, никога. Направете дарение

Често задавани въпроси

Какво представляват моделите, базирани на енергия?

An energy-based model (EBM) is a machine learning model that assigns a single number, called energy, to every possible configuration of its variables, with low energy meaning compatible or likely and high energy meaning unlikely. Because the energies never have to be normalized into probabilities that sum to one, EBMs are very flexible, which is why they serve as a common framework for classifiers, Hopfield networks, Boltzmann machines and score-based generative models.

В модел, базиран на енергия, как енергията е свързана с вероятността за точка от данни?

Чрез разпределението на Болцман p(x) е пропорционално на exp(-E(x)), така че на правдоподобните данни се приписва ниска енергия и висока вероятност.

Какво затруднява обучението на базирани на енергия модели?

Изчисляването на Z изисква сумиране или интегриране на exp(-E(x)) върху цялото входно пространство, което обикновено е неосъществимо, налагайки приблизителни методи за обучение.

Кой метод за вземане на проби обикновено се използва за вземане на проби от EBM?

Динамиката на Langevin итеративно премества пробите надолу по енергийния градиент, като същевременно добавя шум, сближавайки се към региони с ниска енергия.

Защо методи като съвпадение на резултати могат да избегнат изчисляването на функцията за разделяне Z?

Тъй като Z не зависи от x, диференцирането на log p(x) по отношение на x го анулира, оставяйки само енергийния градиент, който е проследим.

Кой от тях е класически енергиен модел?

Мрежите на Hopfield са ранен енергиен модел, който съхранява модели като нискоенергийни състояния и ги извиква чрез минимизиране на енергията.

Продължавай да учиш

Свързани ръководства

Още ръководства, избрани за тази тема

Получете безплатен AI сертификатКратък безплатен курс и изпит. Без разходи, без карта. Около 2,5 часа.Започнете безплатно →