Length-Induced Embedding Collapse in PLM-based Models
Text embeddings from PLM-based models enable a wide range of applications, yet their performance often degrades on longer texts.In this paper, we introduce a phenomenon we call Length Collapse, where embeddings of longer texts tend to cluster together.This clustering results in a distributional inconsistency between the embeddings of short and long texts.We further investigate how these differences contribute to the performance decline observed with longer texts across various downstream tasks.Through a rigorous theoretical analysis of the self-attention mechanism, which acts as a low-pass filter in PLM-based models, we demonstrate that as text length increases, the strength of low-pass filtering intensifies, causing embeddings to retain more low-frequency components.As a result, input token features become more similar, leading to clustering and ultimately the collapse of embeddings for longer texts.To address this issue, we propose a simple method, TempScale, which mitigates the Length Collapse phenomenon.By narrowing the gap in low-pass filtering rates between long and short texts, TempScale ensures more consistent embeddings across different text lengths.This approach leads to performance improvements of 0.94% on MTEB and 1.10% on LongEmbed, which focuses specifically on long-context retrieval, providing strong evidence for the validity of our analysis.The source code is available at https://github. com/Yuqi-Zhou/Length_Collapse.
Authors
- Jun Xu (ORCID: https://orcid.org/0000-0001-7170-111X)
- Sunhao Dai (ORCID: https://orcid.org/0009-0002-7549-0860)
- Zhenhua Cao (ORCID: https://orcid.org/0000-0002-5247-8896)
- Xiao Zhang (ORCID: https://orcid.org/0000-0003-0824-9284)
- Yuqi Zhou (ORCID: https://orcid.org/0009-0009-5656-8475)
Institutions
- Beijing Academy of Artificial Intelligence (CN)
- Renmin University of China (CN)
Publication Details
- Published
- 2025-01-01
- DOI
- https://doi.org/10.18653/v1/2025.acl-long.1396
- Primary Topic
- Characterization and Applications of Magnetic Nanoparticles
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Renmin University of China
- Ministry of Education, India