arxiv.org·2026/10/26
为把大模型部署到内存与算力受限的边缘设备,SALAAD这类非结构化稀疏+低秩(SLR)压缩虽能弹性省参,却因不规则访存无法转化为加速。S-SALAAD用ADMM在训练中联合学习权重、低秩与结构化稀疏分量,支持block、N:M、行/列稀疏且不改架构;训练用I-SVD降低低秩更新开销,EGAR按能量自适应分配预算并交替细化L与S,从单检查点零样本压缩到多预算,推理为block稀疏开发CPU/GPU核。在Llama2-1B预训练与Nemotron-3-Super-120B到Qwen3-1.7B蒸馏上,对比Wanda、SparseGPT、SliceGPT、SVD-LLM、LORO、SALAAD和vanilla,峰值GPU显存降42.4%、吞吐提升1.99×,Qwen蒸馏同样推进Pareto前沿;零样本下游准确率随预算收紧缓慢退化。块粒度宜随模型规模增大,但N:M固定容量限制弹性,专用核目前仅覆盖block稀疏。
为什么重要 对边缘部署大模型直接有用:训练时注入结构化稀疏,省去校准压缩,推理内存与延迟同步下降,适合资源受限场景。适合 适合关注大模型边缘部署、模型压缩与训练/推理系统的工程师与研究者。
规模定律持续奖励模型规模的增加,但由此带来的计算和内存需求推动推理更加依赖云基础设施。然而,不可靠的网络连接、通信延迟和隐私要求可能使这种依赖不切实际,从而促使在边缘设备上本地执行。
本地部署要求模型在有限内存中容纳,并利用可用计算资源高效执行。最近的工作表明,学习稀疏加低秩表示能够将单个检查点弹性压缩到多个内存预算。然而,更少的参数不一定转化为更快的推理。非结构化稀疏分量中的索引开销和不规则内存访问可能抵消计算节省。
为了解决这一关键部署障碍,我们提出 S-SALAAD,一个统一框架,在训练期间联合学习模型权重及其低秩和结构化稀疏分量。该框架支持一系列硬件兼容的结构化稀疏模式。它将这种结构学习与预算感知压缩以及专用 CPU 和 GPU 内核相结合,利用所学结构进行加速,从而在多个内存预算下从单个检查点实现高效推理。
在 Llama2 和 Qwen3 模型上的实验(覆盖预训练和蒸馏)表明,相对于所评估的基线,S-SALAAD 在多个硬件平台上推进了困惑度、内存和推理延迟方面的经验 Pareto 前沿。这些结果表明,结构化学习能够将弹性可压缩性转化为更低的内存需求和更低的推理延迟,从而实现在边缘设备上实际部署大模型。
Scaling laws continue to reward increases in model scale, while the resulting computational and memory demands drive greater reliance on cloud infrastructure for inference.
Yet unreliable network connectivity, communication delays, and privacy requirements can make such reliance impractical, motivating local execution on edge devices.
Local deployment requires models to fit within limited memory and execute efficiently with the available computational resources.
Recent work has shown that learning sparse-plus-low-rank representations enables elastic compression of a single checkpoint to multiple memory budgets.
However, fewer parameters do not necessarily translate into faster inference. Indexing overhead and irregular memory access in unstructured sparse components can offset computational savings.
To address this critical deployment barrier, we introduce S-SALAAD, a unified framework that jointly learns the model weights and their low-rank and structured sparse components during training.
The framework supports a family of hardware-compatible structured sparsity patterns.
It couples this structure learning with budget-aware compression and dedicated CPU and GPU kernels that exploit the learned structure for acceleration, enabling efficient inference from a single checkpoint across memory budgets.
Experiments on Llama2 and Qwen3 models, covering both pretraining and distillation, show that S-SALAAD advances the empirical Pareto frontier in perplexity, memory, and inference latency across multiple hardware platforms relative to the evaluated baselines.
These results show that structured learning can translate elastic compressibility into reduced memory requirements and lower inference latency, enabling practical deployment of large models on edge devices.
Llama 内存降
500M 峰值 GPU 内存
Llama 吞吐升
500M 吞吐
Qwen 内存降
800M 峰值 GPU 内存
Qwen 吞吐升
800M 吞吐
训练开销
术语解释暖启动迭代奇异值分解,从缓存右奇异基V开始单步正交迭代,降低训练开销。 vs vanilla
注释 · 图4 对比 dense 模型,在 Quadro RTX 6000 上测得峰值内存与吞吐。出自 §5.1 / 图4
注释 · 困惑度越低越好;更细粒度块在紧预算时保留更多容量灵活性。出自 图6(b) / §5.2



EVIDENCE / METHODS
表格
在 Quadro RTX 6000 上,S-SALAAD 同时降低峰值内存并提升吞吐。
| 设置 | 峰值内存 GB | 吞吐 ktoken/s |
|---|---|---|
| Llama-1B dense 500M | 4.08 | 29.35 |
| Llama-1B S-SALAAD 500M | 2.35 | 58.47 |
| Qwen3-1.7B dense 800M | 5.58 | 24.19 |
| Qwen3-1.7B S-SALAAD 800M | 3.82 | 39.22 |
注释 · 数值来自图4;吞吐单位为 ktoken/s,内存单位为 GB。出自 图4 / §5.1
原版排版与图表 · 手机无法内嵌时可在浏览器打开。PDF 插件内选文不能传给网页助手。