Latent encoder

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

Latent encoder

Learned latent queries

×

learned

1,024 learned vectors, the same for every sequence, repeated over the batch.

ub,t=Utu_{b,t}=U_t
out
uu [B batch, T positions, 768 d]
parameters
786,432
Parameters

786,432 parameters owned by this component

Exact per-instance accounting
ComponentCalculationParameters
table1,024×7681{,}024 \times 768786,432
Per-instance total786,432

Derived from the pinned implementation’s constructors and configuration. The checkpoint’s tensors have not been counted. 306,011,162 of these parameters are trained; frozen GPT-2 and the decoder’s mask vector, 124,442,112, are not.

A table of 1,024 rows of 768 features, initialized with standard deviation 0.02.

Explanation

The table is a parameter read whole, not looked up by token: row tt becomes the latent of position tt.

Sources2

Direct evidence

The encoder owns the learned latent queries, a position table, an input and an output RMSNorm and the blocks.

The implementation, pinned at revision e18c42acdfb7, not yet public

Direct evidence

The encoder adds positions to the features, normalizes the learned queries, applies the blocks, each attending to the features, and normalizes the result.

The implementation, pinned at revision e18c42acdfb7, not yet public