atlas

Research map / Architectures and efficiency / Attention and transformers

Modeling Language Tokens as Functionals of Semantic Fields

ICML 2024 · Architectures and efficiency › Attention and transformers · approach: Efficient language transformers

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LasF, a Transformer-free nonlinear module that models language tokens as functionals over semantic fields, used both to replace final layers of pretrained LMs and as a flat from-scratch language model.

Problem

Transformers are parameter-heavy and deep, while existing Transformer-free models (SSMs) rely on linear token relations and lack a neuro-biological connection to how the brain represents semantic relations.

Approach

Represent tokens via neuronal states in multiple semantic fields and model interactions between fields as a functional (field-level metric matrix weighting neuron-level relational matrices), yielding a nonlinear module that replaces final Transformer layers or whole stacks, enabling flat (shallow) models.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace the output layer with LasFALBERT-base · SQuAD2.0F1: 79.15 -> 80.33Helped
Replace the output layer with LasFRoBERTa-base · SQuAD2.0F1: 82.32 -> 83.35Helped
Replace the output layer with LasFDeBERTa-base · SQuAD2.0F1: 84.50 -> 85.54Helped
Replace the output layer with LasFLLaMA-7B · CSQAaccuracy: 57.8 -> 62.2Helped
Replace the output layer with LasFLLaMA-13B · PIQAaccuracy: 80.1 -> 82.1Helped
Replace the output layer with LasFLLaMA-13B · SIQAaccuracy: 50.4 -> 51.8Helped
Use L1-norm instead of the default neuron-level metric µ (ablation)2-layer LasF · WikiText-103perplexity: 17.2 -> 17.6Hurt
Use L1-norm instead of the default neuron-level metric µ (ablation)2-layer LasF · PennTreebankperplexity: 49.2 -> 49.5Hurt
Use only L1-norm in the field-level metric set µk (ablation)2-layer LasF · WikiText-103perplexity: 17.2 -> 18.3Hurt
Use only L1-norm in the field-level metric set µk (ablation)2-layer LasF · PennTreebankperplexity: 49.2 -> 52.2Hurt
Use truncated length-related projections D and DS (ablation)2-layer LasF · WikiText-103perplexity: 17.2 -> 17.6Hurt
Use truncated length-related projections D and DS (ablation)2-layer LasF · PennTreebankperplexity: 49.2 -> 49.3No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: SQuAD2.0 CommonsenseQA PIQA SIQA WikiText-103 Penn Treebank WikiQA

Benchmarks: SQuAD2.0 CommonsenseQA PIQA SIQA WikiText-103 Penn Treebank

Metrics: EM F1 accuracy perplexity parameter count training and inference time

Baselines: RoBERTa-large ALBERT-xxlarge UnifiedQA ALBERT+HeadHunter QA-GNN S4 H3 Hyena Transformer GPT-2 small Reformer

Builds on: state-space models Transformer Pei & Wang (2023)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy