arXiv:2608.15504cs.LGstat.ML2026-08

提出高效鲁棒训练框架PERO,提升加密流量分类模型对高风险误判的防御能力。

PERO: Efficient Robust Post-Training Foundation Models for Encrypted Traffic Classification

论文配图:PERO: Efficient Robust Post-Training Foundation Models for Encrypted Traffic Classification
图 1 · 摘自论文原文
  • 用轻量代理估算样本风险,筛选高危样本进行优化
  • 在多个数据集上实现更优鲁棒性,计算开销降低显著
  • 适合对安全误判敏感的网络监控场景

加密流量分类对网络安全至关重要,但实际部署中对罕见但高损失的误判(如将恶意流量误判)极为敏感。现有的加密流量基础模型虽整体性能优异,但采用经验风险最小化等标准目标常忽略高风险尾部事件,且常用指标难以反映其在风险敏感场景下的脆弱性。直接应用条件风险价值等鲁棒优化目标进行后训练对大模型计算开销巨大,因需耗时识别高损失样本。为此,本文提出预评估鲁棒优化(PERO)框架,通过轻量代理估计样本级风险,仅选取高风险样本更新基础模型,将风险评估与昂贵的大模型优化解耦。在典型加密流量数据集上的大量实验表明,PERO在保持或超越现有优秀鲁棒后训练方法的同时,显著降低计算与内存开销。

原文摘要 · Abstract (English)

Encrypted traffic classification is vital for network security, yet real-world deployments are inherently sensitive to rare but high-loss errors such as misclassification of malicious traffic. The encrypted traffic foundation model, as a promising general-purpose technique, can achieve impressive overall performance. However, employing standard objectives such as empirical risk minimization often overlooks high-risk tail events, and commonly used performance metrics hardly reflect robustness limitations in risk-sensitive scenarios. Directly applying robust optimization objectives, such as conditional value-at-risk, to post-training is computationally prohibitive for large models, as identifying high-loss samples exhausts substantial computation. To this end, we propose Pre-Evaluation Robust Optimization (PERO), an efficient robust post-training framework for encrypted traffic foundation models. PERO employs a lightweight proxy to estimate sample-wise risk and selects a subset of high-risk samples to update the foundation model, decoupling risk estimation from expensive large-model optimization. Extensive experiments on typical encrypted traffic datasets show that PERO achieves competitive or superior robustness and average performance compared to outstanding robust post-training methods, while significantly reducing computational and memory costs.

加密流量鲁棒训练基础模型安全分类

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