Temporal Sparse Autoencoders: Leveraging the Sequential Nature of Language for Interpretability

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Bhalla, Usha, Oesterling, Alex, Verdun, Claudio Mayrink, Lakkaraju, Himabindu, Calmon, Flavio P.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912927353667584
author Bhalla, Usha
Oesterling, Alex
Verdun, Claudio Mayrink
Lakkaraju, Himabindu
Calmon, Flavio P.
author_facet Bhalla, Usha
Oesterling, Alex
Verdun, Claudio Mayrink
Lakkaraju, Himabindu
Calmon, Flavio P.
contents Translating the internal representations and computations of models into concepts that humans can understand is a key goal of interpretability. While recent dictionary learning methods such as Sparse Autoencoders (SAEs) provide a promising route to discover human-interpretable features, they often only recover token-specific, noisy, or highly local concepts. We argue that this limitation stems from neglecting the temporal structure of language, where semantic content typically evolves smoothly over sequences. Building on this insight, we introduce Temporal Sparse Autoencoders (T-SAEs), which incorporate a novel contrastive loss encouraging consistent activations of high-level features over adjacent tokens. This simple yet powerful modification enables SAEs to disentangle semantic from syntactic features in a self-supervised manner. Across multiple datasets and models, T-SAEs recover smoother, more coherent semantic concepts without sacrificing reconstruction quality. Strikingly, they exhibit clear semantic structure despite being trained without explicit semantic signal, offering a new pathway for unsupervised interpretability in language models.
format Preprint
id arxiv_https___arxiv_org_abs_2511_05541
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Temporal Sparse Autoencoders: Leveraging the Sequential Nature of Language for Interpretability
Bhalla, Usha
Oesterling, Alex
Verdun, Claudio Mayrink
Lakkaraju, Himabindu
Calmon, Flavio P.
Computation and Language
Artificial Intelligence
Machine Learning
Translating the internal representations and computations of models into concepts that humans can understand is a key goal of interpretability. While recent dictionary learning methods such as Sparse Autoencoders (SAEs) provide a promising route to discover human-interpretable features, they often only recover token-specific, noisy, or highly local concepts. We argue that this limitation stems from neglecting the temporal structure of language, where semantic content typically evolves smoothly over sequences. Building on this insight, we introduce Temporal Sparse Autoencoders (T-SAEs), which incorporate a novel contrastive loss encouraging consistent activations of high-level features over adjacent tokens. This simple yet powerful modification enables SAEs to disentangle semantic from syntactic features in a self-supervised manner. Across multiple datasets and models, T-SAEs recover smoother, more coherent semantic concepts without sacrificing reconstruction quality. Strikingly, they exhibit clear semantic structure despite being trained without explicit semantic signal, offering a new pathway for unsupervised interpretability in language models.
title Temporal Sparse Autoencoders: Leveraging the Sequential Nature of Language for Interpretability
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2511.05541