5 papers
SoftCache: Adaptive Step Size Caching for Diffusion Transformers
Shimon Malnick, Nadav Orenstein, Ohad Fried +1
Diffusion and flow models generate a sample over iterative steps, each a full pass through a large network, and that pass dominates generation cost. Training-free caching accelerat…
Optimal Transport Flow Matching by Design
Shimon Malnick, Matan Rusanovsky, Ohad Fried +1
Flow matching models learn to transport samples from a simple prior distribution to a complex data distribution. When prior-data pairs are coupled via optimal transport (OT), the l…
Talking Points: Describing and Localizing Pixels
Matan Rusanovsky, Shimon Malnick, Shai Avidan
Vision-language models have achieved remarkable success in cross-modal understanding. Yet, these models remain limited to object-level or region-level grounding, lacking the capabi…
Memories of Forgotten Concepts
Matan Rusanovsky, Shimon Malnick, Amir Jevnisek +2
Diffusion models dominate the space of text-to-image generation, yet they may produce undesirable outputs, including explicit content or private data. To mitigate this, concept abl…
Taming Normalizing Flows
Shimon Malnick, Shai Avidan, Ohad Fried
We propose an algorithm for taming Normalizing Flow models - changing the probability that the model will produce a specific image or image category. We focus on Normalizing Flows…