← 全部深度解读
📖 深度解读 · PERSONALIZATION

Modular Representation Compression: Adapting LLMs for Efficient and Effective Recommendations

Yunjia Xi, Menghui Zhu, Jianghao Lin, Bo Chen, Ruiming Tang, Yong Yu, Weinan Zhang · Huawei / SJTU · arXiv 2604.18146 · 发表 2026-04-20
8 分钟阅读时长 2914字 4.95综合分 第 2 名04-22 Personalization 日报
目录 · 6 节1问题与动机2方法详解3核心技术贡献4实验结果分析5局限与未来方向6工业实践意义
相关 5新颖 4实用 5证据 5时效 5
"中间层优于最终层" 的发现对 LLM+推荐的工程实践有重要指导意义:直接使用 LLM 最后一层 embedding 做推荐可能并非最优,应该系统性地评估各层表征质量。MARC 的模块化设计使其可以作为即插即用组件集成到现有推荐 pipeline 中,降低 LLM 表征的存储和计算成本。2.82% eCPM 提升在搜索广告场景下是非常可观的收益,验证了 LLM 表征压缩在工业界的实用价值。

1. 问题与动机

随着 LLM 在推荐系统中的应用日益广泛,一个关键工程挑战浮现:LLM 生成的高维表征(如 Llama3-8B 的 4096 维)带来了巨大的存储和计算开销。以实际数据衡量,采用 LLM 表征相比 BERT 级别模型会导致 140 倍的存储增长和 20 倍的训练时间增加。

现有解决方案通常采用降维压缩(如 PCA)来缩减表征维度,但作者发现了一个反直觉现象——中间层表征优势(Mid-layer Representation Advantage, MRA):LLM 中间层生成的表征在推荐任务上系统性地优于最终层输出。这意味着现有压缩方法在压缩已经退化的最终层表征,从根本上限制了压缩效果。

MRA 的发现动摇了"使用 LLM 最后一层 embedding"这一推荐系统工程中的默认实践,揭示了 LLM 内部在微调过程中会自发形成功能模块化——前中层负责通用表征学习,最终层被任务损失驯化为特定任务的适配器。

2. 方法详解

MARC(Modular Representation Compression)将 LLM 内部隐式的模块化重构为显式、可控的三模块设计:

模块 1:表征模块(Representation Module)

  • LLM 骨架生成原始表征 r_u, r_i ∈ ℝ^(d_o)(如 d_o=4096)
  • 关键设计决策:不对 LLM 输出施加直接的任务损失。任务监督信号仅通过外部模块的反向传播间接流入 LLM,防止最终层坍塌为任务特定预测器
  • 使用 LoRA 进行高效微调,冻结大部分 LLM 参数

模块 2:压缩模块(Compression Module)

  • 专用压缩网络 g(·):c_i = g(r_i), c_u = g(r_u)
  • 将 d_o 维压缩到 d_c 维(如 4096→128,实现 32 倍压缩)
  • 核心约束——HSIC(Hilbert-Schmidt Independence Criterion):

- L_HSIC = HSIC(X,Y) = 1/(n-1)² Tr(K_X J K_Y J)
- K_X, K_Y 是基于高斯核 k(x_i,x_j) = exp(-‖x_i-x_j‖²/2σ²) 的核矩阵
- J = I_n - 1/n·I_n·I_n^T 为中心化矩阵

  • HSIC 的优势在于能处理不同维度间的信息保持度度量,避免简单投影导致的信息扭曲
  • 压缩网络架构:MLP [256, 128]

模块 3:任务适配模块(Task Adaptation Module)

  • 输入:压缩后的表征 + 显式交互特征
  • 预测公式:ŷ = f(c_i, c_u, |c_i − c_u|, c_i ⊙ c_u)
  • 包含元素级差值和乘积的显式交互 + MLP 隐式交互
  • 交叉熵损失:L_match = −Σ[y·log(ŷ) + (1−y)·log(1−ŷ)]
  • 关键作用:吸收任务特定的优化压力,保护 LLM 骨架不被迫特化

总损失函数:L = L_match + α·L_HSIC(α=0.01)

训练细节:在完整训练集中随机采样 200,000 条数据进行训练;下游集成使用 MoE(2-5 个专家/任务)。

3. 核心技术贡献

贡献 1:首次发现并系统验证 MRA 现象。 通过跨模型(Qwen2-1.5B、Phi-2、Qwen2.5-3B、Qwen2-7B、Llama3-8B)和跨数据集(MovieLens-1M、MovieLens-25M、Yelp)的实验,证明中间层表征优势具有普遍性,而非个别模型的特例。该发现改变了"总是使用最后一层"的工程默认实践。

贡献 2:基于功能模块化理论的解释框架。 将 MRA 归因于 LLM 在微调过程中自发形成的功能分工:前中层形成通用表征学习模块,最终层演化为任务适配模块。这一理论框架不仅解释了 MRA,还指导了 MARC 的架构设计。

贡献 3:HSIC 约束的信息保持压缩。 与传统降维方法(PCA、MRL)不同,HSIC 在不同维度空间间度量信息依赖性,避免了强制线性映射导致的信息扭曲。消融实验显示 HSIC 是 MARC 中最关键的组件。

贡献 4:工业级搜索广告场景的在线验证。 2.82% 的 eCPM 提升在竞争激烈的搜索广告场景中属于显著收益,且推理延迟与现有系统相当。

4. 实验结果分析

CTR 预测任务(对比 MRL、PCA 等基线):

数据集MARC-C (128d)MRL (128d)全维 Frozen (4096d)
ML-1M0.8011 AUC0.79220.7948
ML-25M0.82330.81860.8197
Yelp0.74270.73720.7389

关键发现:128 维的 MARC-C 超越 4096 维的冻结 LLM 表征(ML-1M 上 +0.63%),说明压缩过程反而通过去除冗余噪声提升了信息密度。

重排序任务:MARC-C 在 MAP@5 和 NDCG@5 上均显著优于 MRL(p<0.05),MAP@5 从 0.7395 提升至 0.7445。

检索任务:MARC-C 在 HitRate@10 上比 MRL 提升 12.67%(0.3177→0.3580),展示跨推荐阶段的泛化能力。

消融分析显示:

  • HSIC 约束影响最大——移除后性能大幅下降
  • 显式交互特征对压缩表征尤为重要(对原始高维表征影响有限)
  • 模块化结构本身即可缓解 MRA,不依赖特定损失函数

维度敏感性:在 16-512 维范围测试中,128 维为最佳点。256 维以上性能反而下降,表明存在信息饱和效应。

在线 A/B 测试:在大规模商业搜索广告场景(千万级用户和广告)中,7 天 10% 用户分流实验取得 2.82% eCPM 提升,推理延迟与现有系统持平。

5. 局限与未来方向

作者隐含承认的局限:(1)MRA 现象目前仅在推荐微调场景中验证,是否推广到其他领域未知;(2)HSIC 的计算复杂度为 O(n²)(n 为 batch size),虽然训练数据采样至 200K 控制了成本,但在更大规模数据上可能成为瓶颈;(3)最优层选择策略(哪一个中间层最优)未被深入探讨——论文展示了层级性能曲线但未提出自适应选择机制。

未涉及但值得关注的方向:(4)LoRA rank 与 HSIC 效果的交互关系;(5)压缩模型在增量更新(新用户/新物品到达)场景下的表现;(6)α 超参数的自适应选择(当前固定为 0.01)。

6. 工业实践意义

对搜索广告系统的直接价值:MARC 解决了 LLM 表征在在线推理中的核心瓶颈——存储和延迟。32 倍压缩(4096→128)意味着 embedding cache 的内存需求降低一个数量级,使 LLM 增强的推荐系统在延迟敏感的在线服务中变得可行。

工程落地建议:

  • 立即可行:对现有 LLM+推荐 pipeline 进行层级表征质量评估,确认是否存在 MRA 现象——若存在,切换到中间层表征可获得"免费"的性能提升
  • 中期规划:部署 MARC 的三模块架构,重点是 HSIC 约束的实现(核矩阵计算可用 GPU 高效并行化)
  • 注意事项:128 维可能并非所有场景的最优维度,需根据具体业务数据量和特征空间大小进行调优

与 RankUp 的互补性:RankUp 从架构端提升表征秩(使表征更丰富),MARC 从压缩端保持表征质量(使丰富表征更高效)。两者可在同一推荐 pipeline 中协同使用:先用 RankUp 提升原始表征质量,再用 MARC 压缩存储。

2.82% eCPM 提升的商业意义:在搜索广告场景中,eCPM 每提升 1% 通常对应千万级年化收入增长。2.82% 的提升在行业内属于强有力的验证结果,特别是考虑到这是在已有强基线上的增量收益。

← 全部深度解读 出现于报告:2026-04-22 · Personalization #2 (4.95) 原文 ↗