数据STUDIO

卡尔曼滤波器简明解释

Image

Image

数据有噪声? 掌握这个算法,让你的预测更智能。

你是否曾经遇到过这样的场景:无人机在空中晃动,GPS数据跳来跳去;自动驾驶汽车传感器读数相互矛盾;机器人定位时,里程计和激光雷达数据对不上……

这些问题的核心,都是如何从“有噪声”的数据中提取“真实”信号。

今天要介绍的 卡尔曼滤波器(Kalman Filter),就是解决这类问题的“数学瑞士军刀”。它能优雅地融合预测与测量,在不确定性中给出最优估计。

一、卡尔曼滤波器:一句话说清是什么

卡尔曼滤波器是一个算法,用于预测物体随时间变化的“状态”(如位置、速度等),即使在传感器数据充满噪声和不确定性的情况下。

这里的“状态”可以是你关心的任何随时间变化的量:

  • 位置(一维坐标、GPS经纬度)
  • 速度、加速度
  • 方向、角度
  • 温度、海拔、电池电量……

核心思想就像一位经验丰富的船长:他会根据船的当前速度和方向(预测),结合偶尔看到的灯塔位置(有误差的测量),在脑海里绘制出最可能的航线。预测给了他连续性,测量帮他修正偏差,两者结合,路线越来越准。

二、从一维开始:直观理解贝叶斯更新

卡尔曼滤波器的数学基础是贝叶斯推理。我们先用一个简单的一维定位例子来感受它的魔力。

想象你在一条长长的走廊里蒙眼行走,只能靠偶尔摸墙来感知位置。你心里有两个信息源:

  1. 你的步伐(预测):根据步数和步长,你推测自己走了大概10米,但不太确定(方差大)。
  2. 你的触摸(测量):你摸到墙上的一个标志,感觉在12米处,但手感可能不准(也有误差)。

卡尔曼滤波器会怎么做?它不会简单地取平均值(11米),而是“加权平均”:谁更确定(方差小),就听谁的更多一点。最终得出的估计,其确定性会比任何一个单独信息来源都高!

现在,请看下图——一维卡尔曼滤波器中的贝叶斯更新——它直观地展示了先验知识和新观察结果如何结合起来,形成更清晰、更自信的估计。

按回车键或点击查看完整尺寸的图片

Image

先验信念:

这是在纳入新测量值之前对状态的估计。它具有一定的均值(例如,来自预测)和方差(不确定性)。

测量似然:

这表示在真实状态为 x 的情况下观察到测量值 z 的概率。它通常具有较小的方差(假设传感器精度很高)。

经过测量并应用贝叶斯规则后,卡尔曼滤波器将产生一个更精确的新估计值:

后验:

结合两种信息来源的新估计值。

  • 均值会偏移到先前均值和测量值之间的某个位置。
  • 方差减小,因为现在你有了更多信息(预测+测量)。

为什么后部更窄(更确定)

卡尔曼滤波器最强大的方面之一是它如何通过结合两个信息来源来减少不确定性:你的先验信念和测量结果。

卡尔曼滤波器计算更新后的(后验)方差如下:

在哪里:

  • 是先验分布的方差(预测不确定性)。
  • 是测量值的方差(传感器噪声)。

该公式保证:

换句话说,更新后的信念(后验概率)更有把握——它的不确定性比单独的预测或测量都要小。

在先验值和测量值相差甚远的极端情况下,卡尔曼滤波器仍然可以产生更可靠的估计值,如下图所示:

Image

代码时间:一维卡尔曼滤波

下面的代码完美诠释了这个过程。我们从一个极其不确定的初始位置开始,通过反复“预测-更新”,你会看到不确定性(方差)像雪球融化一样迅速缩小。

随着置信度的提高,不确定性(方差)会减小!
随着置信度的提高,不确定性(方差)会减小!
# 1D卡尔曼滤波核心函数
# 测量更新:融合先验估计与新的观测值
defupdate_belief(prior_mean, prior_var, meas_value, meas_var):
"""
    参数:
    prior_mean: 预测的平均值 (μ_prior)
    prior_var:  预测的方差 (σ²_prior)
    meas_value: 测量值 (z)
    meas_var:   测量的方差 (σ²_meas)

    返回:
    [新的平均值, 新的方差]
    """

# 核心公式:加权平均,权重为方差的倒数(不确定性越小,权重越大)
    combined_mean = (meas_var * prior_mean + prior_var * meas_value) / (prior_var + meas_var)
# 新方差:融合后信息更多,不确定性必然降低
    combined_var = 1 / (1 / prior_var + 1 / meas_var)
return [combined_mean, combined_var]

# 预测步骤:根据运动模型预测下一个状态
defpredict_state(current_mean, current_var, motion_change, motion_var):
"""
    参数:
    motion_change: 控制输入或预计的移动量 (u)
    motion_var:    运动模型的不确定性 (σ²_motion)
    """

    predicted_mean = current_mean + motion_change
    predicted_var = current_var + motion_var  # 运动增加了不确定性
return [predicted_mean, predicted_var]

# ===== 实战模拟 =====
# 模拟数据:传感器读数(有噪声的真实位置)
sensor_readings = [5.0, 6.0, 8.0, 9.0]
# 模拟数据:每一步施加的控制(移动量)
motions_applied = [1.0, 2.0, 2.0, 1.0]

# 噪声参数
sensor_noise = 3.0# 测量噪声方差,值越大传感器越不准
motion_uncertainty = 2.0# 运动模型噪声方差,值越大模型越不可靠

# 初始估计:我们完全不知道在哪,所以方差极大
estimated_position = 0.0
position_uncertainty = 500.0

print("=== 开始卡尔曼滤波 ===")
for i in range(len(sensor_readings)):
    print(f"\n--- 第 {i+1} 步 ---")

# 1. 更新步骤:融合传感器数据
    estimated_position, position_uncertainty = update_belief(
        estimated_position, position_uncertainty, 
        sensor_readings[i], sensor_noise
    )
    print(f"更新后: 位置={estimated_position:.2f}, 不确定性={position_uncertainty:.2f}")

# 2. 预测步骤:根据运动模型预测
    estimated_position, position_uncertainty = predict_state(
        estimated_position, position_uncertainty, 
        motions_applied[i], motion_uncertainty
    )
    print(f"预测后: 位置={estimated_position:.2f}, 不确定性={position_uncertainty:.2f}")

# 最终结果
print(f"\n=== 最终估计 ===")
print(f"位置: {estimated_position:.2f} ± {position_uncertainty**0.5:.2f} (标准差)")

运行这段代码,你会看到:尽管初始猜测离真实值很远(0 vs 5),但仅仅几步之后,估计值就迅速收敛到真实轨迹附近,而且不确定性从500骤降到个位数。这就是卡尔曼滤波的威力——快速学习,持续优化。

三、升维思考:从1D到多维的华丽转身

现实世界很少只有一维。机器人在平面移动需要(x, y),加上速度就是(vx, vy)。这时,1D的标量就变成了向量,方差也升级为协方差矩阵。

关键升级点:

  • 状态是向量:例如 x = [位置x, 位置y, 速度x, 速度y]ᵀ
  • 不确定性是矩阵:协方差矩阵 P 不仅记录每个状态的不确定度,还记录它们之间的相关性(比如位置和速度通常是相关的)。

例如,状态向量可以像这样:

x = [x_position, 
     y_position, 
     x_velocity, 
     y_velocity]

多维卡尔曼滤波的“骨架”

多维卡尔曼滤波保持相同的“预测-更新”节奏,但换上了线性代数的装备:

多维卡尔曼滤波器中使用的标准符号:

  • x:状态估计(向量,例如位置、速度)

按回车键或点击查看完整尺寸的图片

Image
  • P:协方差矩阵(状态的不确定性)
  • F:状态转移矩阵(状态如何演变,例如,对于匀速运动模型和一个时间步长,F 如下所示:)
Image

我们可以将 x(状态估计)与 F 相乘,得到新的位置。

Image
  • u:控制输入(外部运动指令)
  • z:实际传感器测量值
  • H:测量函数(传感器与状态的关系)

如果测量无法覆盖状态中的所有特征,例如,状态空间包含 3 个元素 [x, y, z],则我们的传感器只能测量 x 和 y。此时 H 值将为

Image

将状态空间 [x, y, z] 映射到测量空间 [x_m, y_m]

  • R:测量噪声协方差
  • I:单位矩阵(用于更新步骤)

核心方程(了解即可,代码已封装)

预测步骤:

  1. x' = F @ x + u (基于模型预测新状态)
  2. P' = F @ P @ F.T (传播不确定性)

更新步骤:

  1. y = z - H @ x (创新:测量与预测的差异)
  2. S = H @ P @ H.T + R (差异的不确定性)
  3. K = P @ H.T @ inv(S) (计算卡尔曼增益)
  4. x = x + K @ y (更新状态估计)
  5. P = (I - K @ H) @ P (更新不确定性,自信度提升!)

代码实战:2D位置+速度估计

假设我们有一个移动机器人,传感器只能测量它的位置(x,y),但我们还想估计它的速度。这正是卡尔曼滤波的拿手好戏!

import numpy as np

classKalmanFilter2D:
"""一个简单的2D卡尔曼滤波器,估计位置和速度"""

def__init__(self, dt=1.0):
"""
        参数:
        dt: 时间步长(秒)
        """

        self.dt = dt

# 状态向量: [x, y, vx, vy]
        self.x = np.array([[0.], [0.], [0.], [0.]])

# 状态协方差矩阵 P: 初始时非常不确定
        self.P = np.eye(4) * 1000

# 状态转移矩阵 F: 恒定速度模型
# x_new = x + vx*dt
# y_new = y + vy*dt
# vx_new = vx
# vy_new = vy
        self.F = np.array([
            [1, 0, dt, 0],
            [0, 1, 0, dt],
            [0, 0, 1, 0],
            [0, 0, 0, 1]
        ])

# 观测矩阵 H: 我们只能观测到位置(x, y)
        self.H = np.array([
            [1, 0, 0, 0],
            [0, 1, 0, 0]
        ])

# 观测噪声协方差 R: 假设位置测量有约1单位的噪声
        self.R = np.eye(2) * 1.0

# 过程噪声协方差 Q: 模型不完美,速度可能有微小变化
        self.Q = np.eye(4) * 0.01

# 单位矩阵
        self.I = np.eye(4)

defpredict(self):
"""预测步骤:基于运动模型预测下一时刻状态"""
        self.x = self.F @ self.x
        self.P = self.F @ self.P @ self.F.T + self.Q
return self.x[:2].flatten()  # 返回预测的位置

defupdate(self, z):
"""
        更新步骤:用新的测量值修正估计

        参数:
        z: 测量向量 [meas_x, meas_y]
        """

        z = np.array(z).reshape(2, 1)

# 计算创新(测量残差)
        y = z - self.H @ self.x

# 创新协方差
        S = self.H @ self.P @ self.H.T + self.R

# 卡尔曼增益
        K = self.P @ self.H.T @ np.linalg.inv(S)

# 更新状态估计
        self.x = self.x + K @ y

# 更新协方差估计
        self.P = (self.I - K @ self.H) @ self.P

return self.x.flatten()  # 返回更新后的完整状态

# ===== 模拟一个移动物体的轨迹 =====
if __name__ == "__main__":
# 创建滤波器
    kf = KalmanFilter2D(dt=1.0)

# 模拟真实轨迹(匀速直线运动)
    true_states = []
for t in range(10):
        true_x = t * 2.0# 每秒向右移动2单位
        true_y = t * 1.0# 每秒向上移动1单位
        true_states.append([true_x, true_y, 2.0, 1.0])

# 模拟带噪声的测量
    np.random.seed(42)  # 固定随机种子,使结果可重复
    measurements = []
for true_x, true_y, _, _ in true_states:
# 添加高斯噪声到真实位置
        noisy_x = true_x + np.random.randn() * 1.5
        noisy_y = true_y + np.random.randn() * 1.5
        measurements.append([noisy_x, noisy_y])

    print("=== 2D卡尔曼滤波演示 ===\n")
    print("时间 | 真实位置 | 测量值(噪声) | 卡尔曼估计(位置) | 估计速度")
    print("-" * 70)

    estimates = []
for i in range(len(measurements)):
# 预测
        pred_pos = kf.predict()

# 更新(使用带噪声的测量)
        z = measurements[i]
        updated_state = kf.update(z)

# 记录结果
        estimates.append(updated_state)

# 打印
        true_x, true_y = true_states[i][0], true_states[i][1]
        meas_x, meas_y = z
        est_x, est_y, est_vx, est_vy = updated_state

        print(f"t={i:2d} | ({true_x:5.1f},{true_y:5.1f}) | "
f"({meas_x:5.1f},{meas_y:5.1f}) | "
f"({est_x:5.1f},{est_y:5.1f}) | "
f"v=({est_vx:4.1f},{est_vy:4.1f})")

    print("\n=== 效果分析 ===")
    print("即使测量值有明显噪声(见第3、7步),卡尔曼滤波估计的轨迹依然平滑准确。")
    print("更重要的是:我们从只能测位置的传感器,成功估计出了速度信息!")

这段代码的神奇之处在于:

我们只给了滤波器带噪声的位置测量,但它却自动“学习”并输出了相当准确的速度估计。这就是卡尔曼滤波从间接信息中推断隐藏状态的能力。

下图展示了二维卡尔曼滤波器如何估计位置和速度随时间的变化,并可视化了每一步的相关椭圆:

  • 红点:每次更新后的估计(位置,速度)状态。
  • 蓝色椭圆:协方差椭圆,显示位置和速度之间的不确定性和相关性。
  • ρ:协方差矩阵中位置和速度之间的皮尔逊相关系数。

按回车键或点击查看完整尺寸的图片

Image

随着筛选的进行:

  • 椭圆逐渐缩小,意味着测量次数越多,不确定性越小。
  • 相关系数 ρ 趋于稳定(例如,~0.84),表明人们对位置和速度之间的关系越来越有信心。

四、卡尔曼滤波在现实中的威力

卡尔曼滤波器自1960年由Rudolf Kalman提出以来,已成为工程领域的基石技术:

  1. 阿波罗登月计划:导航计算机的核心算法
  2. GPS接收机:融合卫星信号与惯性测量
  3. 自动驾驶:融合摄像头、雷达、激光雷达数据
  4. 无人机稳定:结合IMU(惯性测量单元)与视觉数据
  5. 股票价格预测:过滤市场噪声,识别趋势
  6. 生物信号处理:提取EEG/ECG中的有效成分

五、要点总结与进阶方向

记住卡尔曼滤波的三大精髓:

  1. 预测+更新:两步舞步,缺一不可
  2. 不确定性量化:用协方差矩阵坦诚表达“我不知道什么”
  3. 最优融合:根据不确定性自动调整对预测和测量的信任权重

想深入探索?这里有方向:

  • 扩展卡尔曼滤波(EKF):处理非线性系统(如机器人旋转)
  • 无迹卡尔曼滤波(UKF):更优雅的非线性处理方法
  • 粒子滤波:应对极端非高斯、多模态的分布

写在最后

卡尔曼滤波器之美,在于它用简洁的数学框架,解决了工程中普遍存在的“数据融合”难题。它不要求数据完美,只要求你诚实地量化自己的不确定性。

你在哪些项目中遇到过数据融合的挑战?尝试过用卡尔曼滤波解决吗?欢迎在评论区分享你的经验和问题!

🏴‍☠️宝藏级🏴‍☠️ 原创公众号『数据STUDIO』内容超级硬核。公众号以Python为核心语言,垂直于数据科学领域,包括可戳👉Python|MySQL|数据分析|数据可视化|机器学习与数据挖掘|爬虫等,从入门到进阶!

长按👇关注- 数据STUDIO -设为星标,干货速递ImageImage