Saved in:
Bibliographic Details
Main Authors: Varam, Dara, Abuhani, Diaa A., Zualkernan, Imran, AlDamani, Raghad, Khalil, Lujain
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.11418
Tags: Add Tag
No Tags, Be the first to tag this record!
Table of Contents:
  • Flow Matching (FM) generative models offer efficient simulation-free training and deterministic sampling, but their practical deployment is challenged by high-precision parameter requirements. We adapt optimal transport (OT)-based post-training quantization to FM models, minimizing the 2-Wasserstein distance between quantized and original weights, and systematically compare its effectiveness against uniform, piecewise, and logarithmic quantization schemes. Our theoretical analysis provides upper bounds on generative degradation under quantization, and empirical results across five benchmark datasets of varying complexity show that OT-based quantization preserves both visual generation quality and latent space stability down to 2-3 bits per parameter, where alternative methods fail. This establishes OT-based quantization as a principled, effective approach to compress FM generative models for edge and embedded AI applications.