What Is FS-DFM?
FS-DFM (Few-Step Discrete Flow-Matching) is a diffusion language model introduced by Apple in collaboration with The Ohio State University for fast generation of long texts. The model trains with the number of sampling steps as an explicit parameter, enabling it to generate high-quality text within fewer steps. FS-DFM combines a reliable update rule and strong teacher guidance to ensure accurate probability updates without overshooting. In language modeling benchmarks, FS-DFM achieves the perplexity level of a 1024-step discrete flow baseline with 8-step sampling, while speeding up sampling by 128 times, significantly improving efficiency and throughput.
Main Features of FS-DFM
-
Efficient Sampling: Achieves the effect of a traditional 1024-step diffusion model with 8-step sampling, with a 128x speedup.
-
Long Text Generation: Suitable for generating long texts, addressing the efficiency bottleneck of traditional autoregressive models in long-sequence generation.
-
Stability and Controllability: Through optimized sampling rules and teacher guidance, it ensures a stable and easily controllable generation process.
Technical Principles of FS-DFM
-
Based on the Discrete Flow-Matching (DFM) Framework: The technical foundation of FS-DFM is Discrete Flow-Matching (DFM), which generates text by learning the probability path from a noise distribution to a target distribution. DFM uses the properties of continuous-time Markov chains (CTMC) to provide a method for parallel text generation, which can significantly improve generation efficiency compared to traditional autoregressive models.
-
Explicit Sampling Steps: FS-DFM treats the number of sampling steps as an explicit parameter, training the model to maintain consistent generation quality under different step budgets. This means the model achieves quality comparable to traditional methods (such as 1024 steps) with fewer steps (such as 8 steps), greatly reducing the computational resources and time required for generation.
-
Reliable Update Rule: To ensure the stability and accuracy of few-step generation, FS-DFM introduces a reliable update rule. The rule avoids the problem of overshooting by controlling the direction and magnitude of probability updates, making the generation process more stable.
-
Cumulative Scalar: FS-DFM introduces the concept of a cumulative scalar, which provides the correct probability flow for each finite step by integrating the scheduler's rate over a time interval. This enables the model to achieve effective updates in few-step generation, maintaining sufficient update strength in the early steps and avoiding stagnation in the generation process.
FS-DFM Project Address
- arXiv Technical Paper:https://arxiv.org/pdf/2509.20624
Application Scenarios of FS-DFM
-
Content Creation: Rapidly generate high-quality long texts, used for creating articles, stories, news reports, etc., helping content creators improve efficiency.
-
Intelligent Customer Service: In intelligent customer service systems, quickly generate detailed replies, improving response speed and user experience.
-
Language Translation: Through efficient generation of long texts, it is used for translating lengthy documents, improving translation efficiency and quality.
-
Creative Writing: Provides inspiration for writers and creative professionals, quickly generating story outlines, scripts, or poems, etc., stimulating creative ideas.
-
Education: Generates educational materials, such as course outlines, teaching cases, or practice questions, assisting teachers in preparing teaching content.