A language model’s internal activity can be dominated by one direction, called the first principal component (PC1). That can make a reconstruction score look excellent even when much of the other activity is missed, and make the most important features change between training runs.
(SAEs) rebuild a model’s internal activity using a small set of learned features. The researchers trained them after removing PC1 from the input, then added that direction back to the reconstruction.
One SAE explained 99.96% of total variation while missing 33% of the variation beyond PC1. Removing PC1 before training made top-ranked features more consistent and improved reconstruction and model behavior, so evaluations better reflected what the SAE captured beyond the dominant direction.
