arXiv:2507.06092cs.CRcs.AI2025-07中稿 · the 2026 ACM Asia …

用生成式AI合成数据,显著提升安全分类器在小样本下的性能

Taming Data Challenges in ML-based Security Tasks Using Generative AI

  • 用生成式AI生成合成数据,增强分类器泛化能力
  • 在仅180个训练样本时仍实现最高32.6%的性能提升
  • 适合数据稀缺或需快速应对概念漂移的安全场景

基于机器学习的监督分类器广泛应用于安全任务,其改进多集中于算法层面,而数据挑战影响分类器性能的问题未受足够重视。本文探讨生成式AI(GenAI)能否缓解这些数据难题。提出利用GenAI生成合成数据以扩充训练集,提升分类器泛化性。在7个不同的安全任务上,采用6种先进GenAI方法进行评估,并引入一种名为Nimai的新方案,实现高度可控的数据合成。结果表明,GenAI可显著提升分类器性能,在训练样本仅约180个的极端数据受限情况下,性能提升达32.6%。此外,GenAI能支持部署后快速适应概念漂移,调整过程所需标注极少。然而,部分GenAI方案在特定任务上难以初始化;我们识别出噪声标签、类分布重叠和稀疏特征向量等任务特性会阻碍生成式AI带来的性能增益。本研究将推动面向安全任务的下一代GenAI工具发展。

原文摘要 · Abstract (English)

Machine learning-based supervised classifiers are widely used for security tasks, and their improvement has been largely focused on algorithmic advancements. We argue that data challenges that negatively impact the performance of these classifiers have received limited attention. We address the following research question: Can developments in Generative AI (GenAI) address these data challenges and improve classifier performance? We propose augmenting training datasets with synthetic data generated using GenAI techniques to improve classifier generalization. We evaluate this approach across 7 diverse security tasks using 6 state-of-the-art GenAI methods and introduce a novel GenAI scheme called Nimai that enables highly controlled data synthesis. We find that GenAI techniques can significantly improve the performance of security classifiers, achieving improvements of up to 32.6% even in severely data-constrained settings (only ~180 training samples). Furthermore, we demonstrate that GenAI can facilitate rapid adaptation to concept drift post-deployment, requiring minimal labeling in the adjustment process. Despite successes, our study finds that some GenAI schemes struggle to initialize (train and produce data) on certain security tasks. We also identify characteristics of specific tasks, such as noisy labels, overlapping class distributions, and sparse feature vectors, which hinder performance boost using GenAI. We believe that our study will drive the development of future GenAI tools designed for security tasks.

生成式AI安全分类小样本学习数据合成

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