Grooves without labels: influence functions on a class-blind STL-10 model

The CIFAR result showed the loss-landscape grooves carry a coarse animate/inanimate abstraction. But that model was trained on labelled data — could the abstraction be leaking in through the label distribution somehow? This piece removes labels entirely: a VQ-tokenised autoregressive transformer trained on STL-10's 100,000 unlabelled images at 64×64 (twice CIFAR's resolution), which never sees a single class label. We then probe it with the labelled test set purely for evaluation. Three findings:

§1Setup — no labels, higher resolution

STL-10 ships a 100k-image unlabelled split precisely for unsupervised work. We train a VQ-VAE (f=4, 1024-code book) on it, giving each 64×64 image a 16×16 = 256-code grid, then train a decoder-only transformer on those code streams — the same token-AR recipe as CIFAR, just larger sequences and never a label in sight. The labelled test set (10 classes) is tokenised only to serve as evaluation probes.

tokeniser
VQ-VAE f=4, codebook 1024, 64×64×3 → 16×16 = 256 codes
model
decoder-only transformer d256 · 6 layers · 8 heads, 256 tokens, vocab 1024, tied embeddings
training
AdamW, 30 epochs, on the 100k UNLABELLED split — class-blind
BIF
SGLD, 8 chains × 1,200 draws, ε per-checkpoint, two independent runs; probes = 1,000 labelled test images (100/class → 51/class balanced × 10)
estimator
block-centred covariance (essential at cold cells — see §2)
artefacts
arcadia-impact/small-abstraction-test-{logs, mnist-ar} on HF; STL VQ under stl-vq/
STL-10 originals and VQ reconstructions at 64px: blurry but object and scene structure preserved.
Fig. 1 · VQ reconstructions at 64px — blurrier than the originals but object/scene structure survives into the code stream.
Two rows of free samples from the class-blind model at epochs 5 and 30, decoded through the VQ: painterly but with recognizable animals, and sky/water/mountain scenes.
Fig. 2 · Free rollouts from the class-blind model. By epoch 30 the samples are painterly but legible — four-legged animals, birds, and sky/water/mountain scenes — learned with no class supervision at all.

§2Here the block estimator is mandatory

On CIFAR the naive covariance estimator was already fine. Not here. At the cold checkpoints (the early cells, where SGLD calibrated to a very small step) the chain wanders slowly and a global common mode dominates: every probe correlates with every other at ≈ 0.9, and within-class is indistinguishable from between-class. The raw estimator is highly reproducible (it faithfully re-measures the common mode) but carries no class signal. Block-centring — high-passing that slow mode away — is what surfaces the class contrast.

Left: raw covariance within and between-class correlations both near 0.9 at early epochs, indistinguishable. Right: block-centred, within-class sits clearly above between-class at every epoch.
Fig. 3 · Raw vs block-centred, per checkpoint. Raw (left): within ≈ between ≈ 0.9 at the cold early cells — all common mode, no class structure. Block-centred (right): within-class stands above between-class throughout. The raw-vs-block ranking is checkpoint-dependent, and here block is essential.

§3Grooves: present, weak, fading

With block-centring, same-class probes co-fluctuate more than cross-class at every checkpoint, and the contrast fades from training as before (block-5 fractional contrast +0.85 at epoch 2 → +0.15 at epoch 30). The absolute correlations are small, as expected for a class-blind model at higher resolution.

Within, between and difference (groove signal) block-centred correlations across epochs 2, 10, 30; the groove signal is positive but shrinks with training.
Fig. 4 · Block-centred groove signal (within − between) across training — positive throughout, largest early.

§4The point: a label-blind model still grooves by category

Clustering the checkpoint-ensembled block BIF: ten-way recovery is at chance (ARI ≈ 0.05), as on CIFAR. But the two-way animate/inanimate split lands at ARI ≈ 0.37 — one cluster 86% animal, the other 74% vehicle — while clustering the pixels themselves gives ≈ 0.00. And the ensemble is reproducible: two independent SGLD runs' ensemble matrices agree at r ≈ 0.93. The coarse semantic abstraction is genuinely there, recovered without labels at either training or clustering time, and it is not visible in appearance.

Left: BIF 10-way ARI 0.05, BIF animal/vehicle 0.37, pixel animal/vehicle 0.00. Right: the two clusters are 86% and 74% pure for animal and vehicle.
Fig. 5 · Left: fine class at chance; the coarse animal/vehicle split recovered by the BIF (0.37) and not by pixels (0.00). Right: the two top-level clusters are animal- and vehicle-majority.

A model that never saw a label still carves a loss landscape whose grooves encode the animate/inanimate distinction — recoverable unsupervised, three-times better than from pixels. The abstraction comes from generative training alone.

The hierarchy below is interactive — hover any cluster to see its member images; click to pin, click away or Esc to dismiss. The green/orange rails mark animal- vs vehicle-majority. As k grows the top animate/inanimate split holds while the per-class colours stay mixed (fine-class ARI ≈ 0.06 throughout).

k = 2 ARI 0.07 k = 3 ARI 0.07 k = 4 ARI 0.07 k = 6 ARI 0.06 k = 8 ARI 0.06 k = 10 ARI 0.07 animal-majority vehicle-majority hover a cluster to see its images · click to pin
Fig. 6 · Interactive cluster hierarchy of the class-blind STL BIF, k = 2…10 (ARI under each column is vs the ten fine classes — chance throughout). Rails: green = animal-majority, orange = vehicle-majority. Hover for member images.

§5What it means, and the limits

This is the cleanest version of the thread's recurring finding: the animate/inanimate abstraction shows up in the loss geometry of a model trained with no labels at all, so it cannot be an artefact of label supervision — it emerges from modelling the images themselves. It is also more than surface appearance (pixels recover it at chance). The limits are the same as CIFAR's: it is a coarse abstraction, not the fine taxonomy, and a 512/1024-code VQ discards the within-category detail that would separate a cat from a dog. The natural next probe is a better tokeniser (more codes / higher resolution) to ask whether finer structure ever appears, or the product-structure question — whether the grooves factor into independent axes (identity × pose, say) rather than a single coarse partition. The next piece scales to 200 classes to push on exactly this.

Code · small-abstraction-test (class-blind token-AR pipeline, generalised VQ grid, block-centred SGLD estimator).
Artefacts · HF small-abstraction-test-logs (SGLD traces, BIF matrices, 30-epoch checkpoints) · VQ tokeniser under stl-vq/.
Compute · 1× H100 ≈ 2.5 h ≈ $8 (VQ + class-blind AR train + three gated BIF cells at 8×1,200 draws). Pod shared with the next (Tiny ImageNet) run.