如何计算数据流中任意元素的频率?
这个问题也是大数据场景下的一个经典问题,称为频率估计 (Frequency Estimation) 问题。
方案 1: HashMap
用一个 HashMap 记录每个元素的出现次数,每来一个元素,就把相应的计数器增 1。这个方法在大数据的场景下不可行,因为元素太多,单机内存无法存下这个巨大的 HashMap。
方案 2: Count-Min Sketch
Count-Min Sketch 算法流程:
- 选定 d 个 hash 函数,开一个 dxm 的二维整数数组作为哈希表
- 对于每个元素,分别使用 d 个 hash 函数计算相应的哈希值,并对 m 取余,然后在对应的位置上增 1,二维数组中的每个整数称为 sketch
- 要查询某个元素的频率时,只需要取出 d 个 sketch, 返回最小的那一个(其实 d 个 sketch 都是该元素的近似频率,返回任意一个都可以,该算法选择最小的那个)
这个方法的思路和 Bloom Filter 比较类似,都是用多个 hash 函数来降低冲突。
- 空间复杂度 O(dm)。Count-Min Sketch 需要开一个 dxm 大小的二位数组,所以空间复杂度是 O(dm)
- 时间复杂度 O(n)。Count-Min Sketch 只需要一遍扫描,所以时间复杂度是 O(n)
Count-Min Sketch 算法的优点是省内存,缺点是对于出现次数比较少的元素,准确性很差,因为二维数组相比于原始数据来说还是太小,hash 冲突比较严重,导致结果偏差比较大。
Count-Min Sketch 算法对于低频的元素,结果不太准确,主要是因为 hash 冲突比较严重,产生了噪音,例如当 m=20 时,有 1000 个数 hash 到这个 20 桶,平均每个桶会收到 50 个数,这 50 个数的频率重叠在一块了。Count-Mean-Min Sketch 算法做了如下改进:
- 来了一个查询,按照 Count-Min Sketch 的正常流程,取出它的 d 个 sketch
- 对于每个 hash 函数,估算出一个噪音,噪音等于该行所有整数 (除了被查询的这个元素) 的平均值
- 用该行的 sketch 减去该行的噪音,作为真正的 sketch
- 返回 d 个 sketch 的中位数
class CountMeanMinSketch {
// initialization and addition procedures as in CountMinSketch
// n is total number of added elements
long estimateFrequency(value) {
long e[] = new long[d]
for(i = 0; i < d; i++) {
sketchCounter = estimators[i][ hash(value, i) ]
noiseEstimation = (n - sketchCounter) / (m - 1)
e[i] = sketchCounter – noiseEstimator
}
return median(e)
}
}
Count-Mean-Min Sketch 算法能够显著的改善在长尾数据上的精确度。
参考:
https://soulmachine.gitbooks.io/system-design/content/cn/bigdata/frequency-estimation.html