Bytes Are All You Need: Transformers Operating Directly On File Bytes

Published: 30 Jun 2024, Last Modified: 30 Jun 2024Accepted by TMLREveryoneRevisionsBibTeX
Abstract: Modern deep learning approaches usually utilize modality-specific processing. For example, the most common deep learning approach to image classification involves decoding image file bytes into an RGB tensor which is passed into a neural network. Instead, we investigate modality-independent representation learning by performing classification directly on file bytes, without the need for decoding files at inference time. This enables models to operate on various modalities without any hand-designed, modality-specific processing. Our model, ByteFormer, improves ImageNet Top-1 classification accuracy by $5\%$ (from $72.2\%$ to $77.33\%$) relative to DeIT models of similar size. Compared to Perceiver IO, our model requires absolutely no modality-specific processing at inference time, and uses an order of magnitude fewer parameters at equivalent accuracy on ImageNet. We demonstrate that the same ByteFormer architecture can perform audio classification without modifications or modality-specific preprocessing. We achieve $95.42\%$ classification accuracy on the Speech Commands V2 dataset (comparable to the state-of-the-art accuracy of $98.7\%$). Additionally, we demonstrate that ByteFormer can operate jointly on images and audio, handling joint classification without explicit knowledge of the input modality. We release our code at https://github.com/apple/corenet/tree/main/projects/byteformer.
Submission Length: Regular submission (no more than 12 pages of main content)
Changes Since Last Submission: De-anonymized camera-ready version.
Code: https://github.com/apple/corenet/tree/main/projects/byteformer
Assigned Action Editor: ~Evan_G_Shelhamer1
Submission Number: 2311
Loading