← Search

Goutham Rajendran

8 accepted papers

2024

Do LLMs dream of elephants (when told not to)? Latent concept association and associative memory in transformers

NeurIPS 2024poster

Large Language Models (LLMs) have the capacity to store and recall facts. Through experimentation with open-source models, we observe that this ability to retrieve facts can be easily manipulated by changing contexts, even without altering their factual meanings. These findings highlight that LLMs m…

Cited by 6SourcePDFScholar
2024

From Causal to Concept-Based Representation Learning

NeurIPS 2024poster

To build intelligent machine learning systems, modern representation learning attempts to recover latent generative factors from data, such as in causal representation learning. A key question in this growing field is to provide rigorous conditions under which latent factors can be identified and th…

Cited by 2SourcePDFScholar
2024

On the Origins of Linear Representations in Large Language Models

ICML 2024poster

An array of recent works have argued that high-level semantic concepts are encoded "linearly" in the representation space of large language models. In this work, we study the origins of such linear representations. To that end, we introduce a latent variable model to abstract and formalize the conce…

Cited by 25SourcePDFScholar
2023

Learning Linear Causal Representations from Interventions under General Nonlinear Mixing

NeurIPS 2023oral

We study the problem of learning causal representations from unknown, latent interventions in a general setting, where the latent distribution is Gaussian but the mixing function is completely general. We prove strong identifiability results given unknown single-node interventions, i.e., without hav…

Cited by 71SourcePDFScholar
2022

Identifiability of deep generative models without auxiliary information

NeurIPS 2022accept

We prove identifiability of a broad class of deep latent variable models that (a) have universal approximation capabilities and (b) are the decoders of variational autoencoders that are commonly used in practice. Unlike existing work, our analysis does not require weak supervision, auxiliary informa…

Cited by 63SourcePDFScholar
2022

Sub-exponential time Sum-of-Squares lower bounds for Principal Components Analysis

NeurIPS 2022accept

Principal Components Analysis (PCA) is a dimension-reduction technique widely used in machine learning and statistics. However, due to the dependence of the principal components on all the dimensions, the components are notoriously hard to interpret. Therefore, a variant known as sparse PCA is often…

Cited by 15SourcePDFScholar
2021

Learning latent causal graphs via mixture oracles

NeurIPS 2021poster

We study the problem of reconstructing a causal graphical model from data in the presence of latent variables. The main problem of interest is recovering the causal structure over the latent variables while allowing for general, potentially nonlinear dependencies. In many practical problems, the dep…

2021

Structure learning in polynomial time: Greedy algorithms, Bregman information, and exponential families

NeurIPS 2021poster

Greedy algorithms have long been a workhorse for learning graphical models, and more broadly for learning statistical models with sparse structure. In the context of learning directed acyclic graphs, greedy algorithms are popular despite their worst-case exponential runtime. In practice, however, th…

Cited by 21SourcePDFScholar