跳转至

LLaVA-FA: Learning Fourier Approximation for Compressing Large Multimodal Models

会议: ICLR 2026
arXiv: 2602.00135
代码: 无
领域: 多模态大模型
关键词: 模型压缩, 傅里叶变换, 低秩分解, 量化, 多模态语言模型

一句话总结

提出 LLaVA-FA,一种在频域进行联合低秩加量化权重近似的高效多模态大模型压缩方法,利用傅里叶变换的去相关性和共轭对称性实现更紧凑准确的权重表示,并引入 PolarQuant(极坐标量化)和 ODC(可选对角校准)方案,在多个基准上以最少的激活参数和计算成本超越现有高效多模态模型。

研究背景与动机

大型多模态模型(LMMs)在视觉-语言任务上表现出色,但其巨大的计算和内存成本阻碍了实际部署。例如 LLaVA 70B 模型训练需要超过 800 GPU小时(A100)。

现有压缩方法的问题

低秩分解与量化解耦:现有方法(如 LoRD、ASVD、LQER 等)将低秩分解和量化独立处理。低秩选择阶段对后续量化噪声视而不见,导致重建误差复合叠加

多模态冗余未充分利用:与纯文本 LLM 不同,大型视觉-语言模型额外携带图像编码器的跨模态适配器,每个新视觉领域的适配器秩膨胀,使得相同的低秩加量化方案对多模态模型仍显"臃肿"

校准数据依赖:许多压缩方法需要大规模校准数据集来估计 Hessian 矩阵

核心问题:如何更激进地压缩可学习参数,同时保持多模态模型的性能?

关键观察:傅里叶变换在数据压缩中有强大的表达能力——极稀疏的频谱信息就能恢复高保真信号。更重要的是,即使对于缺乏空间语义的权重矩阵,傅里叶变换也能有效处理近似问题。作者发现: - LMM 权重矩阵在频域具有更紧凑的奇异值分布 - 在相同秩下,频域低秩近似的累积误差小于空间域 - 傅里叶变换的共轭对称性可节省近一半可学习参数

方法详解

整体框架

LLaVA-FA 的核心思想是将权重矩阵的压缩从空间域转移到频域: 1. 对权重矩阵进行离散傅里叶变换(DFT) 2. 在频域进行低秩分解 3. 对复数矩阵进行极坐标量化(PolarQuant) 4. 可选的对角校准(ODC)方案消除大规模校准数据需求

关键设计

  1. 频域低秩分解:利用 DFT 的去相关能力

    • 将权重矩阵 \(W\) 变换为 \(\hat{W} = \text{DFT}(W)\)
    • 在频域对 \(\hat{W}\) 进行截断 SVD 得到低秩近似 \(\hat{W}_r = U_r \Sigma_r V_r^H\)
    • 理论保证:频域中同一秩的低秩近似的 Frobenius 误差小于空间域(论文给出了严格证明)
    • 共轭对称性:实数矩阵的 DFT 具有共轭对称性,只需存储一半系数,进一步压缩参数量
    • 能量集中:DFT 将信息集中到少数频率分量,使截断更有效
  2. PolarQuant(极坐标量化):专为复数矩阵设计的量化方案

    • 频域低秩分解后得到复数矩阵,传统实数量化方案不适用
    • PolarQuant 将每个复数 \(z = r e^{i\theta}\) 分解为幅度 \(r\)相位 \(\theta\)
    • 对幅度和相位分别进行均匀量化,各使用独立的缩放因子
    • 相比直接量化实部和虚部,极坐标表示保留了复数结构,避免了相位信息在低比特下的严重损失
    • 支持低至 2-4 bit 的极低比特量化
  3. ODC(Optional Diagonal Calibration):无需大规模校准数据的校准方案

    • 传统方法需要完整 Hessian 矩阵来校准压缩误差,计算成本高且需要大量代表性数据
    • ODC 利用深度网络 Hessian 矩阵常呈对角主导或低秩结构的经验观察
    • 用行/列均值近似完整 Hessian,大幅降低计算复杂度
    • 使压缩过程完全不需要校准数据,或仅需极少量数据

损失函数 / 训练策略

LLaVA-FA 是一种后训练压缩方法,不需要额外的微调或训练: - 直接对预训练权重进行 DFT → 低秩截断 → PolarQuant 量化 - ODC 校准也是一次性的前向计算 - 压缩后的模型可直接部署推理,无需重训练

这使得压缩流程极为轻量和实用。

实验关键数据

主实验

在多个视觉-语言基准上评估,包括理解型和幻觉型任务:

方法 激活参数 VQAv2 GQA SQA POPE 平均
LLaVA-1.5 (原始) 7B 基线 基线 基线 基线 基线
ASVD + Q ~2B 较低 较低 较低 较低 较低
LQER + Q ~2B 中等 中等 中等 中等 中等
LLaVA-FA 最少 最高 最高 最高 最高 最高

LLaVA-FA 在保持最少激活参数和最低计算成本的同时,在所有基准上超越了现有高效多模态模型。

消融实验

配置 效果 说明
空间域 vs. 频域低秩 频域更优 相同秩下重建误差更小
实部/虚部量化 vs. PolarQuant PolarQuant 更优 保持复数结构,相位信息更完整
无 ODC vs. 有 ODC ODC 提升明显 尤其在低比特量化时校准效果显著
仅低秩 vs. 低秩+量化 联合最优 频域中联合优化避免误差复合
不同比特宽度 4bit最佳性价比 2bit性能下降明显,4bit接近全精度

关键发现

  1. 频域低秩近似确实优于空间域:这不仅是实验现象,论文提供了理论证明——DFT 的去相关性使奇异值衰减更快
  2. 共轭对称性带来"免费"的2倍压缩:无需任何精度损失即可将参数量减半
  3. PolarQuant 对低比特至关重要:在 2-4 bit 下,直接量化实部虚部会导致相位信息严重扭曲
  4. ODC 消除了校准数据瓶颈:对角近似在深度网络 Hessian 上足够准确,使压缩变得真正"即插即用"

亮点与洞察

  1. 开创性的频域压缩视角:将神经网络权重压缩从空间域转到频域是全新思路,傅里叶变换的去相关、共轭对称和能量集中三大特性被充分利用
  2. 理论与实践结合:不仅有实验效果,还有严格的数学证明支撑频域低秩近似优于空间域
  3. 端到端一致的设计哲学:从低秩分解到量化再到校准,每一步都在频域/复数域中一致处理,避免了域间转换带来的信息损失
  4. 极低部署门槛:后训练压缩、无需校准数据、无需微调,使方法具有很强的实用性
  5. 对多模态模型的针对性设计:明确指出多模态模型比纯文本 LLM 面临更大的压缩挑战(跨模态适配器冗余)

局限与展望

  1. 推理时需要频域-空间域转换:虽然压缩了存储,推理时可能需要额外的 DFT/IDFT 计算开销
  2. 主要在 LLaVA 系列验证:是否适用于其他架构(如 Qwen-VL、InternVL)需要进一步验证
  3. 极低比特(1-2 bit)表现:在极低比特下性能损失仍然明显,可能需要结合知识蒸馏等技术
  4. 硬件支持:复数运算和极坐标量化在现有推理引擎中可能缺乏原生加速支持
  5. 动态场景适应:固定的截断秩可能不适合所有层,自适应秩选择策略值得探索

相关工作与启发

  • LoRA / QLoRA:低秩适配的先驱工作,但在空间域操作
  • ASVD / FWSVD:基于 SVD 的权重压缩,分别处理低秩和量化
  • LQER:联合低秩加量化方法,但在空间域
  • GPTQ / AWQ:仅量化方案,不涉及低秩分解

本文的核心启发:将信号处理中成熟的频域分析工具引入深度学习模型压缩,打开了一个全新方向。未来可能拓展到其他变换域(如小波变换)或其他模型架构。

评分

  • 新颖性: ⭐⭐⭐⭐⭐
  • 实验充分度: ⭐⭐⭐⭐
  • 写作质量: ⭐⭐⭐⭐
  • 价值: ⭐⭐⭐⭐

相关论文