English

Three-Body Scattering for Generative Modeling

Machine Learning 2026-07-20 v1 Computer Vision and Pattern Recognition

Abstract

Modern generative models typically rely on an adversarial critic, a prescribed noise-to-data path, or an autoregressive factorization. Instead, we show that a proper distributional energy can induce sample-level motion and provide direct regression supervision for a one-step generator. Three-Body Scattering Modeling (TBSM) for generation turns the energy distance into a constant-size per-projectile interaction: each projectile is attracted toward one real source and repelled from one independently generated source. Conditioned on the projectile and its condition, its expectation equals the 22-Wasserstein gradient-flow velocity of 12DE2(Pθ,Q)\frac12D_E^2(P_{\theta},Q). A batch of BB frozen-target events yields O(B)O(B) sample-level losses, each using one reference for its condition instead of the minibatch-wide all-pairs field used by methods such as Drifting Models. Tracking this conditional expectation online can reduce field noise. Using scattering in frozen image features, TBSM trains one-step generators on ImageNet-256, achieving FID=2.23{}=2.23 with pixel-space PixelDiT-XL and FID=1.63{}=1.63 with latent-space DiT-XL at NFE=1{}=1. We provide a design map relating diffusion-related supervision, Drift-like dynamics, and GAN-like objectives. These results establish tracked scattering as a route to high-dimensional one-step generation. Code: https://github.com/sp12138/TBSM.

Cite

@article{arxiv.2607.18198,
  title  = {Three-Body Scattering for Generative Modeling},
  author = {Peng Sun and Zhenglin Cheng and Deyuan Liu and Jun Xie and Xinyi Shang and Tao Lin},
  journal= {arXiv preprint arXiv:2607.18198},
  year   = {2026}
}

Comments

31 pages, 5 figures, and 4 tables. Code: https://github.com/sp12138/TBSM