Search papers, labs, and topics across Lattice.
This paper introduces Subspace-Aligned LoRA Training (SALT), a novel hierarchical fine-tuning framework designed to enhance the serving efficiency of ultra-low-rank adapters while maintaining high downstream task performance. By jointly training high-capacity domain centroids and fine-tuning ultra-low-rank task residuals, SALT effectively aligns in-domain task subspaces, allowing for significant accuracy recovery even with minimal rank. The results demonstrate that SALT achieves up to 18.5% absolute accuracy gains over existing compression methods and improves serving throughput by up to 51% under bandwidth constraints.
Achieving high task performance with ultra-low-rank adapters, SALT recovers accuracy while slashing memory usage by up to 16x.
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.
