TO-FLOW: Efficient Continuous Normalizing Flows with Temporal Optimization adjoint with Moving Speed
Not on the map. Why?
Abstract (arXiv)
Continuous normalizing flows (CNFs) construct invertible mappings between an arbitrary complex distribution and an isotropic Gaussian distribution using Neural Ordinary Differential Equations (neural ODEs). It has not been tractable on large datasets due to the incremental complexity of the neural ODE training. Optimal Transport theory has been applied to regularize the dynamics of the ODE to speed up training in recent works. In this paper, a temporal optimization is proposed by optimizing the evolutionary time for forward propagation of the neural ODE training. In this appoach, we optimize the network weights of the CNF alternately with evolutionary time by coordinate descent. Further with temporal regularization, stability of the evolution is ensured. This approach can be used in conjunction with the original regularization approach. We have experimentally demonstrated that the proposed approach can significantly accelerate training without sacrifying performance over baseline models.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Builds on 57 of its 78 references that are in the map. First posted on arXiv 19 Mar 2022.
Builds on
57 of 78 references.
References · 57
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2016Density estimation using Real NVPcited by 2,042
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,545
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,323
- 2014NICE: Non-linear Independent Components Estimationcited by 1,058
- 2017Masked Autoregressive Flow for Density Estimationcited by 684
- 2016Improving Variational Inference with Inverse Autoregressive Flowcited by 640
38 more
- 2019Neural Spline Flowscited by 596
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 594
- 2019Augmented Neural ODEscited by 397
- 2018WaveGlow: A Flow-based Generative Network for Speech Synthesiscited by 309
- 2019Flow++: Improving Flow-Based Generative Models with Variational Dequantization and Architecture Designcited by 262
- 2020Equivariant Flows: Exact Likelihood Generative Learning for Symmetric Densitiescited by 250
- 2020How to train your neural ODE: the world of Jacobian and kinetic regularizationcited by 228
- 2018Neural Autoregressive Flowscited by 208
- 2019BSP-Net: Generating Compact Meshes via Binary Space Partitioningcited by 206
- 2018Neural Importance Samplingcited by 155
- 2019NASA: Neural Articulated Shape Approximationcited by 149
- 2020Riemannian Continuous Normalizing Flowscited by 145
- 2020OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transportcited by 143
- 2020Normalizing Flows on Tori and Spherescited by 117
- 2019GraphNVP: An Invertible Flow Model for Generating Molecular Graphscited by 116
- 2019Noise Flow: Noise Modeling with Conditional Normalizing Flowscited by 108
- 2019Neural SDE: Stabilizing Neural ODE Networks with Stochastic Noisecited by 101
- 2019Unconstrained Monotonic Neural Networkscited by 92
- 2019ANODE: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEscited by 84
- 2019On Robustness of Neural Ordinary Differential Equationscited by 83
- 2020Learning Differential Equations that are Easy to Solvecited by 81
- 2017Maximum Principle Based Algorithms for Deep Learningcited by 78
- 2020Adaptive Checkpoint Adjoint Method for Gradient Estimation in Neural ODEcited by 76
- 2019Latent Normalizing Flows for Discrete Sequencescited by 67
- 2020Targeted free energy estimation via learned mappingscited by 67
- 2020STEER: Simple Temporal Regularization For Neural ODEscited by 58
- 2018FloWaveNet : A Generative Flow for Raw Audiocited by 54
- 2019Leveraging exploration in off-policy algorithms via normalizing flowscited by 46
- 2021MALI: A memory efficient and reverse accurate integrator for Neural ODEscited by 42
- 2020Vid-ODE: Continuous-Time Video Generation with Neural Ordinary Differential Equationcited by 40
- 2019Improving Exploration in Soft-Actor-Critic with Normalizing Flows Policiescited by 38
- 2018Layer-Parallel Training of Deep Residual Neural Networkscited by 35
- 2019Deep learning as optimal control problems: models and numerical methodscited by 34
- 2020Model-based Reinforcement Learning for Semi-Markov Decision Processes with Neural ODEscited by 32
- 2019SNODE: Spectral Discretization of Neural ODEs for System Identificationcited by 26
- 2021Multi-scale Neural ODEs for 3D Medical Image Registrationcited by 18
- 2019Universal audio synthesizer control with normalizing flowscited by 17
- 2018Randomized Value Functions via Multiplicative Normalizing Flowscited by 15
50 most cited of 57 All 57 (JSON)
Built on by
Citing papers per year: 2022: 1; 2023: 0; 2024: 2; 2025: 3; 2026: 2.
8 papers.
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2025X-IL: Exploring the Design Space of Imitation Learning Policiescited by 10
- 2024Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Modelscited by 5
- 2026A Minimum Variance Path Principle for Accurate and Stable Score-Based Density Ratio Estimationcited by 3
- 2024Sparse Variational Student-t Processes for Heavy-tailed Modelingcited by 2
- 2025Efficient Manifold-Constrained Neural ODE for High-Dimensional Datasetscited by 2
- 2026Flow matching on homogeneous spacescited by 1
- 2025Instance Relation Learning Network with Label Knowledge Propagation for Few-shot Multi-label Intent Detectioncited by 0
- 2026Flow matching on homogeneous spaces25 Mar 2026
- 2026A Minimum Variance Path Principle for Accurate and Stable Score-Based Density Ratio Estimation31 Jan 2026
- 2025Instance Relation Learning Network with Label Knowledge Propagation for Few-shot Multi-label Intent Detection9 Oct 2025
- 2025Efficient Manifold-Constrained Neural ODE for High-Dimensional Datasets5 Oct 2025
- 2025X-IL: Exploring the Design Space of Imitation Learning Policies17 Feb 2025
- 2024Sparse Variational Student-t Processes for Heavy-tailed Modeling13 Aug 2024
- 2024Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Models6 Jul 2024
- 2022Flow Matching for Generative Modeling6 Oct 2022