用极低秩适配器实现高性能大模型服务,兼顾速度与效果
Pin Once, Swap Light: Subspace-Aligned Centroid-Residual Training for Efficient Ultra-LoRA Serving

- 先训练统一领域中心,再让用户用极低秩残差微调
- 仅用秩≤2的残差模块,性能接近高秩模型
- 适合多租户大模型服务场景,显著提升吞吐量
现代多租户低秩适配器(LoRA)服务系统同时承载数十到数百个LoRA适配器。尽管强大,却面临服务效率与任务性能的矛盾:高秩适配器性能好,但占用大量GPU显存且需频繁通过PCIe传输;而超低秩适配器(r ≤ 2)虽节省显存和传输开销,但下游任务性能下降明显。为此,我们提出面向服务效率的分层微调框架SALT。首先,提供方在公开数据上联合训练高容量领域中心,并引入新型对齐正则化,将领域内任务子空间统一为一组基;其次,用户在私有数据上基于冻结中心微调超低秩残差适配器;最后推理时,提供方将中心固定在GPU显存中,按需动态交换用户的残差适配器。在多种规模LLM上,SALT以r ≤ 2的残差模块恢复高秩精度,相比当前最优压缩基线绝对提升最高达18.5%,单适配器内存减少最高16倍。集成至vLLM后,在Llama-3.2-3B上,于PCIe带宽受限下吞吐提升51%,在显存受限下提升28%。
原文摘要 · Abstract (English)
Modern multi-tenant Low-Rank Adapters (LoRAs) serving systems concurrently host tens to hundreds of LoRA adapters. Though powerful, this introduces a critical system dilemma between serving efficiency and task performance: higher-rank adapters generally achieve better downstream task performance, but their GPU VRAM footprint and Host-to-Device PCIe swapping overhead severely constrain scalability. Conversely, ultra-low-rank adapters ($r \le 2$) minimize both VRAM footprint and PCIe transfer overhead, but suffer from downstream task performance degradation. To solve this problem, we propose Subspace-Aligned LoRA Training (SALT), a serving efficiency-aware hierarchical fine-tuning framework. Our solution operates in three phases. First, a provider jointly trains high-capacity domain centroids on public data within the domain using a novel alignment regularizer that coheres in-domain task subspaces into a unified basis. Next, users fine-tune ultra-low-rank task residual adapters on private data atop those frozen centroids. Finally, during inference, the provider pins the centroid in GPU VRAM and dynamically swaps in each user's task residual on demand. Across LLMs of varying scales, SALT recovers high-rank accuracy using $r \le 2$ residuals, achieving up to 18.5% absolute accuracy gains over state-of-the-art compression baselines and reducing per-adapter memory by up to 16x. When integrated into vLLM, SALT improves serving throughput by up to 51% under PCIe bandwidth pressure and 28% under GPU VRAM constraints for Llama-3.2-3B.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。