atlas

Research map / Large language models / Instruction and fine-tuning

Efficient Multi-task LLM Quantization and Serving for Multiple LoRA Adapters

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Low-rank adaptation variants · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LoRA-Inlaid, a multi-LoRA serving system with MLGPTQ, a joint quantization method that yields one shared quantized base model for multiple LoRA adapters, plus dynamic task addition and output-length-guided grouped scheduling.

Problem

Quantized base models (GPTQ/AWQ) are task-specific when calibrated with LoRA adapters, so they cannot be shared across tasks in multi-LoRA serving; existing systems also lack dynamic task addition and ignore workload differences across tasks during scheduling.

Approach

MLGPTQ quantizes the base model jointly for all tasks by loading each task's LoRA during the forward pass and max-aggregating per-task inverse Hessian diagonals to choose weight updates; incremental quantization caches the aggregated Hessian so new tasks can be added without interrupting service; scheduling uses output-length prediction with SRTF and task grouping to limit the number of LoRA adapt

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MLGPTQ instead of GPTQ with mixed calibration data for shareable 4-bit quantizationLLaMA2 multi-task quantization · GSM8kAccuracy (%): 31.51 (0) -> 31.62 (0)Helped
Use 4-bit MLGPTQ quantization rather than no quantizationLLaMA2 multi-task quantization · GSM8kAccuracy (%): 32.14 (0) -> 31.62 (0)Hurt
Use MLGPTQ rather than S-LoRALLaMA2-7B serving · RTX 4090, 2 tasks, request rate 5Throughput (reqs/s): 2.93 -> 3.89Helped
Use MLGPTQ rather than S-LoRALLaMA2-7B serving · RTX 4090, 1000 tasks, request rate 5Throughput (reqs/s): 2.78 -> 3.42Helped
Add task grouping to multi-task scheduling (ablation)LoRA-Inlaid · 100 tasksSLO Attainment: +1.16× on averageHelped
Add output length prediction to multi-task scheduling (ablation)LoRA-Inlaid · 100 tasksSLO Attainment: +1.23× on averageHelped
Add prediction-based SRTF scheduling (ablation)LoRA-Inlaid · 100 tasksSLO Attainment: +2.27× on averageHelped
Add multi-task quantization (ablation)LoRA-Inlaid serving LLaMA2-7B · Serving workloadsSLO Attainment: 39%Helped
Add multi-task quantization (ablation)LoRA-Inlaid serving LLaMA2-13B · Serving workloadsmemory feasibilityHelped
Include the target task's calibration set and LoRA adapter during quantization (ablation)MLGPTQ · Machine translation tasksS_BLEUHelped
Add incremental quantization while serving continuesLoRA-Inlaid · LLaMA2-7B, online task additionThroughput: 10%-13% of degradationHurt

Surprising findings

Limitations

Experimental setup

Datasets: OPUS (six translation pairs) XLSum QTSumm tiny-codes GSM8K medical_meadow_mmmlu malicious-600k

Benchmarks: OPUS XLSum QTSumm tiny-codes GSM8K medical_meadow_mmmlu malicious-600k

Metrics: SacreBLEU ROUGE-1 accuracy throughput average latency job completion time SLO attainment

Baselines: RTN GPTQ (tweaked) AWQ (tweaked) vLLM S-LoRA Skip-join MLFQ

Builds on: GPTQ LoRA S-LoRA Punica FastServe

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy