LeAP: Learnable Adaptive Permutation for Feature Selection in Heterogeneous and Sparse Recommender Systems

TL;DR

LeAP以可学习置换和自适应正则,在1.2万维工业模型中零损删去3600+冗余维度。

cs.LG 🔴 高级 2026-05-31 20 次浏览
Yihong Huang Chen Chu Fei Chen Yu Lin Ruiduan Li Zhihao Li
推荐系统 特征选择 可学习置换 稀疏特征 工业部署

核心发现

方法论

LeAP将传统逐特征Permutation Feature Importance改为一次前向传播中的可学习门控。对每个特征做batch-wise shuffle,得到噪声特征x′i;以gi=σ(θi/τ)融合原特征与停止梯度的置换特征:x̃i=gixi+(1−gi)sg(x′i)。再以置换散度Δi=1/B∑b||x(b)i−x′(b)i||2,经EMA平滑后设λi=αΔ̄i,并优化Ltotal=Ltask+∑iλigi。

关键结果

  • 在Avazu、Criteo、ML-1M和AliCCP四个公开数据集上,LeAP在50%和25% Feature Retention下均取得最高SAUC;相较AutoField、LPFS、SFS、SHARK等方法,论文报告其在复杂的Criteo与AliCCP上优势尤其明显。
  • 工业搜索排序模型包含500多个特征字段、超过12,000个维度,模型参数规模达2TB、日请求超过10亿。LeAP删除3,600多个冗余维度,即超过30%,实现核心线上指标Zero Diff,能力约为对比方法的2至10倍。
  • 理论上,若信号强度ΔJ=J(0)−J(1)>λi,门控梯度推动gi趋近1;冗余特征的ΔJ≈0时,正则项推动gi趋近0。相比SHARK的逐一贪心删除,LeAP可联合处理特征耦合。

研究意义

论文针对工业推荐中长期存在的三重矛盾:特征维度异构、极端稀疏和置换评估昂贵。它把可解释的置换重要性与可训练门控结合,使特征筛选能直接嵌入日常训练流程,并支持大规模在线模型的带宽和存储优化。结果显示,特征压缩不必以牺牲长尾个性化信号或业务指标为代价。

技术贡献

核心贡献包括:提出O(1)模型开销级别的Learnable Adaptive Permutation,避免传统方法对N个特征重复推理;提出基于Permutation Divergence的自适应正则,使惩罚随真实扰动规模变化,而非采用统一λ∑gi;通过凸性分析给出门控极化条件ΔJ>λi;设计插件式部署、阈值剪枝和少量微调流程,适配异构原生输入层。

新颖性

与AutoField的Gumbel-Softmax、LPFS的平滑L0和SHARK的高效置换不同,LeAP不是把特征权重大小直接当作重要性,而是学习“替换为保持边缘分布的噪声后损失多少”。其新意在于同时利用可学习置换的效率、Permutation Importance的解释性,以及针对维度和稀疏性的动态惩罚。

局限性

  • 公开数据集上的特征维度被统一,实验主要验证基础LeAP机制;真正的异构、高稀疏优势主要来自工业案例,公开基准缺少完全可复现的对应设置。
  • 理论分析假设期望任务损失J(gi)凸,而深度推荐网络通常非凸;因此门控极化定理解释力强,但不等于对所有训练过程提供严格保证。
  • 论文摘要与给定正文未提供各数据集逐项AUC、SAUC数值及完整超参数,难以精确复核不同压缩率下的统计显著性。

未来方向

后续可在公开基准中构造真实1D—256D异构和99%稀疏设置,报告完整AUC、SAUC、延迟及显存曲线;研究更稳健的非凸优化分析、分组或层级门控,以及跨时间漂移下的在线重新筛选。还可把LeAP扩展到广告、搜索、多模态模型和动态成本约束的联合架构搜索。

AI 总览摘要

工业推荐模型正在被海量特征推向太字节规模:统计量可能只有1维,用户行为嵌入却可达256维。特征越多,模型越可能捕捉复杂兴趣,但训练、存储和线上传输成本也随之上升。传统Lasso、Random Forest和XGBoost难以处理高维嵌入;AutoField、LPFS等门控方法通常假设维度一致;SHARK等置换方法则需要反复推理。在极端稀疏场景中,99%以上的默认值还会让重要的长尾信号看起来无用。

LeAP提出一种插件式方案。它在每个训练批次内随机打乱各特征,保持边缘分布和稀疏模式,再用可学习门控gi在原特征与置换噪声之间插值:x̃i=gixi+(1−gi)sg(x′i)。一次前向传播即可同时评估多项特征,计算复杂度从O(N×模型开销)降至O(1×模型开销)。其关键是Permutation Divergence:用原特征与置换特征的L2距离,经EMA平滑后形成λi=αΔ̄i,从而对高维特征施加更匹配的惩罚,对稀疏但未被扰动的特征保持宽容。

在Avazu、Criteo、ML-1M和AliCCP上,LeAP在50%与25%保留率下均取得最高SAUC。更具工程意义的是,在日请求超过10亿、参数规模2TB、输入超过12,000维的搜索排序模型中,它删除3,600多个冗余维度,超过30%,同时核心业务指标Zero Diff,效果为基线的2至10倍。论文的价值并不只是压缩模型,而是提供了一种把解释性重要性评估转化为可训练、可部署决策的路径;不过公开实验缺少逐项数值,且理论仍依赖凸性假设。

深度分析

研究背景

推荐、广告和搜索模型从少量人工特征发展到数千个标量、类别和行为嵌入。WideDeep等深度架构提升了表达能力,但也造成TB级参数、训练成本和CPU-GPU带宽压力。Lasso、树模型具有解释性,却不适合高维嵌入;AutoField、LPFS、SFS引入可学习掩码,但多采用统一正则;SHARK改善了置换效率,却仍面临重复推理和稀疏偏差。

核心问题

目标是在不显著损失预测性能的情况下识别并删除冗余特征维度。难点包括:1D统计量与256D嵌入无法公平比较;99%默认值使随机置换低估稀疏信号;逐特征Permutation需要O(N)次前向推理;深度网络中的参数权重大小也不等于真实信息重要性。

核心创新

  • �� 可学习置换:在同一批次内打乱特征,并通过门控融合原值和噪声,避免逐项重跑模型。• 自适应正则:以Δi=1/B∑||xi−x′i||2作为数据驱动尺度,经EMA得到λi=αΔ̄i。• 极化理论:当ΔJ>λi时gi趋近1,冗余特征则趋近0。• 工程插件:插入拼接层,收敛后按gi阈值或Top-K直接剪枝,再少量微调。

方法详解

  • �� 输入:F个特征xi∈Rdi,di可不同,批大小为B。• 置换:对每个字段独立采样RandomPermutation(B),形成x′i;它破坏样本标签关联,却保留边缘分布。• 门控:gi=σ(θi/τ),输出x̃i=gixi+(1−gi)sg(x′i),停止梯度避免噪声更新上游表示。• 正则:计算批次L2散度Δi,使用Δ̄(t)i=βΔ̄(t−1)i+(1−β)Δ(t)i平滑,并令λi=αΔ̄i。• 优化:Ltotal=Ltask+∑λigi。• 剪枝:采用gi<0.5的绝对阈值或按gi保留Top-K,移除LeAP后微调。

实验设计

公开数据包括Avazu(40,428,967样本、23字段)、Criteo(45,850,617、39字段)、ML-1M(1,000,209、9字段)和AliCCP(85,316,519、23字段)。骨干网络为WideDeep,采用Search-Retrain协议,比较Lasso、RF、XGBoost、AutoField、LPFS、SFS和SHARK。指标为AUC与SAUC,重点观察50%和25%保留率;工业数据来自长视频平台搜索、曝光、点击和互动日志,并进行Group AUC及线上A/B测试。

结果分析

LeAP在四个公开数据集的两种保留率下均获最高SAUC;论文特别指出,在Criteo和AliCCP上优于掩码方法。工业模型有500多个字段、12,000+维、2TB参数和10亿+日请求,LeAP删除3,600+维且核心指标Zero Diff。传统方法删除少于600维即出现明显退化,说明LeAP对特征耦合、异构维度和稀疏信号更稳健。

应用场景

最直接的应用是搜索排序、推荐和广告模型的输入压缩:在特征拼接层插入模块,利用线上数据累积门控和EMA统计,再按阈值剪枝。它可降低GPU显存读取、CPU-GPU带宽、训练时间和模型存储;适用前提是已有稳定任务模型、可访问批次数据,并允许短暂评估和微调。

局限与展望

论文没有给出四个公开数据集的逐项AUC/SAUC表格数值,且公开数据被统一维度化,限制了外部复现。凸损失假设并不符合一般深度网络;batch shuffle还可能在小批次、强时间相关或分布漂移数据上产生噪声。EMA、温度τ和全量门控增加少量训练开销,长期部署仍需周期性重评估和防止重要特征随业务变化被误删。

通俗解读 非专业人士也能看懂

把推荐模型想成一家大型餐厅,数千个特征就是数千种食材:有的只有一粒盐,有的是整箱蔬菜。传统筛选方法要么把所有食材按同样标准收费,要么每次只拿走一种食材重新做菜,太慢;对很少出现的食材,还可能因为“平时用得少”就误以为它没价值。

LeAP像一位会学习的主厨。每次做一批菜时,它把某种食材的来源标签打乱,再比较原菜谱和“被打乱菜谱”的味道差异。如果味道几乎不变,说明这项食材可能多余;如果变化很大,就保留它。主厨不是逐道菜重做,而是在同一批菜里同时测试所有食材,因此速度快得多。

它还会根据食材被打乱后变化多大来决定检查力度。大箱食材天然变化更明显,小粒食材变化较小;而一种极少使用、打乱后仍几乎不变的食材不会立刻被罚掉,只有确认它对味道确实没有帮助才会删除。最终,保留或删除的选择会变得清晰。真实系统中,LeAP从超过12,000个“食材位置”中删掉3,600多个,同时没有降低核心业务指标。

简单解释 像给14岁少年讲一样

想象你在玩一个推荐游戏:系统要根据玩家的兴趣决定下一条视频。它手里有一万多个线索,比如“最近看了什么”“常在哪个时间上线”,还有很长的行为向量。线索太多,游戏运行会变慢,所以要找出真正有用的线索。

LeAP的做法很像把同学名字打乱,再看小组作业会不会变差。它一次把很多线索的顺序打乱,而不是每次只测试一个,因此省时间。然后它给每条线索一个开关:开关接近1,说明线索重要;接近0,说明换成乱序内容也没关系,可以删掉。

有些线索很长,有些只有一个数字;有些线索平时99%的时间都是空白,但偶尔出现时可能非常关键。LeAP不会只看“长度”或“出现次数”,而是看打乱前后到底改变了多少信息。这样,罕见但关键的线索不会因为低频被误删。

论文在Avazu、Criteo、ML-1M和AliCCP上测试,并在一个每天超过10亿请求的搜索系统中使用。系统有12,000多个维度,LeAP删掉3,600多个仍保持核心指标不变。就像整理游戏背包:不是简单丢掉最少使用的物品,而是测试丢掉后战斗力是否真的下降!

术语表

Learnable Adaptive Permutation(可学习自适应置换)

把随机置换从一次性评估操作变成可训练的门控机制。模型学习每个特征应保留原值还是替换为批次噪声。

LeAP的核心模块。

Permutation Feature Importance(置换特征重要性)

打乱某特征并观察预测性能下降,以衡量该特征贡献。它解释性强,但逐特征重复推理成本高。

LeAP试图保留其解释性并降低成本。

Permutation Divergence(置换散度)

原特征与其置换版本之间的平均L2距离:Δi=1/B∑||xi−x′i||2。它衡量置换造成的信息扰动规模。

用于生成自适应正则权重。

EMA(指数移动平均)

用当前统计量和历史统计量的加权平均降低批次噪声。公式为Δ̄t=βΔ̄t−1+(1−β)Δt。

平滑Permutation Divergence。

Gate(门控)

取值在0到1之间的可学习开关。g接近1保留原特征,g接近0使用置换噪声并倾向剪枝。

通过温度缩放Sigmoid获得。

SAUC(归一化AUC)

将各数据集上的AUC除以该数据集所有方法中的最佳AUC,再跨数据集平均。它用于综合比较。

公开实验的主要聚合指标。

开放问题 这项研究留下的未解疑问

  • 1 公开基准已统一特征维度,尚不能充分验证1D与256D混合、99%稀疏条件下的公平性;需要发布带真实异构输入和完整逐项AUC的可复现实验。
  • 2 门控极化证明依赖J(gi)凸性,而工业深度模型通常非凸。未来应研究随机优化、特征交互和分布漂移下的概率保证。
  • 3 论文未量化LeAP训练额外开销、EMA参数敏感性和长期线上重筛选频率;这些因素决定大规模持续部署的实际成本。

应用场景

近期应用

搜索排序模型压缩

在特征拼接层插入LeAP,使用线上日志训练门控,按gi阈值或Top-K删除冗余维度。适用于已有稳定排序模型、需要降低GPU带宽和模型存储的团队;论文案例显示可删3,600+维且Zero Diff。

推荐与广告特征清理

对统计特征、类别嵌入和行为向量统一进行敏感性筛选,避免仅按频率删除长尾特征。完成门控收敛后移除LeAP并小规模微调,可减少训练和在线推理资源。

远期愿景

动态特征生命周期管理

将LeAP接入持续训练和监控系统,结合业务漂移、延迟预算与存储成本周期性重估特征。未来可实现按场景、设备或用户群自适应的特征集合。

多模态模型输入优化

把置换散度扩展到文本、图像、序列和稀疏ID表示,形成统一的可解释输入裁剪框架。主要障碍是跨模态相关性、时间因果关系和更高的评估成本。

原文摘要

Modern industrial recommender systems rely on thousands of heterogeneous features -- ranging from low-dimensional scalars (e.g., statistical value) to high-dimensional embeddings (e.g., user-id embeddings, MLP representations) -- to achieve high-precision predictions. Given the immense computational costs associated with training, efficient feature selection is critical. However, existing methods encounter three primary bottlenecks: (1) they typically assume uniform feature dimensions or require costly mapping to a fixed size; (2) they struggle with extreme sparsity, where the majority of features (e.g., 99%+) remain at default values; and (3) traditional permutation-based approaches are computationally prohibitive in large-scale settings. To address these challenges, we propose LeAP (Learnable Adaptive Permutation), a novel, model-agnostic plug-in module for feature selection. LeAP transforms the inefficient random permutation process into a learnable mechanism, significantly accelerating the evaluation of feature importance. In addition, we introduce an adaptive regularization strategy tailored for heterogeneous dimensions and extreme sparsity, enabling superior feature importance ranking results across asymmetric input spaces. Experiments on four public recommendation datasets demonstrate that LeAP achieves state-of-the-art performance. Furthermore, LeAP has been deployed in a large-scale industrial search ranking model with over a billion daily requests and a 2TB model parameter scale. In this real-world scenario involving 12,000+ total feature dimensions, LeAP successfully identified and removed over 3,600 redundant dimensions without performance degradation, which is 2 to 10 times the ability of compared baseline methods.

cs.LG