Abstract:In this paper, we contend that a natural objective of representation learning is to compress and transform the distribution of the data, say sets of tokens, towards a low-dimensional Gaussian mixture supported on incoherent subspaces. The goodness of such a representation can be evaluated by a principled measure, called sparse rate reduction, that simultaneously maximizes the intrinsic information gain and extrinsic sparsity of the learned representation. From this perspective, popular deep network architectures, including transformers, can be viewed as realizing iterative schemes to optimize this measure. Particularly, we derive a transformer block from alternating optimization on parts of this objective: the multi-head self-attention operator compresses the representation by implementing an approximate gradient descent step on the coding rate of the features, and the subsequent multi-layer perceptron sparsifies the features. This leads to a family of white-box transformer-like deep network architectures, named CRATE, which are mathematically fully interpretable. We show, by way of a novel connection between denoising and compression, that the inverse to the aforementioned compressive encoding can be realized by the same class of CRATE architectures. Thus, the so-derived white-box architectures are universal to both encoders and decoders. Experiments show that these networks, despite their simplicity, indeed learn to compress and sparsify representations of large-scale real-world image and text datasets, and achieve performance very close to highly engineered transformer-based models: ViT, MAE, DINO, BERT, and GPT2. We believe the proposed computational framework demonstrates great potential in bridging the gap between theory and practice of deep learning, from a unified perspective of data compression. Code is available at: <a class="link-external link-https" href="https://ma-lab-berkeley.github.io/CRATE" rel="external noopener nofollow">this https URL</a> .

West: Word Encoded Sequence Transducers

Stacked What-Where Auto-encoders

Bit Cipher -- A Simple yet Powerful Word Representation System that Integrates Efficiently with Language Models

VcLLM: Video Codecs are Secretly Tensor Codecs

Word Equations: Inherently Interpretable Sparse Word Embeddingsthrough Sparse Coding

Word-Level Representation From Bytes For Language Modeling

Multi-word Tokenization for Sequence Compression

Joint Goal for Word Embedding Compression Based on Word Frequency

Extending Context Window of Large Language Models via Semantic Compression

Memory-Augmenting Decoder-Only Language Models through Encoders (Student Abstract)

Word Shape Matters: Robust Machine Translation with Visual Embedding

Tiny Transformers Excel at Sentence Compression

Shifted Chunk Encoder for Transformer Based Streaming End-to-End ASR

Subword and Crossword Units for CTC Acoustic Models

Compressing Neural Language Models by Sparse Word Representations

Multidimensional Byte Pair Encoding: Shortened Sequences for Improved Visual Data Generation

White-Box Transformers via Sparse Rate Reduction: Compression Is All There Is?

Learn Your Tokens: Word-Pooled Tokenization for Language Modeling

Green CWS: Extreme Distillation and Efficient Decode Method Towards Industrial Application

Extreme Model Compression for On-device Natural Language Understanding

Fast Vocabulary Transfer for Language Model Compression