Research map / Probabilistic and scientific modeling / Variational and generative sampling
Distance Marching for Generative Modeling
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes Distance Marching, a time-unconditional generative model that learns a distance-like scalar field plus a direction field with distance-focused losses (one-step and directional eikonal), enabling gradient descent or sphere-tracing sampling on CIFAR-10 and ImageNet.
Problem
Time-unconditional generative models suffer from target ambiguity: the same noisy input maps to multiple noise levels and conflicting denoising directions, and existing time-free methods are fragile or lower quality.
Approach
Learn a distance-like scalar field depending only on position, trained with a one-step loss that inverse-distance-weights closer data targets and a directional eikonal loss that fixes the offset; inference uses sphere tracing or gradient descent without time input.
Key results
- Distance Marching achieves FID 2.23 on unconditional CIFAR-10 with sphere tracing (tied best among time-unconditional methods), versus 3.34 for EM and 3.36 for EqM.
- On class-conditional ImageNet B/2, gradient descent reaches FID 32.16, beating SiT-B/2 standard FM (36.68) and Equilibrium Matching-B/2 (32.85) with 250 steps.
- Distance Marching reaches flow matching's final FID using under 60% of sampling steps, and adaptive GD cuts average steps by 29.7% (250 to 175.8).
Surprising findings
- Gradient descent with a non-curl-free direction field converges without enforcing the vector field to be a gradient of a scalar potential.
- A tiny distance predictor (about 0.1M parameters) achieves OOD AUROC avg 0.79, beating EqM (130M) on average.
Limitations
- Sphere tracing gives slightly worse final ImageNet FID than gradient descent at 250 steps.
- Sample diversity may be limited; MCMC refinement is left as future work.
- Gains on CIFAR-10 are tied with uEDM.
Experimental setup
Datasets: CIFAR-10 ImageNet-256
Benchmarks: CIFAR-10 ImageNet-256 SVHN Textures Constant
Metrics: FID sFID IS AUROC W2 Hausdorff distance Chamfer distance
Baselines: FM OT-CFM SB-CFM Energy Matching Equilibrium Matching uEDM SiT-B/2 DiT-B/2 EDM NCSN++ DDPM++ IGEBM
Builds on: flow matching SiT EDM sphere tracing Equilibrium Matching
Closest papers by meaning
- Rectifying Categorical Flows on Statistical Manifolds for One-Step Generation (NeurIPS 2026 · similarity 0.88)
- What Secrets Do Your Manifolds Hold? Understanding the Local Geometry of Generative Models (ICLR 2025 · similarity 0.88)
- Metric Flow Matching for Smooth Interpolations on the Data Manifold (NeurIPS 2024 · similarity 0.87)
- Categorical Flow Matching on Statistical Manifolds (NeurIPS 2024 · similarity 0.86)
- Diffusion Model's Generalization Can Be Characterized by Inductive Biases toward a Data-Dependent Ridge Manifold (NeurIPS 2026 · similarity 0.86)
- Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes (ICML 2024 · similarity 0.86)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
