Bidirectional Normalizing Flow: From Data to Noise and Back

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lu, Yiyang, Sun, Qiao, Wang, Xianbang, Jiang, Zhicheng, Zhao, Hanhong, He, Kaiming
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918244722409472
author Lu, Yiyang
Sun, Qiao
Wang, Xianbang
Jiang, Zhicheng
Zhao, Hanhong
He, Kaiming
author_facet Lu, Yiyang
Sun, Qiao
Wang, Xianbang
Jiang, Zhicheng
Zhao, Hanhong
He, Kaiming
contents Normalizing Flows (NFs) have been established as a principled framework for generative modeling. Standard NFs consist of a forward process and a reverse process: the forward process maps data to noise, while the reverse process generates samples by inverting it. Typical NF forward transformations are constrained by explicit invertibility, ensuring that the reverse process can serve as their exact analytic inverse. Recent developments in TARFlow and its variants have revitalized NF methods by combining Transformers and autoregressive flows, but have also exposed causal decoding as a major bottleneck. In this work, we introduce Bidirectional Normalizing Flow ($\textbf{BiFlow}$), a framework that removes the need for an exact analytic inverse. BiFlow learns a reverse model that approximates the underlying noise-to-data inverse mapping, enabling more flexible loss functions and architectures. Experiments on ImageNet demonstrate that BiFlow, compared to its causal decoding counterpart, improves generation quality while accelerating sampling by up to two orders of magnitude. BiFlow yields state-of-the-art results among NF-based methods and competitive performance among single-evaluation ("1-NFE") methods. Following recent encouraging progress on NFs, we hope our work will draw further attention to this classical paradigm.
format Preprint
id arxiv_https___arxiv_org_abs_2512_10953
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Bidirectional Normalizing Flow: From Data to Noise and Back
Lu, Yiyang
Sun, Qiao
Wang, Xianbang
Jiang, Zhicheng
Zhao, Hanhong
He, Kaiming
Machine Learning
Computer Vision and Pattern Recognition
Normalizing Flows (NFs) have been established as a principled framework for generative modeling. Standard NFs consist of a forward process and a reverse process: the forward process maps data to noise, while the reverse process generates samples by inverting it. Typical NF forward transformations are constrained by explicit invertibility, ensuring that the reverse process can serve as their exact analytic inverse. Recent developments in TARFlow and its variants have revitalized NF methods by combining Transformers and autoregressive flows, but have also exposed causal decoding as a major bottleneck. In this work, we introduce Bidirectional Normalizing Flow ($\textbf{BiFlow}$), a framework that removes the need for an exact analytic inverse. BiFlow learns a reverse model that approximates the underlying noise-to-data inverse mapping, enabling more flexible loss functions and architectures. Experiments on ImageNet demonstrate that BiFlow, compared to its causal decoding counterpart, improves generation quality while accelerating sampling by up to two orders of magnitude. BiFlow yields state-of-the-art results among NF-based methods and competitive performance among single-evaluation ("1-NFE") methods. Following recent encouraging progress on NFs, we hope our work will draw further attention to this classical paradigm.
title Bidirectional Normalizing Flow: From Data to Noise and Back
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.10953