流式数据中的数学统计量计算

ゝ一纸荒年。 2022-06-13 09:25 182阅读 0赞

在科技飞速发展的今天，每天都会产生大量新数据，例如银行交易记录，卫星飞行记录，网页点击信息，用户日志等。为了充分利用这些数据，我们需要对数据进行分析。在数据分析领域，很重要的一块内容是流式数据分析。流式数据，也即数据是实时到达的，无法一次性获得所有数据。通常情况下我们需要对其进行分批处理或者以滑动窗口的形式进行处理。分批处理也即每次处理的数据之间没有交集，此时需要考虑的问题是吞吐量和批处理的大小。滑动窗口计算表示处理的数据每次向前移N个单位，N小于要处理数据的长度。例如，在语音识别中，每个包处理大约25ms的音频数据，然后以步幅10ms向前移动处理下一个包的数据。语音识别就是一个典型的流式数据通过滑动窗口方式进行处理的例子。在本文中，我们关注N=1的情况，也即每次处理完一个包之后，向前移动一个单位继续处理下一个包，如下图所示。  
![图1 基于滑动窗口的流式数据处理示例][1]  
我们主要关注几个常见的数学统计量：最小（大）值、平均值和中位数。事实上，只要知道了最大值和最小值的求法，很容易计算极差；知道了平均值的求法，就可以很容易地计算方差和标准差。针对上述统计量的计算都有一个naïve算法，也即不考虑前后两个包之间数据重叠，将每个包看成独立的，对每一个包分别计算上述统计量。如果总数据长度为n，每个包的长度为k，则计算上述统计量的复杂度为O(nk)（针对给定数组求中位数的问题，存在复杂度O(k)的算法，实现方法是基于快排进行改进，网上资料很多在此不再做介绍）。我们尝试在naïve算法的基础上降低每个统计量的计算复杂度，下面开始正式的介绍。  
**1. 最小（大）值**  
这是一个经典问题，通常被称为滑动极值问题。问题描述：给定一个长度为n的数列a0,a1,...,an−1和一个整数k，求数列bi=min\{ ai,ai\+1,...,ai\+k−1\}(i=0,1,...,n−k)。  
通过使用单调队列可以在O(n)的时间内解决。单调队列维护数列的下标，队列内的元素满足：  
*设单调队列从头部开始的元素值为xi，则xi<xi\+1且axi<axi\+1。*  
简单来说单调队列就是下标对应的元素是严格递增的顺序（当然在实际应用过程中，可能不严格单调，也可能是递减的顺序）。  
考虑以ai结尾的k个元素，求bi−k\+1。假定单调递增队列中维护了ai之前的k-1个元素相关的最小值下标，为了求bi−k\+1，我们需要将ai和单调队列中元素进行比较。当队列末尾的元素j满足aj≥ai，则不断取出末尾元素，直到队列为空或者aj<ai。 ai不仅会影响bi−k\+1的计算，也会影响后续k-1个bi的计算。如果ai是这一段的最小值，则它在单调队列中就不会被删除，进而可以用O(1)的时间求单个bi。  
当删除单调队列的元素时，需要判断头部元素是否还需要。如果已经脱离计算bi的范围，则可以删除头部元素。求单个bi的值，只需要返回单调队列的头部元素即可。均摊复杂度为O(n)。求最小值的代码如下：

#define MAX_N 100000
    
    static int a[MAX_N];
    static int b[MAX_N];
    static int deque[MAX_N];
    
    void range_min(int n,int k)
    {
        int s=0,t=0;//单调队列的头和尾指针
    
        for (int i=0;i<n;i++)
        {
            //在单调队列的末尾加入i
            while (s<t&&a[deque[t-1]]>=a[i]) t--;//维护严格的单调递增队列
            deque[t++]=i;
    
            if (i-k+1>=0)
            {
                b[i-k+1]=a[deque[s]];
            }
    
            //从单调队列头部删除元素
            if (deque[s]==i-k+1)
            {
                s++;
            }
        }
    }

求滑动最大值只需要将大于等于号改为小于等于号即可，维护一个单调递减队列。通过使用单调队列，流式数据中极值计算的复杂度可以由O(nk)降为O(n)，当每个包的长度很大时，算法的优化效果会非常明显。滑动极值问题具有很广泛的应用，希望大家能知道这个优雅的解法。单调队列还有很多其他应用场景，比如解决《[leetcode之Largest Rectangle in Histogram][leetcode_Largest Rectangle in Histogram]》。此外，在一些动态规划问题中，它也可以用来降低时间复杂度。  
**2. 平均值**  
滑动平均值的计算比较容易优化，我们需要做的就是维护区间内元素的和，除以区间元素个数k即是区间平均值。当计算下一个区间的平均值时，我们先将上一个区间的和减掉上一个区间第一个元素的值，然后加上当前区间最后一个元素的值，然后除以k即是当前区间的平均值。求区间平均值的代码如下：

#define MAX_N 100000
    
    static int a[MAX_N];
    static int b[MAX_N];
    
    void range_mean(int n,int k)
    {
        int sum=0;
        for (int i=0;i<n;i++)
        {
            sum+=a[i];
            if(i-k+1>=0)
            {
                b[i-k+1]=sum/k;
                sum-=a[i-k+1];
            }
        }
    }

很明显可以看出上述代码的复杂度为O(n)。求方差可以采用类似的思路，在求和的同时也求一个平方和，之后采用方差的平方和公式即可求得方差。  
**3. 中位数**  
中位数是一个非常重要的指标，在很多应用中都会用到，但是相比前两个统计量，中位数的优化要麻烦很多。  
在介绍基于滑动窗口的中位数计算之前，我们先看一个类似的问题：也是流式数据求中位数，但是每次都求前面所有数据的中位数。该问题也很经典，出现在剑指offer一书中，具体解法可参考《[数据流中的中位数][Link 1]》。简单来说，就是构造一个最大堆和一个最小堆，最大堆的元素都小于最小堆中的元素，而且最小堆中的元素个数至多比最大堆中的元素个数多1。每次来新元素的时候，根据当前两个堆的元素个数来决定往哪个堆插入元素，在插入的同时保证上面所说的两个前提。插入复杂度是O(log n)，查询复杂度是O(1)。  
基于滑动窗口的中位数计算解法和上面的问题类似，也需要构造一个最大堆和最小堆，同时也满足上面的两个条件，区别就在于我们每次计算完一次中位数之后，都需要从堆中删除一个最老的元素。可以通过和中位数比较来确定删除哪个堆中的元素。通常的堆操作一般是插入和删除堆顶元素，在此需要实现一个函数可以删除任意位置的堆元素，同时保证堆的结构不被破坏，这不是一个困难的问题，实现和删除堆顶元素类似。如果数据是以数组形式一次给定，最老的元素可以通过访问原数组获得，如果流式数据一次只给定一个数据，我们可以通过循环队列保存最近的k个元素来获得最老的元素。代码实现可以参考博客《[找滑动窗口的中位数][Link 2]》，在此就不给出详细代码。每来一个数据都需要执行一次插入和删除，复杂度是O(log k)，所以针对流式数据的中位数问题算法复杂度是O(nlogk)，相比朴素算法也有明显地提升。

[1]: /images/20220613/b7ada05774cf4273bd86262f1befe6c6.png
[leetcode_Largest Rectangle in Histogram]: http://blog.csdn.net/yutianzuijin/article/details/52072427
[Link 1]: http://wiki.jikexueyuan.com/project/for-offer/question-sixty-four.html
[Link 2]: http://blog.csdn.net/addisionyoung/article/details/49804515