English

BERT's output layer recognizes all hidden layers? Some Intriguing Phenomena and a simple way to boost BERT

Computation and Language 2021-02-16 v2 Machine Learning

Abstract

Although Bidirectional Encoder Representations from Transformers (BERT) have achieved tremendous success in many natural language processing (NLP) tasks, it remains a black box. A variety of previous works have tried to lift the veil of BERT and understand each layer's functionality. In this paper, we found that surprisingly the output layer of BERT can reconstruct the input sentence by directly taking each layer of BERT as input, even though the output layer has never seen the input other than the final hidden layer. This fact remains true across a wide variety of BERT-based models, even when some layers are duplicated. Based on this observation, we propose a quite simple method to boost the performance of BERT. By duplicating some layers in the BERT-based models to make it deeper (no extra training required in this step), they obtain better performance in the downstream tasks after fine-tuning.

Keywords

Cite

@article{arxiv.2001.09309,
  title  = {BERT's output layer recognizes all hidden layers? Some Intriguing Phenomena and a simple way to boost BERT},
  author = {Wei-Tsung Kao and Tsung-Han Wu and Po-Han Chi and Chun-Cheng Hsieh and Hung-Yi Lee},
  journal= {arXiv preprint arXiv:2001.09309},
  year   = {2021}
}

Comments

7 pages, 8 figures, 3 tables

R2 v1 2026-06-23T13:20:33.911Z