HƯỚNG DẪN KỸ THUẬT

Mô hình dựa trên năng lượng

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.

Nghe hướng dẫn nàyAI thuật lại · ~9 min

Trên trang nàyđọc 10 phút
  1. Tổng quan
  2. Những điểm chính rút ra
  3. Lặn sâu
  4. Reading a three-class classifier as an energy-based model
  5. Triển khai trong thế giới thực
  6. Tương lai của các mô hình dựa trên năng lượng
  7. Đưa nó vào thực tế
  8. Nguồn tham khảo và đọc thêm
  9. Tiếp tục khám phá
  10. Câu hỏi thường gặp

Tổng quan

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.

Những điểm chính rút ra

  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.

Lặn sâu

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.

04Ví dụ hoạt động

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.

Những gì nó cho thấy

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.

Triển khai trong thế giới thực

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.

Tương lai của các mô hình dựa trên năng lượng

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.

Đưa nó vào thực tế

Tại sao nó quan trọng

  • Chi phí và ngân sách. Các quyết định về kiến ​​trúc sẽ thúc đẩy hiệu suất và chi phí vận hành trong nhiều năm.
  • Quyết định rõ ràng hơn. Giáo dục kỹ thuật giúp các nhóm chọn nhóm phù hợp chứ không chỉ nhóm mới nhất.
  • Kiểm soát chất lượng. Lựa chọn kỹ thuật tốt hơn làm giảm sự cố về độ tin cậy trong sản xuất.

coi chừng

  • Tối ưu hóa một điểm chuẩn có thể che giấu những điểm yếu của hệ thống rộng hơn.
  • Chi phí cơ sở hạ tầng và bảo trì thường được đánh giá thấp.
  • Khoảng cách về bảo mật và khả năng quan sát có thể tăng lên khi hệ thống trở nên phức tạp hơn.

Các bước tiếp theo

  1. Xác định các mục tiêu về độ trễ, chất lượng và chi phí trước khi triển khai.
  2. Điểm chuẩn trong điều kiện tải và dữ liệu thực tế.
  3. Giám sát thiết bị về lỗi, độ lệch và tác động của người dùng.
  4. Chuẩn bị đường dẫn khôi phục và ứng phó sự cố trước khi mở rộng quy mô.

Nguồn tham khảo và đọc thêm

  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

Tiếp tục khám phá

Bản tin miễn phí

Nhận bản tóm tắt AI hàng ngày

Ba câu chuyện AI đã được xác minh vào mỗi buổi sáng các ngày trong tuần, được viết bằng tiếng Anh đơn giản. Miễn phí mãi mãi, không có quảng cáo.

Một email mỗi ngày trong tuần. Hủy đăng ký trong một cú nhấp chuột. Chúng tôi không bao giờ bán hoặc chia sẻ địa chỉ của bạn.

Tự kiểm tra

Take the Energy-Based Models quiz

Phản hồi tức thì về mọi câu trả lời và chứng chỉ có thể chia sẻ với ID có thể xác minh sau khi bạn vượt qua khóa học.

Bắt đầu bài kiểm tra

Hỗ trợ giáo dục AI miễn phí. AI Understanding là tổ chức phi lợi nhuận 501(c)(3) — không bao giờ có quảng cáo, không có tường phí. Hãy quyên góp

Câu hỏi thường gặp

What is Energy-Based Models?

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.

Trong mô hình dựa trên năng lượng, năng lượng liên quan như thế nào đến xác suất của một điểm dữ liệu?

Thông qua phân phối Boltzmann, p(x) tỷ lệ với exp(-E(x)), do đó dữ liệu hợp lý được gán năng lượng thấp và xác suất cao.

Điều gì khiến việc đào tạo các mô hình dựa trên năng lượng trở nên khó khăn?

Việc tính toán Z yêu cầu tính tổng hoặc tích phân exp(-E(x)) trên toàn bộ không gian đầu vào, điều này thường không khả thi, buộc phải áp dụng các phương pháp huấn luyện gần đúng.

Phương pháp lấy mẫu nào thường được sử dụng để lấy mẫu từ EBM?

Động lực học Langevin liên tục di chuyển các mẫu xuống dốc dọc theo gradient năng lượng đồng thời tăng thêm nhiễu, hội tụ về các vùng năng lượng thấp.

Tại sao các phương pháp như so khớp điểm có thể tránh tính toán hàm phân vùng Z?

Vì Z không phụ thuộc vào x, nên vi phân log p(x) theo x sẽ hủy bỏ nó, chỉ để lại gradient năng lượng, có thể điều chỉnh được.

Mô hình nào sau đây là mô hình dựa trên năng lượng cổ điển?

Mạng Hopfield là một mô hình dựa trên năng lượng ban đầu lưu trữ các mẫu ở trạng thái năng lượng thấp và gọi lại chúng bằng cách giảm thiểu năng lượng.