PyTorch for density estimation research
Top 53.6% on sourcepulse
This repository provides PyTorch implementations of several normalizing flow models for density estimation, including BNAF, Glow, MAF, RealNVP, and planar flows. It is targeted at researchers and practitioners in generative modeling and machine learning who need to experiment with or reproduce results from these advanced density estimation techniques. The benefit is a unified codebase for exploring and comparing different normalizing flow architectures.
How It Works
The project reimplements key normalizing flow architectures, each designed for density estimation. BNAF and MAF leverage autoregressive properties for efficient computation, while Glow utilizes invertible 1x1 convolutions. RealNVP employs coupling layers for invertibility. These methods transform a simple base distribution (e.g., Gaussian) into a complex target distribution through a series of invertible transformations, allowing for exact likelihood computation.
Quick Start & Requirements
pip install -r requirements.txt
(specific requirements vary per model).bnaf.py
, glow.py
, maf.py
, and planar_flow.py
are used for training, plotting, evaluation, and generation.Highlighted Details
Maintenance & Community
The repository appears to be a personal project, with no explicit mention of active maintenance, community channels (like Discord/Slack), or a roadmap.
Licensing & Compatibility
The README does not explicitly state a license. Compatibility for commercial use or closed-source linking is not specified.
Limitations & Caveats
The project relies on older versions of PyTorch (1.0) and Python (3.6), which may pose compatibility issues with current environments. The lack of explicit licensing and community support could be a concern for long-term adoption.
4 years ago
1+ week