TopoLM: brain-like spatio-functional organization in a topographic language model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rathi, Neil, Mehrer, Johannes, AlKhamissi, Badr, Binhuraib, Taha, Blauch, Nicholas M., Schrimpf, Martin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916737647116288
author Rathi, Neil
Mehrer, Johannes
AlKhamissi, Badr
Binhuraib, Taha
Blauch, Nicholas M.
Schrimpf, Martin
author_facet Rathi, Neil
Mehrer, Johannes
AlKhamissi, Badr
Binhuraib, Taha
Blauch, Nicholas M.
Schrimpf, Martin
contents Neurons in the brain are spatially organized such that neighbors on tissue often exhibit similar response profiles. In the human language system, experimental studies have observed clusters for syntactic and semantic categories, but the mechanisms underlying this functional organization remain unclear. Here, building on work from the vision literature, we develop TopoLM, a transformer language model with an explicit two-dimensional spatial representation of model units. By combining a next-token prediction objective with a spatial smoothness loss, representations in this model assemble into clusters that correspond to semantically interpretable groupings of text and closely match the functional organization in the brain's language system. TopoLM successfully predicts the emergence of the spatio-functional organization of a cortical language system as well as the organization of functional clusters selective for fine-grained linguistic features empirically observed in human cortex. Our results suggest that the functional organization of the human language system is driven by a unified spatial objective, and provide a functionally and spatially aligned model of language processing in the brain.
format Preprint
id arxiv_https___arxiv_org_abs_2410_11516
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TopoLM: brain-like spatio-functional organization in a topographic language model
Rathi, Neil
Mehrer, Johannes
AlKhamissi, Badr
Binhuraib, Taha
Blauch, Nicholas M.
Schrimpf, Martin
Computation and Language
Neurons in the brain are spatially organized such that neighbors on tissue often exhibit similar response profiles. In the human language system, experimental studies have observed clusters for syntactic and semantic categories, but the mechanisms underlying this functional organization remain unclear. Here, building on work from the vision literature, we develop TopoLM, a transformer language model with an explicit two-dimensional spatial representation of model units. By combining a next-token prediction objective with a spatial smoothness loss, representations in this model assemble into clusters that correspond to semantically interpretable groupings of text and closely match the functional organization in the brain's language system. TopoLM successfully predicts the emergence of the spatio-functional organization of a cortical language system as well as the organization of functional clusters selective for fine-grained linguistic features empirically observed in human cortex. Our results suggest that the functional organization of the human language system is driven by a unified spatial objective, and provide a functionally and spatially aligned model of language processing in the brain.
title TopoLM: brain-like spatio-functional organization in a topographic language model
topic Computation and Language
url https://arxiv.org/abs/2410.11516