跳到正文
arXiv · cs.LG· Mahdi Arab Loodaricheh, Anita Raja, Mohammad Hossein Manshaei·· 19 小时前

不确定性引导的跨模态对齐:面向扩散模型的物联网入侵检测

Uncertainty-Guided Cross-Modal Alignment for Diffusion-Based IoT Intrusion Detection

AI 导读

研究者提出一种可靠性引导的跨模态对齐规则,用归一化 VAE 后验方差在扩散训练前控制各模态的方向性影响,用于不平衡物联网入侵检测中的少数类攻击生成。在 TON-IoT 上,该方法在三种扩散变体中取得最高平均精确率、召回率、密度与覆盖率;仅增强训练集后,真实测试集上 ROC-AUC 达 0.9515、PR-AUC 0.8954、平衡准确率 0.8395、攻击类 F1 为 0.7940。

来源:arXiv · cs.LG · arxiv.org