Identifying Linear Relational Concepts in Large Language Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Chanin, David, Hunter, Anthony, Camburu, Oana-Maria
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913291572346880
author Chanin, David
Hunter, Anthony
Camburu, Oana-Maria
author_facet Chanin, David
Hunter, Anthony
Camburu, Oana-Maria
contents Transformer language models (LMs) have been shown to represent concepts as directions in the latent space of hidden activations. However, for any human-interpretable concept, how can we find its direction in the latent space? We present a technique called linear relational concepts (LRC) for finding concept directions corresponding to human-interpretable concepts by first modeling the relation between subject and object as a linear relational embedding (LRE). We find that inverting the LRE and using earlier object layers results in a powerful technique for finding concept directions that outperforms standard black-box probing classifiers. We evaluate LRCs on their performance as concept classifiers as well as their ability to causally change model output.
format Preprint
id arxiv_https___arxiv_org_abs_2311_08968
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Identifying Linear Relational Concepts in Large Language Models
Chanin, David
Hunter, Anthony
Camburu, Oana-Maria
Computation and Language
Artificial Intelligence
Transformer language models (LMs) have been shown to represent concepts as directions in the latent space of hidden activations. However, for any human-interpretable concept, how can we find its direction in the latent space? We present a technique called linear relational concepts (LRC) for finding concept directions corresponding to human-interpretable concepts by first modeling the relation between subject and object as a linear relational embedding (LRE). We find that inverting the LRE and using earlier object layers results in a powerful technique for finding concept directions that outperforms standard black-box probing classifiers. We evaluate LRCs on their performance as concept classifiers as well as their ability to causally change model output.
title Identifying Linear Relational Concepts in Large Language Models
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2311.08968