Taming Outlier Tokens in Diffusion Transformers

Apple ML Research··

We study outlier tokens in Diffusion Transformers (DiTs) for image generation. Prior work has shown that Vision Transformers (ViTs) can produce a small number of high-norm tokens that attract disproportionate attention while carrying limited local information, but their role in generative models remains underexplored. We show that this phenomenon appears in both the encoder and denoiser of modern Representation Autoencoder (RAE)-DiT pipelines: pretrained ViT encoders can produce outlier represen...

Read full article →

Related Articles

OpenAI’s o1 correctly diagnosed 67% of ER patients vs. 50-55% by triage doctors
donsupreme · Hacker News · 3mo ago
Accelerating Gemma 4: faster inference with multi-token prediction drafters
amrrs · Hacker News · 3mo ago
A couple million lines of Haskell: Production engineering at Mercury
unignorant · Hacker News · 3mo ago
Using “underdrawings” for accurate text and numbers
samcollins · Hacker News · 3mo ago
ProgramBench: Can language models rebuild programs from scratch?
jonbaer · Hacker News · 3mo ago