TexShape: Information Theoretic Sentence Embedding for Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kale, Kaan, Esfahanizadeh, Homa, Elias, Noel, Baser, Oguzhan, Medard, Muriel, Vishwanath, Sriram
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914792657125376
author Kale, Kaan
Esfahanizadeh, Homa
Elias, Noel
Baser, Oguzhan
Medard, Muriel
Vishwanath, Sriram
author_facet Kale, Kaan
Esfahanizadeh, Homa
Elias, Noel
Baser, Oguzhan
Medard, Muriel
Vishwanath, Sriram
contents With the exponential growth in data volume and the emergence of data-intensive applications, particularly in the field of machine learning, concerns related to resource utilization, privacy, and fairness have become paramount. This paper focuses on the textual domain of data and addresses challenges regarding encoding sentences to their optimized representations through the lens of information-theory. In particular, we use empirical estimates of mutual information, using the Donsker-Varadhan definition of Kullback-Leibler divergence. Our approach leverages this estimation to train an information-theoretic sentence embedding, called TexShape, for (task-based) data compression or for filtering out sensitive information, enhancing privacy and fairness. In this study, we employ a benchmark language model for initial text representation, complemented by neural networks for information-theoretic compression and mutual information estimations. Our experiments demonstrate significant advancements in preserving maximal targeted information and minimal sensitive information over adverse compression ratios, in terms of predictive accuracy of downstream models that are trained using the compressed data.
format Preprint
id arxiv_https___arxiv_org_abs_2402_05132
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TexShape: Information Theoretic Sentence Embedding for Language Models
Kale, Kaan
Esfahanizadeh, Homa
Elias, Noel
Baser, Oguzhan
Medard, Muriel
Vishwanath, Sriram
Computation and Language
Information Theory
With the exponential growth in data volume and the emergence of data-intensive applications, particularly in the field of machine learning, concerns related to resource utilization, privacy, and fairness have become paramount. This paper focuses on the textual domain of data and addresses challenges regarding encoding sentences to their optimized representations through the lens of information-theory. In particular, we use empirical estimates of mutual information, using the Donsker-Varadhan definition of Kullback-Leibler divergence. Our approach leverages this estimation to train an information-theoretic sentence embedding, called TexShape, for (task-based) data compression or for filtering out sensitive information, enhancing privacy and fairness. In this study, we employ a benchmark language model for initial text representation, complemented by neural networks for information-theoretic compression and mutual information estimations. Our experiments demonstrate significant advancements in preserving maximal targeted information and minimal sensitive information over adverse compression ratios, in terms of predictive accuracy of downstream models that are trained using the compressed data.
title TexShape: Information Theoretic Sentence Embedding for Language Models
topic Computation and Language
Information Theory
url https://arxiv.org/abs/2402.05132