【论文阅读】Detecting Spacecraft Anomalies Using LSTMs and Nonparametric Dynamic Thresholding
Metadata
authors:: Kyle Hundman, Valentino Constantinou, Christopher Laporte, Ian Colwell, Tom Soderstrom container:: Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining year:: 2018 DOI:: 10.1145/3219819.3219845 rating:: ⭐⭐⭐ share:: true comment:: 重点在于通过设定目标函数,最优化σsigmaσ,以此确定阈值范围。
前言
基于预测和阈值的方法来做异常检测,在常规的方法中,我们往往会使用一种最朴素的方法来确定阈值,也就是设置一个常数,如果超过这个常数,那么认为某个点是「异常」的。
论文《Detecting Spacecraft Anomalies Using LSTMs and Nonparametric Dynamic Thresholding》 提到了一种动态设置阈值的方法,最近也是学习实现了一下。
你可以在这里找到原文:
Detecting Spacecraft Anomalies Using LSTMs and Nonparametric Dynamic Thresholding
Errors and Smoothing
在论文中,作者首先使用 LSTM 对序列进行预测,得到预测值。
每一步
可以得到一个预测值
,计算预测误差为 e
而后就可以得到一个误差序列,序列的长度为
。
简单来说,也就是实际值
与预测值y
之间的误差的绝对值序列,每次取时间窗口大小为hhh进行异常检测。
对上述误差序列进行指数加权滑动平均(EWMA),这一操作的目的在于:LSTM 的预测中经常出现的误差峰值,当数据发生突变时,往往不能很好地被预测到,这就会使得e(t)e^{(t)}e(t)变得非常大,但其实这是一个非常常见的现象。通过 EWMA 就可以很好地消除这一现象,得到以下序列:
Threshold Calculation and Anomaly Scoring
在将误差经过平滑处理之后,开始进行确定阈值。
总体来说,这个方法的思想也很简单,就是用到了「均值」和「标准差」。
我们通过下面这个式子来确定阈值
其中 z 是人为规定的常数,阈值是通过下式决定的:
其中:
具体来说,作者通过枚举
,每次计算得到一个阈值,使得上式取到最大值。其中
和
分别是去除异常点前后「均值」和「标准差」的变化。分母
为序列中异常值的个数,
为异常值连续序列的个数。
简单来说,过程是这样的,首先枚举zzz,然后计算得到一个阈值,然后计算去除超过阈值的点的前后均值和标准差的变化,并通过上式计算得分,使得得分最大的zzz也就是我们需要的。
Pruning Anomalies
另外,作者提出了一种修剪异常点的方法。
选取异常序列中的最大值序列和正常序列的最大值降序排列,组成序列emaxe_{max}emax
对
中的连续两个序列计算:
接着是指定一个最小下降率
,若
,则认为
所属的连续序列是正常值,否则还是异常值。
如上图所示,设p=0.1p=0.1p=0.1,Anomaly 2 的斜率为 (0.01396–0.01072)/0.01396=0.23>p(0.01396 – 0.01072)/0.01396 = 0.23 > p(0.01396–0.01072)/0.01396=0.23>p,因此依旧是异常,而 Anomaly 1 的斜率为
,所以将其视为正常值。
这一流程的想法在于,如果异常值中的值与正常序列的最大值差距不大,则他们可能只是正常的抖动,而不是真正意义上的异常值。
代码实现
你可以在这里找到论文原文的代码:khundman/telemanom
由于论文原文中动态阈值部分的代码很大程度上与其他部分结合在了一起,导致阅读体验有所不佳。为了需要,我也仔细阅读了原文代码,参考原文代码对动态阈值部分进行了封装,并对几个点进行了修改。
Data and calculations for a specific window of prediction errors. Includes finding thresholds, pruning, and scoring anomalous sequences for errors and inverted errors (flipped around mean) - significant drops in values can also be anomalous.
异常窗口类
代码语言:javascript复制class ErrorWindow():
def __init__(self, e_s):
self.e_s = e_s
self.i_anom = np.array([]) # 窗口中异常点的下标
self.E_seq = np.array([]) # 连续异常点的首尾坐标
self.non_anom_max = float('-inf') # 窗口中非异常点的最大值
self.sd_lim = 12.0 # z 范围的最大值
self.sd_threshold = self.sd_lim # 最佳的 z
self.mean_e_s = np.mean(self.e_s)
self.sd_e_s = np.std(self.e_s)
self.epsilon = self.mean_e_s self.sd_lim * self.sd_e_s # 阈值
self.p = 0.15 # 连续误差序列最大值的波动率
def find_epsilon(self):
'''寻找最佳的 z 值'''
e_s = self.e_s
max_score = float('-inf')
# 遍历寻找最佳 z
for z in np.arange(1.5, self.sd_lim, 0.5):
# 计算阈值
epsilon = self.mean_e_s (self.sd_e_s * z)
# 除去异常点后的序列
pruned_e_s = e_s[e_s < epsilon]
# 大于阈值的点的下标
i_anom = np.argwhere(e_s >= epsilon).reshape(-1)
# # 设置缓冲区
# buffer = np.arange(1, 2)
# # 将每个异常点周围的值(缓冲区)添加到异常序列中
# i_anom = np.concatenate(
# (i_anom, np.array([i buffer for i in i_anom]).flatten(),
# np.array([i - buffer for i in i_anom]).flatten()))
# i_anom = i_anom[(i_anom < len(e_s)) & (i_anom >= 0)]
i_anom = np.sort(np.unique(i_anom))
# 如果存在异常点
if len(i_anom) > 0:
# 得到连续异常点的下标
groups = [list(group) for group in mit.consecutive_groups(i_anom)]
# 得到连续异常点的首尾坐标
E_seq = [(g[0], g[-1]) for g in groups if not g[0] == g[-1]]
# groups: [[1, 2, 3, 4], [6, 7], [9, 10]]
# E_seq: [(1, 4), (6, 7), (9, 10)]
# 计算去除异常点前后均值,方差的变化
mean_delta = (self.mean_e_s - np.mean(pruned_e_s)) / self.mean_e_s
sd_delta = (self.sd_e_s - np.std(pruned_e_s)) / self.sd_e_s
# 计算得分
score = (mean_delta sd_delta) / (len(E_seq)**2 len(i_anom))
if score >= max_score and len(E_seq) < 6 and len(i_anom) < (len(e_s) / 2):
max_score = score
self.sd_threshold = z
self.epsilon = self.mean_e_s z * self.sd_e_s
def compare_to_epsilon(self):
'''获取当前窗口小于阈值的最大值'''
e_s = self.e_s
epsilon = self.epsilon
# 找到异常点的下标
i_anom = np.argwhere(e_s >= epsilon).reshape(-1)
if len(i_anom) == 0:
return
# buffer = np.arange(1, 2)
# i_anom = np.concatenate((i_anom, np.array([i buffer for i in i_anom]).flatten(),
# np.array([i - buffer for i in i_anom]).flatten()))
# i_anom = i_anom[(i_anom < len(e_s)) & (i_anom >= 0)]
i_anom = np.sort(np.unique(i_anom))
# 获取当前窗口小于阈值的最大值
window_indices = np.setdiff1d(np.arange(0, len(e_s)), i_anom)
non_anom_max = np.max(np.take(e_s, window_indices))
groups = [list(group) for group in mit.consecutive_groups(i_anom)]
E_seq = [(g[0], g[-1]) for g in groups if not g[0] == g[-1]]
self.i_anom = i_anom
self.E_seq = E_seq
self.non_anom_max = non_anom_max
def prune_anoms(self):
'''修剪异常点'''
E_seq = self.E_seq
e_s = self.e_s
non_anom_max = self.non_anom_max
if len(E_seq) == 0:
return
# 得到每个连续异常序列中的最大值
E_seq_max = np.array([max(e_s[e[0]:e[1] 1]) for e in E_seq])
E_seq_max_sorted = np.sort(E_seq_max)[::-1]
# 每个连续异常序列中的最大值 非异常点的最大值
E_seq_max_sorted = np.append(E_seq_max_sorted, [non_anom_max])
i_to_remove = np.array([])
for i in range(0, len(E_seq_max_sorted) - 1):
# 在异常序列中最大误差之间的最小百分比下降
if (E_seq_max_sorted[i] - E_seq_max_sorted[i 1])
/ E_seq_max_sorted[i] < self.p:
i_to_remove = np.append(
i_to_remove,
np.argwhere(E_seq_max == E_seq_max_sorted[i])).astype(int)
else:
i_to_remove = np.array([])
i_to_remove.sort()
if len(i_to_remove) > 0:
E_seq = np.delete(E_seq, i_to_remove, axis=0)
if len(E_seq) == 0:
self.i_anom = np.array([])
return
indices_to_keep = np.concatenate(
[range(e_seq[0], e_seq[1] 1) for e_seq in E_seq])
mask = np.isin(self.i_anom, indices_to_keep)
self.i_anom = self.i_anom[mask]
划窗异常检测
代码语言:javascript复制def detect_anomaly(actual, pred, window_size: int):
'''异常检测
Args:
-------
actual(array): 真实值
pred(array): 预测值
window_size(int): 窗口大小
Returns:
-------
anomaly_list(List): 异常点序号
'''
e = abs(actual - pred)
smoothing_window = int(window_size * 0.05)
e_s = pd.DataFrame(e).ewm(span=smoothing_window).mean().values.flatten()
anomaly_list = np.array([])
for i in range(len(e_s) // window_size):
cur = np.array(e_s[i * window_size:(i 1) * window_size])
window = ErrorWindow(cur)
window.find_epsilon()
window.compare_to_epsilon()
if len(window.i_anom) == 0:
continue
window.prune_anoms()
if len(window.i_anom) == 0:
continue
window.i_anom = np.sort(np.unique(window.i_anom))
anomaly_list = np.append(anomaly_list,
window.i_anom i * window_size).astype('int')
return anomaly_list.tolist()
关于 inverted errors
在论文中没有提到的一个重要部分是「inverted errors」,也就是翻转的误差,具体来说也就是根据误差的均值线进行翻转,以此得到一个 inverted errors
。
e_s_inv = [mean_e_s (mean_e_s - e) for e in e_s]
具体来说,原文代码针对 errors
和 inverted errors
这两个序列分别进行了异常检测的步骤:确定阈值,修剪异常点。
window.find_epsilon()
window.find_epsilon(inverse=True)
window.compare_to_epsilon()
window.compare_to_epsilon(inverse=True)
window.prune_anoms()
window.prune_anoms(inverse=True)
但让我疑惑的点是,为什么要对翻转的序列再次进行异常检测?
可以预见的一种情况是:某个点相对附近一些点的误差非常小,但是在翻转之后,它就有很大的可能被认为是异常点。难道这个点不应该是正常点吗?在计算误差时,本身就使用了绝对值,为什么仍要根据误差的均值进行翻转?
这也是让我困惑的一点,因此在上面的代码中,我将 inverted errors
部分的代码去掉了。
关于 buffer
在原文代码中,另外还有一个点,也就是在上面的代码中被我注释掉的 buffer
缓冲区。事实上,我能理解但没完全理解。
具体来说,原文代码中将那些超出阈值的异常点检测出来之后,又进行了一步操作:将这个异常点前后 buffer
个元素加入异常点序列,然后在进行后面的步骤。
确实,异常点周围的点同样很可能也是异常点,即便它没有超过阈值。这样操作也可以为后面异常值的修剪做准备,我相信会有很大一批这样的点被修剪。
但是,两者之间异常点的检测数量差异仍然是巨大的,有很大一部分小误差点被认为是异常点,事实上也是如此。这也是让我纠结的一个点。获取调整窗口的值会让这些异常点的数量减少,但是这并不能解决根本问题。最后还是先注释掉吧。
参考资料
- Detecting Spacecraft Anomalies Using LSTMs and Nonparametric Dynamic Thresholding
- khundman/telemanom