Semantic Compression for Word and Sentence Embeddings using Discrete Wavelet Transform

Rana Salama, Abdou Youssef, Mona Diab


Abstract
Wavelet transforms, a powerful mathematical tool, have been widely used in different domains, including Signal and Image processing, to unravel intricate patterns, enhance data representation, and extract meaningful features from data. Tangible results from their application suggest that Wavelet transforms can be applied to NLP capturing a variety of linguistic and semantic properties.In this paper, we empirically leverage the application of Discrete Wavelet Transforms (DWT) to word and sentence embeddings. We aim to showcase the capabilities of DWT in analyzing embedding representations at different levels of resolution and compressing them while maintaining their overall quality.We assess the effectiveness of DWT embeddings on semantic similarity tasks to show how DWT can be used to consolidate important semantic information in an embedding vector. We show the efficacy of the proposed paradigm using different embedding models, including large language models, on downstream tasks. Our results show that DWT can reduce the dimensionality of embeddings by 50-93% with almost no change in performance for semantic similarity tasks, while achieving superior accuracy in most downstream tasks. Our findings pave the way for applying DWT to improve NLP applications.
Anthology ID:
2024.findings-acl.945
Volume:
Findings of the Association for Computational Linguistics ACL 2024
Month:
August
Year:
2024
Address:
Bangkok, Thailand and virtual meeting
Editors:
Lun-Wei Ku, Andre Martins, Vivek Srikumar
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
15963–15977
Language:
URL:
https://aclanthology.org/2024.findings-acl.945
DOI:
Bibkey:
Cite (ACL):
Rana Salama, Abdou Youssef, and Mona Diab. 2024. Semantic Compression for Word and Sentence Embeddings using Discrete Wavelet Transform. In Findings of the Association for Computational Linguistics ACL 2024, pages 15963–15977, Bangkok, Thailand and virtual meeting. Association for Computational Linguistics.
Cite (Informal):
Semantic Compression for Word and Sentence Embeddings using Discrete Wavelet Transform (Salama et al., Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-acl.945.pdf