蓄水池抽样,首先有它的应用和它的神奇之处,其次这个也是机器学习领域面试的热门试题。
问题一(引子):流式数据(Streaming Data),数据长度为n但不知道,如何从中等概率随机选择一个数据?
解法:我们以概率1选择第一个数据,以1/2的概率选择第二个数据,以此类推,以1/m的概率选择第m个对象(如果后面某一数据一旦选中,替换掉以前选中的数据)。当所有数据流过时,每个对象具有相同的被选中的概率1/n。
证明:第m个数据最终被选中的概率P = 选择m的概率*其后的所有对象都不被选中的概率,即
问题二(蓄水池抽样):流式数据(Streaming Data),数据长度为n但不知道,如何从中等概率随机选择k(k < n)个数据?
解法:先选中前k个数据(当作一个蓄水池),然后以k/(k+1)的概率选择第k+1个数据,以k/(k+2)的概率选择第k+2个数据,以此类推以k/m的概率选择第m个数据,一旦后面有某个数据被选中,则随机替换蓄水池中的一个数据。最终每个数据被选中的概率均为k/n。
证明:第m个数据选中的概率 = 选择第m个数据的概率其后所有数据(不被选中的概率+被选中的概率不替换第m个数据的概率)
公式推导
网友评论