PaceLLM: Brain-Inspired Large Language Models for Long-Context Understanding

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Kangcong, Ye, Peng, Tu, Chongjun, Zhang, Lin, Song, Chunfeng, Wu, Jiamin, Yang, Tao, Zheng, Qihao, Chen, Tao
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914462132338688
author Li, Kangcong
Ye, Peng
Tu, Chongjun
Zhang, Lin
Song, Chunfeng
Wu, Jiamin
Yang, Tao
Zheng, Qihao
Chen, Tao
author_facet Li, Kangcong
Ye, Peng
Tu, Chongjun
Zhang, Lin
Song, Chunfeng
Wu, Jiamin
Yang, Tao
Zheng, Qihao
Chen, Tao
contents While Large Language Models (LLMs) demonstrate strong performance across domains, their long-context capabilities are limited by transient neural activations causing information decay and unstructured feed-forward network (FFN) weights leading to semantic fragmentation. Inspired by the brain's working memory and cortical modularity, we propose PaceLLM, featuring two innovations: (1) a Persistent Activity (PA) Mechanism that mimics prefrontal cortex (PFC) neurons' persistent firing by introducing an activation-level memory bank to dynamically retrieve, reuse, and update critical FFN states, addressing contextual decay; and (2) Cortical Expert (CE) Clustering that emulates task-adaptive neural specialization to reorganize FFN weights into semantic modules, establishing cross-token dependencies and mitigating fragmentation. Extensive evaluations show that PaceLLM achieves 6% improvement on LongBench's Multi-document QA and 12.5-17.5% performance gains on Infinite-Bench tasks, while extending measurable context length to 200K tokens in Needle-In-A-Haystack (NIAH) tests. This work pioneers brain-inspired LLM optimization and is complementary to other works. Besides, it can be generalized to any model and enhance their long-context performance and interpretability without structural overhauls.
format Preprint
id arxiv_https___arxiv_org_abs_2506_17310
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PaceLLM: Brain-Inspired Large Language Models for Long-Context Understanding
Li, Kangcong
Ye, Peng
Tu, Chongjun
Zhang, Lin
Song, Chunfeng
Wu, Jiamin
Yang, Tao
Zheng, Qihao
Chen, Tao
Neurons and Cognition
Computation and Language
Neural and Evolutionary Computing
While Large Language Models (LLMs) demonstrate strong performance across domains, their long-context capabilities are limited by transient neural activations causing information decay and unstructured feed-forward network (FFN) weights leading to semantic fragmentation. Inspired by the brain's working memory and cortical modularity, we propose PaceLLM, featuring two innovations: (1) a Persistent Activity (PA) Mechanism that mimics prefrontal cortex (PFC) neurons' persistent firing by introducing an activation-level memory bank to dynamically retrieve, reuse, and update critical FFN states, addressing contextual decay; and (2) Cortical Expert (CE) Clustering that emulates task-adaptive neural specialization to reorganize FFN weights into semantic modules, establishing cross-token dependencies and mitigating fragmentation. Extensive evaluations show that PaceLLM achieves 6% improvement on LongBench's Multi-document QA and 12.5-17.5% performance gains on Infinite-Bench tasks, while extending measurable context length to 200K tokens in Needle-In-A-Haystack (NIAH) tests. This work pioneers brain-inspired LLM optimization and is complementary to other works. Besides, it can be generalized to any model and enhance their long-context performance and interpretability without structural overhauls.
title PaceLLM: Brain-Inspired Large Language Models for Long-Context Understanding
topic Neurons and Cognition
Computation and Language
Neural and Evolutionary Computing
url https://arxiv.org/abs/2506.17310