FocusLLM: Precise Understanding of Long Context by Dynamic Condensing

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Zhenyu, Zhang, Yike, Pan, Tengyu, Sun, Yutao, Duan, Zhichao, Fang, Junjie, Han, Rong, Wang, Zixuan, Wang, Jianyong
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912166986121216
author Li, Zhenyu
Zhang, Yike
Pan, Tengyu
Sun, Yutao
Duan, Zhichao
Fang, Junjie
Han, Rong
Wang, Zixuan
Wang, Jianyong
author_facet Li, Zhenyu
Zhang, Yike
Pan, Tengyu
Sun, Yutao
Duan, Zhichao
Fang, Junjie
Han, Rong
Wang, Zixuan
Wang, Jianyong
contents Empowering LLMs with the ability to precisely understand long contexts is crucial for many downstream applications. However, handling long contexts with conventional transformer architecture requires substantial training and inference resources. Existing context condensing methods cannot accurately understand the full context, as there is a considerable amount of information loss in the condensing process. To address these issues, we present FocusLLM, a framework designed to extend the fixed context length of any decoder-only LLM, allowing the model to focus on relevant information from very long sequences. FocusLLM first divides long text input into chunks based on the model's original context length. It then employs the dynamic condensing process to distill crucial information from each chunk. Ultimately, through the novel parallel decoding mechanism, FocusLLM can integrate the extracted information into its local context. FocusLLM stands out for great training efficiency and versatility: trained with an 8K input length and with much less training cost than previous methods, FocusLLM exhibits superior performance across downstream tasks and maintains strong language modeling ability when handling extensive long texts, even up to 400K tokens. Our code is available at https://github.com/leezythu/FocusLLM.
format Preprint
id arxiv_https___arxiv_org_abs_2408_11745
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FocusLLM: Precise Understanding of Long Context by Dynamic Condensing
Li, Zhenyu
Zhang, Yike
Pan, Tengyu
Sun, Yutao
Duan, Zhichao
Fang, Junjie
Han, Rong
Wang, Zixuan
Wang, Jianyong
Computation and Language
Artificial Intelligence
Empowering LLMs with the ability to precisely understand long contexts is crucial for many downstream applications. However, handling long contexts with conventional transformer architecture requires substantial training and inference resources. Existing context condensing methods cannot accurately understand the full context, as there is a considerable amount of information loss in the condensing process. To address these issues, we present FocusLLM, a framework designed to extend the fixed context length of any decoder-only LLM, allowing the model to focus on relevant information from very long sequences. FocusLLM first divides long text input into chunks based on the model's original context length. It then employs the dynamic condensing process to distill crucial information from each chunk. Ultimately, through the novel parallel decoding mechanism, FocusLLM can integrate the extracted information into its local context. FocusLLM stands out for great training efficiency and versatility: trained with an 8K input length and with much less training cost than previous methods, FocusLLM exhibits superior performance across downstream tasks and maintains strong language modeling ability when handling extensive long texts, even up to 400K tokens. Our code is available at https://github.com/leezythu/FocusLLM.
title FocusLLM: Precise Understanding of Long Context by Dynamic Condensing
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2408.11745