Text Representation Distillation via Information Bottleneck Principle

Yanzhao Zhang, Dingkun Long, Zehan Li, Pengjun Xie


Abstract
Pre-trained language models (PLMs) have recently shown great success in text representation field. However, the high computational cost and high-dimensional representation of PLMs pose significant challenges for practical applications. To make models more accessible, an effective method is to distill large models into smaller representation models. In order to relieve the issue of performance degradation after distillation, we propose a novel Knowledge Distillation method called IBKD. This approach is motivated by the Information Bottleneck principle and aims to maximize the mutual information between the final representation of the teacher and student model, while simultaneously reducing the mutual information between the student model’s representation and the input data. This enables the student model to preserve important learned information while avoiding unnecessary information, thus reducing the risk of over-fitting. Empirical studies on two main downstream applications of text representation (Semantic Textual Similarity and Dense Retrieval tasks) demonstrate the effectiveness of our proposed approach.
Anthology ID:
2023.emnlp-main.888
Volume:
Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing
Month:
December
Year:
2023
Address:
Singapore
Editors:
Houda Bouamor, Juan Pino, Kalika Bali
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
14372–14383
Language:
URL:
https://aclanthology.org/2023.emnlp-main.888
DOI:
10.18653/v1/2023.emnlp-main.888
Bibkey:
Cite (ACL):
Yanzhao Zhang, Dingkun Long, Zehan Li, and Pengjun Xie. 2023. Text Representation Distillation via Information Bottleneck Principle. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, pages 14372–14383, Singapore. Association for Computational Linguistics.
Cite (Informal):
Text Representation Distillation via Information Bottleneck Principle (Zhang et al., EMNLP 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.emnlp-main.888.pdf
Video:
 https://aclanthology.org/2023.emnlp-main.888.mp4