mlsanityData to 14 Oct 2026

Comparing SGLD and a fixed-noise predictor-corrector adaptation in canonical joint energy-based models on CIFAR-10

arXiv 2608.05025 ↗First posted 5 Aug 2026cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Joint Energy-Based Models (JEM) unify classification and generation within a single network and support out-of-distribution (OOD) detection. Canonical JEM training relies on stochastic gradient Langevin dynamics (SGLD); a theoretically motivated alternative, the Predictor-Corrector (PC) sampler, has not previously undergone a systematic replication test on the canonical model. We reproduce canonical JEM on WideResNet-28-10 without normalisation layers on two independent runs and test a fixed-noise PC adaptation - with the degenerate annealed-noise predictor replaced by a deterministic gradient step - across three protocols: the adapted sampler replacing SGLD throughout the full training trajectories (115-132 epochs); cold-start generation (FID); and refinement-style multi-OOD detection (AUROC). The reconstruction reaches 92.88% test accuracy and buffer-FID 44.46 (canonical: 92.90% and 38.40). We document two failure modes: catastrophic late-training divergence with the signature of the canonical outlier-buffer mechanism (all four runs), and run-dependent SVHN OOD-discrimination dynamics. No consistent method-level advantage of the adaptation over SGLD is observed on any protocol: refinement AUROC differences stay below 0.007 across ten checkpoint-OOD pairs; seeded cold-start generation favours SGLD by about five FID points; on the training protocol a hierarchical seed-by-image bootstrap gives a 95% confidence interval on the macro-averaged AUROC difference that contains zero, while a seed-level equivalence test with two runs per method cannot establish formal equivalence. The training-protocol data are consistent both with equivalence and with a small directional effect. This outcome is consistent with theory: the guarantees of the annealed-noise PC framework do not transfer to the constant-noise regime of canonical JEM.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 16 references that are in the map. First posted on arXiv 5 Aug 2026.

Builds on

14 of 16 references.

References · 14

  1. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  2. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
  3. 2020Energy-based Out-of-distribution Detectioncited by 1,299
  4. 2018Do Deep Generative Models Know What They Don't Know?cited by 443
  5. 2019Your Classifier is Secretly an Energy Based Model and You Should Treat it Like Onecited by 405
  6. 2024Classifier-Free Guidance is a Predictor-Correctorcited by 84
  7. 2025Composition and Control with Distilled Energy Diffusion Models and Sequential Monte Carlocited by 20
  8. 2023Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihoodcited by 19
  9. 2023EGC: Image Generation and Classification via a Diffusion Energy-Based Modelcited by 16
  10. 2022Towards Bridging the Performance Gaps of Joint Energy-based Modelscited by 15
  11. 2024Diffusion for Out-of-Distribution Detection on Road Scenes and Beyondcited by 9
  12. 2024Improving Adversarial Energy-Based Model via Diffusion Processcited by 7
2 more
  1. 2025Your Classifier Can Do More: Towards Balancing the Gaps in Classification, Robustness, and Generationcited by 1
  2. 2025Scalable Energy-Based Models via Adversarial Training: Unifying Discrimination and Generationcited by 1

Built on by

Not cited yet.