Denoiser

Pinned implementation and paper LDLM @ e18c42a, not yet public · arXiv:2605.07933v1 · OpenWebText · 1024 · checkpoint, tokenizer and saved statistics not verified · How to read

Denoiser

Output projection

×

learned

Map the modulated features to the predicted clean latents.

z^0=xoutWout+bout{\htmlData{light=output output-projection}{\hat z_0}}={\htmlData{light=final-modulate}{x_{\mathrm{out}}}}W_{\mathrm{out}}+b_{\mathrm{out}}
in
xoutx_{\mathrm{out}} [B batch, T positions, 768 d]
out
z^0\hat z_0 [B batch, T positions, 768 d]
parameters
590,592
Parameters

590,592 parameters per instance.

Exact per-instance accounting
ComponentCalculationParameters
Weight matrix768×768768 \times 768589,824
Output bias768768768
Per-instance total590,592

The reference module holds the weight output by input, [out, in].

Derived from the configured matrix dimensions; checkpoint tensors have not been independently counted.

Explanation

Its weights and bias start at zero, so an untrained denoiser predicts zero latents.

Sources1

Direct evidence

The output layer: gain-only LayerNorm, a shift and a scale from the time condition, and an affine projection initialized at zero.

The implementation, pinned at revision e18c42acdfb7, not yet public