Understanding temperature tuning in energy-based models
Not on the map. Why?
Abstract (arXiv)
Generative models of complex systems often require post-hoc parameter adjustments to produce useful outputs. For example, energy-based models for protein design are sampled at an artificially low ''temperature'' to generate novel, functional sequences. This temperature tuning is a common yet poorly understood heuristic used across machine learning contexts to control the trade-off between generative fidelity and diversity. Here, we develop an interpretable, physically motivated framework to explain this phenomenon. We demonstrate that in systems with a large ''energy gap'' - separating a small fraction of meaningful states from a vast space of unrealistic states - learning from sparse data causes models to systematically overestimate high-energy state probabilities, a bias that lowering the sampling temperature corrects. More generally, we characterize how the optimal sampling temperature depends on the interplay between data size and the system's underlying energy landscape. Crucially, our results show that lowering the sampling temperature is not always desirable; we identify the conditions where raising it results in better generative performance. Our framework thus casts post-hoc temperature tuning as a diagnostic tool that reveals properties of the true data distribution and the limits of the learned model.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 47 references that are in the map. First posted on arXiv 9 Dec 2025.
Builds on
22 of 47 references.
References · 22
- 2017Attention Is All You Needcited by 59,932
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2015Variational Inference with Normalizing Flowscited by 2,161
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applicationscited by 1,107
- 2022Beyond neural scaling laws: beating power law scaling via data pruningcited by 519
- 2020Hopfield Networks is All You Needcited by 461
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2018A high-bias, low-variance introduction to Machine Learning for physicistscited by 108
10 more
- 2015How (not) to Train your Generative Model: Scheduled Sampling, Likelihood, Adversary?cited by 76
- 2023Not All Semantics are Created Equal: Contrastive Self-supervised Learning with Automatic Temperature Individualizationcited by 25
- 2024Transcendence: Generative Models Can Outperform The Experts That Train Themcited by 18
- 2022Stochastic Constrained DRO with a Complexity Independent of Sample Sizecited by 16
- 2013Searching for collective behavior in a network of real neuronscited by 9
- 2018Boltzmann Encoded Adversarial Machinescited by 3
- 2023Exact minimax entropy models of large-scale neuronal activitycited by 3
- 2023If there is no underfitting, there is no Cold Posterior Effectcited by 2
- 2023Biases in Inverse Ising Estimates of Near-Critical Behaviourcited by 1
- 2024Ratio Divergence Learning Using Target Energy in Restricted Boltzmann Machines: Beyond Kullback--Leibler Divergence Learningcited by 1
Built on by
Not cited yet.