about
ETT: Expanding the Long Context Understanding Capability of LLMs at Test-Time (arxiv.org)
4 points by PaulHoule on Jul 14, 2025 | hide | past | pdf | discuss on HN

In plain words: To read long texts, the model fine-tunes its own weights on the input, split into overlapping chunks, so a short-context model handles more text with constant memory. This stretched context 32 times, from 1k to 32k tokens, raising accuracy by up to 30 percent.

Abstract

Transformer-based Language Models' computation and memory overhead increase quadratically as a function of sequence length. The quadratic cost poses challenges when employing LLMs for processing long sequences. In this work, we introduce \ourmodelacronym~(Extend at Test-Time), method for extending the context length of short context Transformer-based LLMs, with constant memory requirement and linear computation overhead. ETT enable the extension of the context length at test-time by efficient fine-tuning the model's parameters on the input context, chunked into overlapping small subsequences. We evaluate ETT on LongBench by extending the context length of GPT-Large and Phi-2 up to 32 times, increasing from 1k to 32k tokens. This results in up to a 30 percent improvement in the model's accuracy. We also study how context can be stored in LLM's weights effectively and efficiently. Through a detailed ablation study, we examine which Transformer modules are most beneficial to fine-tune at test-time. Interestingly, we find that fine-tuning the second layer of the FFNs is more effective than full fine-tuning, leading to a further improvement in the models' accuracy.

Kiarash Zahirnia, Zahra Golpayegani, Walid Ahmed, Yang Liu
arXiv:2507.06313 · cs.CL · submitted Jul 8, 2025 · updated Oct 21, 2025
abstract · pdf · html

add comment on HN