Saved in:
Bibliographic Details
Main Authors: Xu, Yige, Guo, Xu, Zeng, Zhiwei, Miao, Chunyan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.04519
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914965898657792
author Xu, Yige
Guo, Xu
Zeng, Zhiwei
Miao, Chunyan
author_facet Xu, Yige
Guo, Xu
Zeng, Zhiwei
Miao, Chunyan
contents Large language models (LLMs) have brought a great breakthrough to the natural language processing (NLP) community, while leading the challenge of handling concurrent customer queries due to their high throughput demands. Data multiplexing addresses this by merging multiple inputs into a single composite input, allowing more efficient inference through a shared forward pass. However, as distinguishing individuals from a composite input is challenging, conventional methods typically require training the entire backbone, yet still suffer from performance degradation. In this paper, we introduce RevMUX, a parameter-efficient data multiplexing framework that incorporates a reversible design in the multiplexer, which can be reused by the demultiplexer to perform reverse operations and restore individual samples for classification. Extensive experiments on four datasets and three types of LLM backbones demonstrate the effectiveness of RevMUX for enhancing LLM inference efficiency while retaining a satisfactory classification performance.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04519
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RevMUX: Data Multiplexing with Reversible Adapters for Efficient LLM Batch Inference
Xu, Yige
Guo, Xu
Zeng, Zhiwei
Miao, Chunyan
Computation and Language
Large language models (LLMs) have brought a great breakthrough to the natural language processing (NLP) community, while leading the challenge of handling concurrent customer queries due to their high throughput demands. Data multiplexing addresses this by merging multiple inputs into a single composite input, allowing more efficient inference through a shared forward pass. However, as distinguishing individuals from a composite input is challenging, conventional methods typically require training the entire backbone, yet still suffer from performance degradation. In this paper, we introduce RevMUX, a parameter-efficient data multiplexing framework that incorporates a reversible design in the multiplexer, which can be reused by the demultiplexer to perform reverse operations and restore individual samples for classification. Extensive experiments on four datasets and three types of LLM backbones demonstrate the effectiveness of RevMUX for enhancing LLM inference efficiency while retaining a satisfactory classification performance.
title RevMUX: Data Multiplexing with Reversible Adapters for Efficient LLM Batch Inference
topic Computation and Language
url https://arxiv.org/abs/2410.04519