跳到正文
OpenReview · Representation learning· Phai Vu Dinh; Quang Uy Nguyen; Dinh Thai Hoang; Diep N. Nguyen; Son Pham Bao; Eryk Dutkiewicz·· 2024-12-31精选AI 评分73

用于网络攻击检测中可分离表征学习的 Twin Auto-Encoder 模型

Twin Auto-Encoder Model for Learning Separable Representation in Cyberattack Detection

AI 导读

基于摘要分析。本文研究网络攻击检测中不同类别的表征混叠问题,尤其关注类别数量增加时的区分困难,提出 Twin Auto-Encoder(TAE),通过构造并学习动态表示目标,使不同类别的表征进一步分离。主要研究对象是网络安全数据,而非 EEG 或 BCI。 机制上,TAE 先将输入映射到潜在空间,再以确定性方式将不同类别的样本推得更远,形成 representation targets;随后由 decoder 学习将输入投影到这些目标,训练后也由 decoder 提取表征。作者将表示目标称为动态 codeword:与不包含输入数据信息的固定 codeword 相比,它会在每个训练 epoch 后针对每个样本更新。具体位移规则、目标更新公式、损失函数及标签参与方式尚未验证。 作者报告,实验涵盖七个 IoT botnet 数据集、两个网络 IDS 数据集、三个 malware 数据集、一个 cloud DDoS 数据集,以及用于考察类别数量增加情形的十个人工数据集。作者报告,相对其所称的现有先进模型,攻击检测 Accuracy 和 F-score 提升约 2%;摘要未明确这一提升是相对百分比还是百分点,也未说明 F-score 的计算口径。作者另报告 IoT 攻击检测平均 Accuracy 最高达 96.1%,模型大小约 1 MB,单样本表征提取平均运行时间约 2.6E-07 秒。数据划分、基线配置、平均方式、计时硬件与批处理条件,以及消融和统计信息尚未验证,不能据此直接判断部署效率。 平台推测(待验证):可检验的迁移假设是,依赖类别信息构造动态表示目标,可能有助于缓解 EEG 分类中的类别重叠。可复用的是目标更新与表征分离思路,输入编码及目标位移尺度需针对 EEG 改造;低信噪比、被试差异、通道变化和小样本可能使目标更新不稳定,或放大与类别相关的伪特征。小规模验证可在固定的 Cross-subject 运动想象划分中,保持编码器与训练预算一致,对照无目标位移、固定目标和动态目标,检查 Balanced Accuracy 与跨随机种子稳定性。原领域结果不等于 BCI 有效性,已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对其通过逐轮更新的动态表示目标缓解多类别表征混叠的机制与实现,但摘要中的性能及效率结果仍需结合评估协议验证,尚无 EEG/BCI 有效性证据。

来源:OpenReview · Representation learning · openreview.net