기술 가이드

에너지 기반 모델

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 min read
  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 브리핑 받기

매주 평일 아침마다 3개의 검증된 AI 스토리를 일반 영어로 작성합니다. 무료이며, 광고 없음.

매주 평일마다 한 번 이메일을 받습니다. 한 번 클릭으로 구독 해지. 주소를 판매하거나 공유하지 않습니다.

자신을 테스트해 보세요

Take the Energy-Based Models quiz

모든 답변에 즉각적인 피드백과, 과정을 통과하면 검증 가능한 ID가 있는 공유 가능한 인증서를 제공합니다.

퀴즈 시작

무료 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에 의존하지 않기 때문에 x에 대해 log p(x)를 미분하면 이를 취소하고 다루기 쉬운 에너지 기울기만 남깁니다.

다음 중 고전적인 에너지 기반 모델은 무엇입니까?

Hopfield 네트워크는 패턴을 저에너지 상태로 저장하고 에너지를 최소화하여 이를 호출하는 초기 에너지 기반 모델입니다.