Repo

IDM

Runs 1-step or 2-step text-to-image sampling from one checkpoint and improves text-heavy prompts using EMA feature-moment matching plus internal DMD.

GitHub ↗Oct 29, 2026Paper ↗

  • cv
  • text-to-image
  • distillation
  • long-text-rendering
Method pipeline showing DMD loss and feature-moment loss architecture
Method pipeline showing DMD loss and feature-moment loss architecture

What it is

One- and two-step text-to-image generation improved by matching internal feature moments, without an external encoder. One checkpoint serves both 1 and 2 NFE sampling. Generate images in 1 or 2 transformer evaluations while improving long-text rendering by adding feature-moment matching in the generator's own EMA feature space on top of internal DMD.

Key results

  • 94.25% LongText-CN and 90.20% LongText-EN at 2 NFE
  • 0.25s at 1 NFE and 0.30s at 2 NFE on one A100, BF16
  • EMA moment loss adds +0.41 LongText-EN over internal DMD alone

Shows that a generator's own EMA features can replace an external encoder for moment matching in few-step text-to-image distillation.

One useful ML email a week.

Worth your time, not all of your time.

We track new ML work across papers, code, and data, then keep only the pieces worth a closer look. Selected, not collected.

Papers.Data.Code

The Weekly ML Signal

Selected, not collected.

One useful ML email a week: the papers, repos and datasets worth your attention.

Worth your time, not all of your time.