C$^{2}$R: Cross-sample Consistency Regularization Mitigates Feature Splitting and Absorption in Sparse Autoencoders
About
Sparse Autoencoders (SAEs) are widely used to interpret large language models by decomposing activations into sparse, human-understandable features, but scaling to large dictionaries exposes fundamental challenges. Systematic studies reveal pervasive feature splitting that fragments coherent concepts into non-atomic latents and widespread feature absorption that creates arbitrary exceptions in general features, severely compromising latent reliability. These issues stem from inconsistent latent assignment across samples: without cross-sample constraints, per-sample optimization often allows a single underlying concept to be inconsistently distributed across multiple redundant or interfering latents. To address this, we introduce C$^2$R (\underline{\textbf{C}}ross-sample \underline{\textbf{C}}onsistency \underline{\textbf{R}}egularization). C$^2$R explicitly encourages that each semantic feature is consistently represented by a unified latent across the batch by penalizing the co-activation of directionally similar latents. Comprehensive evaluation demonstrates that C$^2$R effectively mitigates both splitting and absorption while, crucially, preserving reconstruction fidelity, providing a principled solution that enhances latent interpretability without degrading model performance. Source code is available at https://github.com/hr-jin/Cross-sample-Consistency-Regularization.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Causal intervention | Gemma-2-2B (layer 12) | SCR10.47 | 5 | |
| Sparse Autoencoder Feature Evaluation | OpenWebText 1B tokens Gemma-2-2B (train) | KL Score0.9631 | 5 | |
| Sparse Autoencoding | Gemma-2-2B activations (layer 20) | KL Score0.9727 | 5 | |
| Sparse Autoencoder Quality Evaluation | The Pile Gemma-2-2B (test) | KL Score0.9618 | 4 | |
| Sparse Autoencoding | Gemma-2-2B (layer 12) | KL Score0.9628 | 4 | |
| Sparse Autoencoder Disentanglement and Interpretability | Llama-3-8B (layer 20) | KL Score0.9858 | 4 | |
| Sparse Autoencoder Evaluation | Qwen3-8B layer 20 | KL Score0.9874 | 4 |