User Embeddings are Superpositions of Interpretable Behavioral Modes
Jonathan Kozaczuk, Zheng Dong, Noyan Evirgen, Keivan Majidi, Eddie K. H. Ng, Chris Porter, et al.
arXiv:2611.08487v1Modern recommendation and personalization systems are built on user embeddings: dense vectors that summarize a person’s past behavior and preferences. These embeddings are extremely useful because they compress a lot of signal into a form that downstream models can work with efficiently. But that same compression is also the problem. A dense vector is hard to inspect, hard to explain, and hard to query in human terms. If a system recommends something surprising, it is not obvious what behavior in the embedding led to that decision. And if an analyst wants to find a group of users who care about, say, dog products or binge-watching, the embedding space usually cannot be searched directly with natural language. This paper argues that the opacity is not accidental. Instead, user embeddings can be viewed as superpositions of interpretable behavioral modes. That idea comes from mechanistic interpretability, where researchers have found that neural networks often represent many concepts in overlapping, distributed directions rather than in neat one-feature-per-neuron form. The authors suggest that user embeddings behave similarly: a single user vector may mix together several meaningful modes, such as being a casual dog owner versus a devoted dog owner, or a commuter versus an occasional rider. If that is true, then sparse dictionary learning methods, especially sparse autoencoders, should be able to recover those hidden modes after the fact, without changing the original recommendation model. To understand the setting, it helps to think about how user embeddings are created. In the simplest case, a user’s history is a set or sequence of interactions, and an embedding function turns that history into a vector. Sometimes that function is classical, like a matrix factorization or count-based representation. Sometimes it is a deep encoder, such as a transformer over the interaction sequence. Either way, the embedding is meant to preserve useful behavioral information while reducing dimensionality. The paper’s key intuition is that if users are generated from a sparse mixture of latent behavioral modes, then averaged or pooled embeddings should inherit that mixture structure. In other words, the embedding of a user who is 70 percent one kind of shopper and 30 percent another kind of shopper should lie near a linear combination of the corresponding behavioral directions. The paper starts from a simple generative model. Imagine a population of users, each of whom draws interactions from a small subset of shared modes. Those modes are interpretable user types with their own distributions over observable behavior. The model is mixed-membership rather than one-user-one-cluster: a user can belong to multiple modes at once, with different weights. That matters because real behavior is rarely categorical. People are not just “a sports fan” or “not a sports fan”; they may be mostly interested in one category but also have secondary interests. The authors note that when the observations are discrete, this reduces to the same kind of structure used in topic models like LDA, except that here the “words” are behavioral events and the “topics” are behavioral modes. From there, the paper shows why embeddings should exhibit superposition. If the user embedding is formed by averaging per-interaction features, then the expected embedding becomes a sparse linear combination of mode centroids. Those centroids do not have to be orthogonal. In fact, they often should not be. Two modes may overlap in behavior, and even unrelated modes must share representational space when the embedding dimension is smaller than the number of latent modes. That overlap is exactly what superposition means here: multiple meaningful directions are packed into the same space. The authors extend the same logic to deep encoders using the superposition hypothesis. If a transformer’s internal representations already encode many features in overlapping directions, then pooling those representations over a user’s history should produce a user vector that is again a mixture of mode-level directions. In the cases they study, they use mean pooling over final-layer positions, so this argument applies directly. Given that structure, the method is straightforward in spirit. Train or take a pre-existing user embedding system, freeze it, and then decompose the resulting embeddings with sparse dictionary learning. The paper focuses on sparse autoencoders, which learn a dictionary of latent directions and a sparse code for each user. Each latent is supposed to capture one behavioral mode or a small composite of related modes. Because the decomposition is post-hoc, it does not require access to the upstream model’s internals or any retraining of the recommender. That makes it model-agnostic and practical for systems already in production. A nice feature of this approach is that the learned latents can be interpreted after the fact. The authors describe automatically naming and querying the recovered modes in natural language. In effect, once the sparse dictionary is learned, one can ask what users activate a given latent, inspect those users’ behaviors, and attach a human-readable label to the mode. That also enables search over the embedding space using text-like queries, even though the original embeddings were never trained on text. This is important because it turns a purely numeric representation into something an analyst can interrogate. The experiments are designed to test both recovery and interpretability. First, the authors validate mode recovery on synthetic data where the ground-truth modes are known. They generate user behavior from the mixed-membership model and then embed it in two ways: with simple mean-pooling and with a trained transformer encoder. Because the true latent modes are known, they can directly measure how well the sparse decomposition recovers the underlying mixture weights. This is a stronger test than simply checking whether the learned factors look plausible. It lets them ask whether the method actually finds the right behavioral structure. On this synthetic setup, sparse autoencoders recover the modes better than classical sparse baselines. The paper specifically compares against baselines including a soft-frozen-decoder variant discussed in prior criticism of SAEs. The important point is that the superiority is not universal in every possible regime, but in their moderate-overcompleteness setting the equivalence claimed by some random-baseline arguments does not hold. So the result here is that, at least for these user embeddings and this level of compression, the trained SAE is doing more than a trivial dictionary fit. The second set of experiments uses two real-world datasets, with both classical and transformer-based user embeddings. Since real data do not come with ground-truth behavioral modes, the authors use external interpretability tests. One of the main tests is zero-shot natural-language audience retrieval. The idea is to use the recovered mode labels to search for audiences without any seed users and without training on text-aligned supervision. This is a meaningful benchmark because it checks whether the decomposition produces modes that are not only mathematically sparse, but also usable by a human. If a latent can be named something like “devoted dog owners” and then used to retrieve users that fit that description, that is much stronger evidence of interpretability than just visualizing a latent vector. The paper reports that the recovered features are more behaviorally distinct than K-means-style encodings, and that retrieval performance matches or exceeds classical sparse and text-embedding baselines. The takeaway is not that the method magically solves all interpretability problems, but that the sparse decomposition gives a useful handle on user behavior that older clustering or embedding-based approaches do not provide. K-means gives hard clusters, which can be too coarse for mixed behavior. Text embeddings can support semantic search, but they require text in the first place. Here, the method works directly on behavioral embeddings and still supports natural-language querying. What do these results actually show? First, they support the central hypothesis that user embeddings do contain meaningful behavioral modes in superposition. Second, they show that sparse dictionary learning is a viable way to recover those modes, at least in the settings tested. Third, they show that the recovered modes are not just mathematically convenient; they can be named and used for downstream tasks like audience retrieval. And because the whole pipeline is post-hoc, it can be applied to existing systems without changing the model that produced the embedding in the first place. There are also some important limitations and caveats. The authors are careful to note that their theoretical argument is exact only for embedding functions that average per-observation features. For nonlinear sequence encoders, the claim relies on the superposition hypothesis and is therefore approximate rather than guaranteed. They also point out that when attention or other contextual effects make the mapping from behavior to embedding less linear, the recovered latents may correspond to finer or composite structure rather than one clean ground-truth mode per latent. That is not necessarily a failure, but it means interpretation depends on the encoder and the granularity it chooses to represent. Another limitation is that the paper’s strongest quantitative validation comes from synthetic data, where ground truth is available. That is appropriate and important, but real-world behavior is messier. In practice, modes may overlap heavily, labels may be ambiguous, and the “right” granularity of a behavioral mode may depend on the application. The authors also mention that single-position readouts like last-token or CLS embeddings are not directly covered by their linear pooling argument, so the analysis is strongest for mean-pooled representations. Future work would need to test how far the framework extends to those settings. Still, the broader message is compelling. User embeddings are not just opaque blobs; they may be sparse mixtures of interpretable behavioral directions. If that is true, then the same tools that have been useful for interpreting language models can help open up recommendation and personalization systems as well. This paper makes that case with a clear theoretical story, controlled synthetic validation, and real-world evidence that the recovered modes can support human-facing tasks like semantic search and audience characterization. In short, it offers a practical route from dense user vectors to understandable behavioral structure, without requiring the original system to be redesigned.
Also spotted that day
Previous daily papers