arXiv:2601.15754cs.CRcs.AI2026-01

用分块梯度提升法实现高效稳定的恶意软件特征筛选

CAFE-GB: Scalable and Stable Feature Selection for Malware Detection via Chunk-wise Aggregated Gradient Boosting

  • 分块处理数据,用梯度提升模型局部估算特征重要性并聚合
  • 在两个大数据集上降低95%以上特征维度,性能与全特征相当
  • 适合需要可解释性与高可扩展性的大规模恶意软件检测场景

高维恶意软件数据常存在特征冗余、不稳定性及可扩展性差的问题,影响机器学习检测系统的有效性与可解释性。为解决此问题,本文提出CAFE-GB(基于梯度提升的分块聚合特征估计)框架,实现高维恶意软件检测中稳定且全局一致的特征排序。该方法将训练数据划分为重叠块,使用梯度提升模型估计局部特征重要性,并聚合得到稳健的全局排名。通过系统化的k选择与稳定性分析独立完成特征预算选取,平衡检测性能与鲁棒性。在两个大规模恶意软件数据集BODMAS和CIC-AndMal2020上评估,基于CAFE-GB筛选特征的分类器在准确率、F1-score、MCC、ROC-AUC和PR-AUC等指标上与全特征基线表现相当,同时特征维度减少超过95%。配对威尔科克森符号秩检验确认性能无显著下降。额外分析显示特征间冗余低,且通过SHAP解释提升了可解释性。运行时与内存分析表明下游分类开销显著降低。总体而言,CAFE-GB为大规模恶意软件检测提供了稳定、可解释且可扩展的特征选择策略。

原文摘要 · Abstract (English)

High-dimensional malware datasets often exhibit feature redundancy, instability, and scalability limitations, which hinder the effectiveness and interpretability of machine learning-based malware detection systems. Although feature selection is commonly employed to mitigate these issues, many existing approaches lack robustness when applied to large-scale and heterogeneous malware data. To address this gap, this paper proposes CAFE-GB (Chunk-wise Aggregated Feature Estimation using Gradient Boosting), a scalable feature selection framework designed to produce stable and globally consistent feature rankings for high-dimensional malware detection. CAFE-GB partitions training data into overlapping chunks, estimates local feature importance using gradient boosting models, and aggregates these estimates to derive a robust global ranking. Feature budget selection is performed separately through a systematic k-selection and stability analysis to balance detection performance and robustness. The proposed framework is evaluated on two large-scale malware datasets: BODMAS and CIC-AndMal2020, representing large and diverse malware feature spaces. Experimental results show that classifiers trained on CAFE-GB -selected features achieve performance parity with full-feature baselines across multiple metrics, including Accuracy, F1-score, MCC, ROC-AUC, and PR-AUC, while reducing feature dimensionality by more than 95\%. Paired Wilcoxon signed-rank tests confirm that this reduction does not introduce statistically significant performance degradation. Additional analyses demonstrate low inter-feature redundancy and improved interpretability through SHAP-based explanations. Runtime and memory profiling further indicate reduced downstream classification overhead. Overall, CAFE-GB provides a stable, interpretable, and scalable feature selection strategy for large-scale malware detection.

恶意软件检测特征选择梯度提升可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。