0views0forks
ESM-C 6B is a large protein model trained on amino acid sequences. A sparse dictionary was fitted to the activations inside its layer 60, and each entry of that dictionary is an "atom": one direction the model uses as it reads a residue. This page shows 24 of them.
Every point is one amino acid residue from one protein, chosen because this atom responds to it strongly, and placed by the three axes along which the atom's residues spread out the most. Most clouds have a dense middle with a few arms reaching out of it. The arms are what the labels describe, and the lobe colouring gives each arm its own colour and leaves the middle grey.
These are co-occurrences measured on sampled residues. The model was never edited here, so a label says where an atom fires and how strongly, rather than what it causes. One quantity in the source files, a per-atom scalar named energy, is still unidentified and is left out of every interpretation.