AlignSAE: Concept-Aligned Sparse Autoencoders
AlignSAE is introduced, a method that aligns SAE features with a predefined ontology through a "pre-train, then post-train"curriculum and achieves disentanglement improvements of $9$--$\infty\times$ over standard SAEs, and further supports multi-hop reasoning and a mechanistic probe of grokking-like generalization dyna...