文章目录
- 目的
- 优化函数
- 求解
- python 实现
目的
HP 滤波是一种分解信号的低频趋势与高频部分的滤波方法:
y t = g t + c t y_t = g_t + c_t yt=gt+ct
其中 y y y 代表原始信号, g t g_t gt 代表信号的低频趋势, c t c_t ct为更高频的周期或噪声。
优化函数
min g ∑ i N ( y i − g i ) 2 + λ ∑ i N − 2 ( g i − 2 g t + 1 + g i + 2 ) 2 (1) \min_g \sum_i^N (y_i - g_i)^2 + \lambda \sum_i^{N-2} (g_i - 2g_{t+1} + g_{i+2})^2 \tag{1} gmini∑N(yi−gi)2+λi∑N−2(gi−2gt+1+gi+2)2(1)
即
min g ∣ ∣ y − g ∣ ∣ 2 + λ ′ ∣ ∣ ∇ 2 g ∣ ∣ 2 (2) \min_\mathbf{g} ||\mathbf{y} - \mathbf{g}||^2 + \lambda' ||\nabla^2\mathbf{g}||^2 \tag{2} gmin∣∣y−g∣∣2+λ′∣∣∇2g∣∣2(2)
一方面是拟合:减小滤波后信号 g \mathbf{g} g 与原信号的误差;另一方面是平滑,限制平滑信号的二阶差分大小。
求解
从(1)(2)式中可以看出,目标函数是未知量的二次函数,所以可以用最小二乘法求解。
首先,用矩阵来表示信号的差分:
∇ g = [ − 1 1 − 1 1 ⋯ − 1 1 ] ( N − 1 ) × N g \nabla \mathbf{g} = \left[ \begin{array}{lllll} -1 &1 & & & \\ & -1 & 1 & & \\ &&\cdots && \\ &&& -1 & 1 \end{array} \right]_{(N-1)\times N} \mathbf{g} ∇g=⎣⎢⎢⎡−11−11⋯−11⎦⎥⎥⎤(N−1)×Ng
∇ 2 g = [ − 1 1 − 1 1 ⋯ − 1 1 ] ( N − 2 ) × ( N − 1 ) ∇ g \nabla^2 \mathbf{g} = \left[ \begin{array}{lllll} -1 &1 & & & \\ & -1 & 1 & & \\ &&\cdots && \\ &&& -1 & 1 \end{array} \right]_{(N-2)\times (N-1)} \nabla \mathbf{g} ∇2g=⎣⎢⎢⎡−11−11⋯−11⎦⎥⎥⎤(N−2)×(N−1)∇g
所以简单记为
∇ 2 g = D g \nabla^2 \mathbf{g} = D \mathbf{g} ∇2g=Dg
问题(2)转换成矩阵形式
min g ∣ ∣ y − g ∣ ∣ 2 + λ ∣ ∣ D g ∣ ∣ 2 = ( y − g ) ⊤ ( y − g ) + λ ( D g ) ⊤ ( D g ) = g ⊤ ( I + λ D ⊤ D ) g − 2 y ⊤ g + y ⊤ y \begin{array}{ll} \min_\mathbf{g} &||\mathbf{y} - \mathbf{g}||^2 + \lambda||D\mathbf{g}||^2\\\\ &= (\mathbf{y} - \mathbf{g})^\top (\mathbf{y} - \mathbf{g}) + \lambda (D\mathbf{g})^\top (D\mathbf{g}) \\\\ &= \mathbf{g}^\top(I + \lambda D^\top D) \mathbf{g} - 2\mathbf{y}^\top \mathbf{g} + \mathbf{y}^\top \mathbf{y} \end{array} ming∣∣y−g∣∣2+λ∣∣Dg∣∣2=(y−g)⊤(y−g)+λ(Dg)⊤(Dg)=g⊤(I+λD⊤D)g−2y⊤g+y⊤y
对上式求梯度得:
( I + λ D ⊤ D ) g = y (I + \lambda D^\top D) \mathbf{g} = \mathbf{y} (I+λD⊤D)g=y
所以解得:
g = ( I + λ D ⊤ D ) − 1 y \mathbf{g} = (I + \lambda D^\top D) ^{-1}\mathbf{y} g=(I+λD⊤D)−1y
python 实现
def hp(y, lamb=10):def D_matrix(N):D = np.zeros((N-1,N))D[:,1:] = np.eye(N-1)D[:,:-1] -= np.eye(N-1)"""D1[[-1. 1. 0. ... 0. 0. 0.][ 0. -1. 1. ... 0. 0. 0.][ 0. 0. -1. ... 0. 0. 0.]...[ 0. 0. 0. ... 1. 0. 0.][ 0. 0. 0. ... -1. 1. 0.][ 0. 0. 0. ... 0. -1. 1.]]"""return DN = len(ts)D1 = D_matrix(N)D2 = D_matrix(N-1)D = D2 @ D1g = np.linalg.inv((np.eye(N)+lamb*D.T@D))@ tsreturn g
测试
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inlineN = 100
t = np.linspace(1,10,N)
ts = np.sin(t) + np.cos(20*t) + np.random.randn(N)*0.1
plt.figure(figsize=(10,12))
for i,l in enumerate([0.1,1,10,100,1000, 10000]):plt.subplot(3,2,i+1)g = hp(ts,l)plt.plot(ts, label='original')plt.plot(g, label='filtered')plt.legend()plt.title('$\lambda$='+str(l))
plt.show()