用强化学习生成难检测异常,让模型越练越强。
Swift Hydra: Self-Reinforcing Generative Framework for Anomaly Detection with Multiple Mamba Models
- 用强化学习在生成模型潜空间生成新异常样本
- 在ADBench上超越现有模型,推理时间短
- 多专家Mamba架构按数据复杂度动态扩容
尽管近年来涌现大量异常检测模型,其对未见异常的泛化能力仍不足,尤其在关键系统中。本文提出Swift Hydra框架,融合生成式AI与强化学习(RL),通过作用于生成模型潜变量的策略,合成可绕过检测模型的新颖、多样异常样本,并用于增强检测模型,提升其应对挑战性异常的能力。框架采用混合专家(MoE)结构的Mamba模型,根据数据复杂度动态调整专家数量,有效捕捉多样化特征分布,同时保持推理时间不变。在ADBench基准上的实证评估显示,该方法优于其他先进模型,且推理延迟较低。研究揭示了强化学习与生成式AI融合在推动异常检测发展中的新范式。
原文摘要 · Abstract (English)
Despite a plethora of anomaly detection models developed over the years, their ability to generalize to unseen anomalies remains an issue, particularly in critical systems. This paper aims to address this challenge by introducing Swift Hydra, a new framework for training an anomaly detection method based on generative AI and reinforcement learning (RL). Through featuring an RL policy that operates on the latent variables of a generative model, the framework synthesizes novel and diverse anomaly samples that are capable of bypassing a detection model. These generated synthetic samples are, in turn, used to augment the detection model, further improving its ability to handle challenging anomalies. Swift Hydra also incorporates Mamba models structured as a Mixture of Experts (MoE) to enable scalable adaptation of the number of Mamba experts based on data complexity, effectively capturing diverse feature distributions without increasing the model's inference time. Empirical evaluations on ADBench benchmark demonstrate that Swift Hydra outperforms other state-of-the-art anomaly detection models while maintaining a relatively short inference time. From these results, our research highlights a new and auspicious paradigm of integrating RL and generative AI for advancing anomaly detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。