TopoNets: High Performing Vision and Language Models with Brain-Like Topography

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Deb, Mayukh, Deb, Mainak, Murty, N. Apurva Ratan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916586345988096
author Deb, Mayukh
Deb, Mainak
Murty, N. Apurva Ratan
author_facet Deb, Mayukh
Deb, Mainak
Murty, N. Apurva Ratan
contents Neurons in the brain are organized such that nearby cells tend to share similar functions. AI models lack this organization, and past efforts to introduce topography have often led to trade-offs between topography and task performance. In this work, we present TopoLoss, a new loss function that promotes spatially organized topographic representations in AI models without significantly sacrificing task performance. TopoLoss is highly adaptable and can be seamlessly integrated into the training of leading model architectures. We validate our method on both vision (ResNet-18, ResNet-50, ViT) and language models (GPT-Neo-125M, NanoGPT), collectively TopoNets. TopoNets are the highest-performing supervised topographic models to date, exhibiting brain-like properties such as localized feature processing, lower dimensionality, and increased efficiency. TopoNets also predict responses in the brain and replicate the key topographic signatures observed in the brain's visual and language cortices. Together, this work establishes a robust and generalizable framework for integrating topography into leading model architectures, advancing the development of high-performing models that more closely emulate the computational strategies of the human brain.
format Preprint
id arxiv_https___arxiv_org_abs_2501_16396
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TopoNets: High Performing Vision and Language Models with Brain-Like Topography
Deb, Mayukh
Deb, Mainak
Murty, N. Apurva Ratan
Machine Learning
Neural and Evolutionary Computing
Neurons and Cognition
Neurons in the brain are organized such that nearby cells tend to share similar functions. AI models lack this organization, and past efforts to introduce topography have often led to trade-offs between topography and task performance. In this work, we present TopoLoss, a new loss function that promotes spatially organized topographic representations in AI models without significantly sacrificing task performance. TopoLoss is highly adaptable and can be seamlessly integrated into the training of leading model architectures. We validate our method on both vision (ResNet-18, ResNet-50, ViT) and language models (GPT-Neo-125M, NanoGPT), collectively TopoNets. TopoNets are the highest-performing supervised topographic models to date, exhibiting brain-like properties such as localized feature processing, lower dimensionality, and increased efficiency. TopoNets also predict responses in the brain and replicate the key topographic signatures observed in the brain's visual and language cortices. Together, this work establishes a robust and generalizable framework for integrating topography into leading model architectures, advancing the development of high-performing models that more closely emulate the computational strategies of the human brain.
title TopoNets: High Performing Vision and Language Models with Brain-Like Topography
topic Machine Learning
Neural and Evolutionary Computing
Neurons and Cognition
url https://arxiv.org/abs/2501.16396