1Local context
Which spatial directions help predict a cell’s contents?
Height and width run within comb sides; depth runs across them. Each option shows the inputs to a separately trained model configuration.
Class embeddings
Each content type becomes a set of learned numerical features, its embedding. The shared table E supplies these features wherever that type occurs.
Masked convolution
The masked convolution combines neighbor features using learned weights for each offset. Its output C describes the local pattern while excluding the masked cell.
How masked convolution works
A kernel assigns weights to positions around the masked cell. Each weight multiplies the features at the matching position; adding these contributions produces a local description. The center weight is zero, so the cell cannot supply its own answer.
Match the kernel to the neighborhood
NegativePositive
Kernel color shows the mean weight at each offset for this context feature.
This figure isolates one learned context feature. For each neighbor, the model computes a dot product between its class embedding and the kernel weights at that offset, then adds the shared baseline.
Predict from the surrounding pattern
Fitted DD1 model
The outlined prediction matches the masked input cell. The animation shows predictions accumulating across the patch.
Real DD1 cells and fitted kernel weights are used here. The probabilities include all context features, global position and interactions. The animation visits positions in sequence; the model computes them in parallel.