paper-with-me

Papers

Layer by Layer: Uncovering Hidden Representations in Language Models

2025-02-04 · Oscar Skean, Md Rifat Arefin, Dan Zhao, Niket Patel, Jalal Naghiyev, Yann Lecun, Ravid Shwartz-Ziv

From extracting features to generating text, the outputs of large language models (LLMs) typically rely on their final layers, following the conventional wisdom that earlier layers capture only low-level cues. However, our analysis shows that intermediate layers can encode even richer representations, often improving performance on a wide range of downstream tasks. To explain and quantify these hidden-layer properties, we propose a unified framework of representation quality metrics based on information theory, geometry, and invariance to input perturbations. Our framework highlights how each model layer balances information compression and signal preservation, revealing why mid-depth embeddings can exceed the last layer's performance. Through extensive experiments on 32 text-embedding tasks and comparisons across model architectures (transformers, state-space models) and domains (language, vision), we demonstrate that intermediate layers consistently provide stronger features. These findings challenge the standard focus on final-layer embeddings and open new directions for model analysis and optimization, including strategic use of mid-layer representations for more robust and accurate AI systems.

📄 PDF Abstract BibTeX arXiv:2502.02013

Code (0)

등록된 구현이 없습니다.

Tasks

State Space Models

Methods 이 논문이 사용한 방법론

Focus 설명 없음

Similar Papers 제목 키워드 기반

Structure Amplification on Multi-layer Stochastic Block Models

2021-07-31 · Xiaodong Xin, Kun He, Jialu Bao, Bart Selman 외

Much of the complexity of social, biological, and engineered systems arises from a network of complex interactions connecting many basic components. Network analysis tools have been successful at uncovering latent struct…

Stochastic Block Model

Jump to Conclusions: Short-Cutting Transformers With Linear Transformations

2023-03-16 · Alexander Yom Din, Taelin Karidi, Leshem Choshen, Mor Geva

Transformer-based language models create hidden representations of their inputs at every layer, but only use final-layer representations for prediction. This obscures the internal decision-making process of the model and…

Decision MakingLanguage ModelingLanguage Modelling

HyPe: Better Pre-trained Language Model Fine-tuning with Hidden Representation Perturbation

2022-12-17 · Hongyi Yuan, Zheng Yuan, Chuanqi Tan, Fei Huang 외

Language models with the Transformers structure have shown great performance in natural language processing. However, there still poses problems when fine-tuning pre-trained language models on downstream tasks, such as o…

Language ModelingLanguage ModellingNatural Language Inference

DySem: Uncovering Dynamic Semantic Components of Large Language Models for Calculating Semantic Textual Similarity

2026-05-28 · Kaijie Zheng, Weiqin Wang, Yile Wang, Hui Huang arxiv

Calculating semantic textual similarity is a foundational task in natural language processing. Current large language models (LLMs) based methods typically rely on extracting last-layer hidden states with fixed dimension…

Semantic Textual SimilaritySemantic SimilarityGeneral Knowledge

Hidden Heroes and Gradient Bloats: Layer-Wise Redundancy Inverts Attribution in Transformers

2026-02-01 · Donald Ye arxiv

Gradient-based attribution is the workhorse of mechanistic interpretability, yet whether it reliably tracks causal importance at the component level remains largely untested. We causally evaluate this assumption across t…