专栏文章

【日报】浅谈随机分布及C++当中的实现

科技·工程参与者 48已保存评论 50

文章操作

快速查看文章及其快照的属性,并进行相关操作。

当前评论
50 条
当前快照
1 份
快照标识符
@mhz5supt
此快照首次捕获于
2025/11/15 01:56
4 个月前
此快照最后确认于
2025/11/29 05:24
3 个月前
查看原文

前言

C++11\text{C++11} 里新增加了一个功能强大的库 random\verb!random!,用于弥补 C\text{C} 里面的随机数生成器的不足。然而,除了提供比起 C\text{C} 来说更加可靠的随机数生成器,random\verb!random! 库里同样提供了很多类,可以利用一个均匀随机位生成器来获得符合特定分布的随机数。如何用算法实现这些东西,就是本文的重点。
为了缩短篇幅,本文并不会去探讨这些特定分布的来龙去脉,也并不会给出一些相关推导。这部分内容并不是文章的重点,读者可以通过百度等方式自行得到。我们关心的是 random\verb!random! 库里采取了什么样的算法。然而,由于作者能力有限(并且网上关于这方面的文章比较少),有些实现确实查阅不到资料。我只能尽量提供可能具有帮助的参考资料。

概念

分布

假如我们有个骰子,六个面上依次标为了 1,2,61,2,\cdots 6。现在抛一次骰子(每个面朝上的概率均等),设抛出来的点数为 XX。容易发现,XX 的值是随机的,并且对于骰子上的六个值,XX 都有一定概率取得。像这样的只可能取有限个或至多可列个值的 XX,称作离散型随机变量。对于刚刚的例子,很容易绘制出 XX 在不同取值时的概率:
X123456P161616161616\def\arraystectch{1.5} \begin{array}{c|c|c|c|c|c|c} X & 1 & 2 & 3 & 4 & 5 & 6 \cr\hline P & \frac{1}{6} & \frac{1}{6} & \frac{1}{6} & \frac{1}{6} & \frac{1}{6} & \frac{1}{6} \end{array}
当然了,XX 的可能取值可能有无数个。比如 P(X=k)=(12)k.k=1,2,P(X=k)=\left(\frac{1}{2}\right)^k.k=1,2,\cdots。这样的 XX 同样是离散型随机变量。
但是换个例子,比如说 XX 是在线段上取的任意一个点的数值,或者是在一个圆上取一个点到圆心的距离,你会发现貌似 XX 取到任意一个值的概率都是 00。但是我们可以描述 XX 取一个集合内的值的概率。比如在数轴上的线段 [5,10][5,10] 内任取一个点,那么它在 [6,8][6,8] 内的概率就是 35\frac{3}{5}。对于这种的 XX,我们称之为非离散型随机变量
但是我们怎么描述多次观测一个非离散型随机变量得到的结果呢?这时候就要用到积分相关知识了。根据上文,我们可以描述一个非离散型随机变量在一定区间内的概率。首先定义累积分布函数 F(x)F(x),表示 XX 的取值在 [0,x][0,x] 内的概率;接着可以定义出 F(x)F(x) 在某点初的导数(假设 F(x)F(x) 在此处可导。本文将会涉及到的所有累积分布函数都在定义域内可导),称作概率密度函数 f(x)=dF(x)dxf(x)=\frac{\mathrm dF(x)}{\mathrm dx}
容易发现,f(x)f(x) 就是 F(x)F(x) 的导数,f(x)f(x) 积分后加上常数 CC 可得 F(x)F(x)。两者是可以互相转化的。下面是个例子:

均匀随机位生成器

接着简要提及一下标准库内的相关概念。
均匀随机位生成器是返回无符号整数值的函数对象,可能结果范围中的每个值都(理想情况)拥有等概率。
从中我们可以得到,均匀随机位生成器应该是一个可以在指定范围内生成均匀随机的随机数的生成器。随机数分布内部需要有一个均匀随机数生成器,这将保证它生成出来的随机数确实是符合相关分布的。除此之外,标准里面还要求了一个均匀随机数生成器 GG 应当拥有以下几个表达式:
  • G::value_type\verb!G::value_type!:均匀随机数生成器返回值的类型。它应该是一个无符号整型。
  • G::min()\verb!G::min()!:返回 GG 可以生成的随机数的最小值。最小值的类型应当为 TT
  • G::max()\verb!G::max()!:返回 GG 可以生成的随机数的最大值。最大值的类型应当为 TT
  • G()\verb!G()!:返回一个值域在 [G::min(),G::max()][\verb!G::min()!,\verb!G::max()!] 里面的类型为 TT 的均匀的随机数。
random\verb!random! 库里已经内置了一些随机数引擎(线性同余算法、梅森缠绕器算法、延迟斐波那契算法),再配合上一些随机数引擎适配器实现好了一些现成的均匀随机位生成器(比如大名鼎鼎的 mt19937\text{mt19937} 等等)。由于这部分内容不是我们的重点,因此不再赘述。有兴趣的读者可以参阅维基百科,以及 cpp reference\textsf{cpp reference} 上面的相关叙述。
那么下面就是我们的主角。当前在 random\verb!random! 库里被提供的那些可生成服从特定分布的随机数的类。
均匀分布
  • uniform_int_distribution \verb!uniform_int_distribution !:生成在固定范围上均匀的整数
  • uniform_real_distribution\verb!uniform_real_distribution!:生成在固定范围上均匀的实数
伯努利分布
  • bernoulli_distribution\verb!bernoulli_distribution!:生成符合伯努利分布的 bool\text{bool} 类型的值。
  • geometric_distribution\verb!geometric_distribution!:生成符合几何分布的整数值。
  • binomial_distribution \verb!binomial_distribution !:生成符合二项分布的整数值。
  • negative_binomial_distribution\verb!negative_binomial_distribution!:生成一个符合负二项分布的整数值。
泊松分布
  • poisson_distribution    \verb!poisson_distribution !:生成符合泊松分布的整数值。
  • exponential_distribution\verb!exponential_distribution!:生成符合指数分布的实数值。
  • gamma_distribution      \verb!gamma_distribution !:生成符合 Γ\Gamma 分布的实数值。
  • weibull_distribution    \verb!weibull_distribution !:生成符合威布尔分布的实数值。
  • extreme_value_distribution\verb!extreme_value_distribution!:生成符合极值分布的实数值。
正态分布
  • normal_distribution   \verb!normal_distribution !:生成符合标准正态(高斯)分布上的实数值。
  • lognormal_distribution\verb!lognormal_distribution!:生成符合对数正态分布的实数值。
  • cauchy_distribution   \verb!cauchy_distribution !:生成符合柯西分布的实数值。
  • fisher_f_distribution \verb!fisher_f_distribution !:生成符合费舍尔分布的实数值。
  • student_t_distribution\verb!student_t_distribution!:生成符合学生分布上的实数值。
  • chi_squared_distribution\verb!chi_squared_distribution!:生成符合 χ2\chi^2 分布的实数值。

从均匀分布开始

考虑一个很简单的问题。你现在手头有一个均匀随机位生成器(不妨设为 GG),可以随机生成 1616 位的无符号整数。现在你要用它实现一个函数 rand(a,b)\operatorname{rand}(a,b),使得它能够返回值在 [a,b][a,b] 内的均匀的随机整数。应该怎么做?
很多人会用一个这样的方式来实现 rand(l,r)\operatorname{rand}(l,r)
CPP
int rand(int a, int b){
    return G() % (b - a + 1) + l;
}
嗯,这个 rand\operatorname{rand} 函数看上去蛮对的。它把 GG 生成的随机数映射到了 [0,ba)[0,b-a) 里面,然后把它加上 aa,最终得到了值在 [a,b][a,b] 内的随机数。但是只要仔细想一想,就会发现它里面存在的漏洞。首先是原理上的错误:GG 映射到 [0,ba)[0,b-a) 是不均匀的。比如说,当 a=0,b=60000a=0,b=60000 时,有两个数字映射到了 23332333233323336233362333),但是仅有一个数字映射到了 5000050000。这样就会导致最终生成 23332333 的概率是生成 5000050000 的两倍。其次还有实现上的漏洞:bab-a 有可能超过 int\text{int} 类型的最大值,发生不可预期的错误。
让我们看看 random\verb!random! 库是如何解决这些问题的。由于 uniform_int_distribution\verb!uniform_int_distribution! 库使用了模板,我们不妨设它的返回值为整型 Int\text{Int}
  • 与刚刚的做法类似,它把计算 [a,b][a,b] 内的随机数的任务转换为了计算 [0,ba)[0,b-a) 内的随机数的任务,再将结果加上 aa。但它使用了一些 template\text{template} 的小技巧。通过获取 Int\text{Int}无符号形式 uInt\text{uInt},来防止发生溢出。这基于一个这样的补码的结论: unsigned(u)={uu02n+uu<0unsigned(u)unsigned(v)={uvu0,v0u(2n+v)u0,v<0(2n+u)(2n+v)u<0,v<0\begin{aligned}\operatorname{unsigned}(u)&= \begin{cases} u & u\ge 0\cr 2^n+u & u<0\cr \end{cases}\cr \operatorname{unsigned}(u)-\operatorname{unsigned}(v)&=\begin{cases} u-v & u\ge0,v\ge0 \cr u-(2^n+v)& u\ge 0,v<0\cr (2^n+u)-(2^n+v) &u<0,v<0 \cr \end{cases} \end{aligned} 其中 uu 是一个 nn 位二进制有符号整数。负数 uu 的补码形式相当于先对它的所有二进制位取反,然后再加 11。取反操作本身就相当于让 2n12^n-1 减去 uu,再加上 11 得到的就是 2nu2^n-u
然后就是对随机数生成器的最大最小值之差 rangeg=GmaxGmin\mathrm{range}_{\mathrm{g}}=G_{\max}-G_{\min} 和现在需要生成的 [a,b][a,b] 的最大最小值之差 rangeu=ba\mathrm{range}_{\mathrm{u}}=b-a 进行讨论。注意:rangeg\mathrm{range}_{\mathrm{g}}rangeu\mathrm{range}_{\mathrm{u}} 比两者的值域少 11,这是为了防止程序中发生溢出。
  • 第一种情况:rangeg>rangeu\mathrm{range}_{\mathrm{g}}>\mathrm{range}_{\mathrm{u}}。这表明,我们需要缩小 GG 生成的随机数的值域,以至于达到 uu 的值域。设 erangeu=rangeu+1\mathrm{erange}_{\mathrm{u}}=\mathrm{range}_{\mathrm{u}}+1,显然不会发生溢出。
    不妨设 rangeg=rerangeu+s\mathrm{range}_{\mathrm{g}}=r\cdot \mathrm{erange}_{\mathrm{u}}+s。断言:只要不断让 GG 生成减去了 GminG_{\min} 的随机数 pp,直到 p[0,rerangeu)p\in[0,r\cdot\mathrm{erange}_{\mathrm{u}}),最终取 p÷rp\div r 即可。显然,最终得到的 pp 肯定是在 [0,rerangeu)[0,r\cdot \mathrm{erange}_{\mathrm{u}}) 里均匀的,通过除法映射到 [0,erangeu)[0,\mathrm{erange}_{\mathrm{u}}) 里也是均匀的,于是可以证明答案的正确性;由于 smin(erangeu1,rangegerangeu)12rangegs\le \min(\mathrm{erange}_{\mathrm{u}}-1,\mathrm{range}_{\mathrm{g}}-\mathrm{erange}_{\mathrm{u}})\le\left\lfloor\frac{1}{2}\mathrm{range}_{\mathrm{g}}\right\rfloor,因此生成的 pp[rrangeu,rangeg][r\cdot \mathrm{range}_{\mathrm{u}},\mathrm{range}_{\mathrm g}] 的概率不超过 12\frac{1}{2}。因此期望只要进行两次生成,就能得到符合要求的随机数。时间复杂度为 O(1)\mathcal O(1)
  • 第二种情况:rangeg<rangeu\mathrm{range}_{\mathrm{g}}<\mathrm{range}_{\mathrm{u}}。这表明,我们需要放大 GG 生成的随机数的值域,以至于达到 uu 的值域。设 erangeg=rangeg+1\mathrm{erange}_{\mathrm{g}}=\mathrm{range}_{\mathrm{g}}+1,显然不会发生溢出。
    不妨设 rangeu=rerangeg+s\mathrm{range}_{\mathrm{u}}=r\cdot \mathrm{erange}_{\mathrm{g}}+s。容易发现,r>0r>0。现在我们随机生成一组 (r[0,r],s[0,rangeg](r'\in[0,r],s'\in[0,\mathrm{range}_{\mathrm{g}}](生成 rr' 可以通过递归处理),那么求出来的 p=rerangeg+sp=r'\cdot\mathrm{erange}_{\mathrm{g}}+s' 肯定是均匀随机地分布在 [0,(r+1)erangeg)[0,(r+1)\cdot \mathrm{erange}_{\mathrm{g}}) 里面的。如何使得 pp[0,rangeu)[0,\mathrm{range}_{\mathrm{u}}) 内呢?用一个最暴力的做法:若 prangeu{p\ge\mathrm{range}_{\mathrm{u}}},那就重新生成,直到 p[0,rangeu)p\in[0,\mathrm{range}_{\mathrm{u}})。答案的正确性比较显然,现在考虑时间复杂度的正确性。
    不妨设 r[2k1,2k)r\in[2^{k-1},2^k)。我们需要重新生成 [0,r][0,r] 内的整数的次数期望为 rangeu1rangeu1rangeg\frac{\mathrm{range}_{\mathrm{u}}-1}{\mathrm{range}_{\mathrm{u}}-1-\mathrm{range}_{\mathrm{g}}},也就是 2k2k1\frac{2^k}{2^k-1} 次。接着我们要做的就是把这些东西全部乘在一起:
    (((21211+1)22221+1))2k2k1+1\left(\cdots \left(\left(\frac{2^1}{2^1-1}+1\right)\cdot \frac{2^2}{2^2-1}+1\right)\cdots\right)\cdot\frac{2^k}{2^k-1}+1
    这个式子最终会无限趋近 kk 加上一个常数,所以总复杂度为 O(lograngegrangeu)\mathcal O(\log_{\mathrm{range}_{\mathrm{g}}}\mathrm{range}_{\mathrm{u}})
  • 第三种情况:rangeg=rangeu\mathrm{range}_{\mathrm{g}}=\mathrm{range}_{\mathrm{u}}。这种情况就很简单了,直接返回 GG 生成的随机数即可,时间复杂度是 O(1)\mathcal O(1)
这里给出一个模拟标准库写法的实现:
CPP
#include<iostream>
#include<random>

typedef unsigned int       u32;
typedef unsigned long long u64;

std::mt19937 MT;
u32 rand4(){ return MT() & 15; }
template <typename T>
T rand(T l, T r){
    typedef typename std::make_unsigned<T>::type U;
    U u = r - l, g = 15, w = 0;
    if(g == u) return l + rand4();
    if(g >  u){
        U eu = u + 1, s = g / eu, t = s * eu;
        do{
            w = rand4();
        }while(w >= t);
        w /= s;
    } else {
        U eg = g + 1, s = u / eg;
        do{
            w = rand((U)0, s) * eg + rand4();
        }while(w >= u);
    }
    return l + w;
}
int C[10];
int main(){
    for(int i = 0; i < 100000; ++i)
        ++C[rand(1, 10) - 1];
    for(int i = 0; i < 10; ++ i)
        std::cout << i + 1 << ":" << C[i] << std::endl;
    return 0;
}

讲完了比较复杂的随机生成 [a,b][a,b] 内整数的实现,生成 [a,b)[a,b) 内浮点数的实现则要简单很多。由于浮点数没有很麻烦的精度溢出的问题,又因为是实数,所以处理起来相当简单:
rand(a,b)=a+(ba)rand0()\operatorname{rand}(a,b)=a+(b-a)\cdot \operatorname{rand}_0()
其中 rand0()\operatorname{rand}_0() 可以生成一个均匀随机的在 [0,1)[0,1) 内的实数。
random\verb!random! 库里还定义了一个函数 generate_canonical\verb!generate_canonical!,它能使用一个均匀随机位生成器 GG 生成 [0,1)[0,1) 内的实数。一种容易想到的生成 [0,1)[0,1) 内随机实数的方法是用 GG 生成一个随机无符号整数 pp,然后让它除去 GmaxGmin+1G_{\max}-G_{\min}+1。看起来很对,但这种做法生成的浮点数熵值不够(举个较为极端的例子:GmaxGmin+1=10G_{\max}-G_{\min}+1=10,那么这种做法生成的随机数只能是 010,110,210,,910\frac{0}{10},\frac{1}{10},\frac{2}{10},\cdots,\frac{9}{10})。解决办法也很简单。首先计算 erange=GmaxGmin+1\mathrm{erange}=G_{\max}-G_{\min}+1(转换成浮点数,就不用担心 +1+1 后的溢出问题了),然后计算出该浮点数类型精度部分所用的二进制位的个数 mm(换言之,就是表示科学计数法指数部分需要使用的二进制位的个数)。显然,每次调用 GG 可以让 log2erange\log_2\mathrm{erange} 位具有较高的熵值。那么我们调用 m/log2erangem/\log_2\mathrm{erange} 次即可。实现较为容易,不再赘述。
random\verb!random! 库里,generate_canonical\verb!generate_canonical! 还被包装进了一个内置的类 __detail::_Adaptor\verb!__detail::_Adaptor! 里,再基于此实现了 uniform_real_distribution\verb!uniform_real_distribution!
我们称值在 [l,r)[l,r) 内均匀随机的随机变量 XX 服从均匀分布。记作 XUnif(l,r)X\sim \mathrm{Unif}(l,r)

复杂的分布

柯西分布

柯西分布的概率密度函数和累计密度函数,分别为:
f(x;x0,γ)=1π(γ(xx0)2+γ2)F(x;x0,γ)=1πarctan(xx0γ)+12\begin{aligned} f(x;x_0,\gamma)&=\frac{1}{\pi}\left(\frac{\gamma}{(x-x_0)^2+\gamma^2}\right)\cr F(x;x_0,\gamma)&=\frac{1}{\pi}\arctan\left(\frac{x-x_0}{\gamma}\right)+\frac{1}{2} \end{aligned}
看上去这些式子非常复杂。但实际上我们可以用逆变换法来解决这样一类问题:
已知一个累积分布函数 F(x)F(x),现在需要计算 ζ\zeta,使得 ζ\zeta 的累积分布函数为 F(x)F(x)。那么我们设 UUnif(0,1)U\sim\mathrm{Unif}(0,1)F(x)F(x) 的逆函数为 F1(x)F^{-1}(x)。大胆猜测 ζ=F1(U)\zeta=F^{-1}(U)。此时, P(ζ<k)=P(F1(U)<k)=P(U<F(k))=F(k)P(\zeta<k)=P(F^{-1}(U)<k)=P(U<F(k))=F(k) 这证明了按照这种方法求出来的 ζ\zeta 确实是符合 FF 的。
UU 的一个观测值 uu 容易获得,因此只要求出 F1(u)F^{-1}(u),就能得到一个 ζ\zeta 的观测值。当然,有时候 F1(u)F^{-1}(u) 比较容易获得,有时候又不那么容易得到。对于后者是一些算法针对的重点。
对于柯西分布的累计密度函数,我们很容易求出 F1(x;x0,γ)F^{-1}(x;x_0,\gamma)
y=1πarctan(xx0γ)+12π(y12)=arctan(xx0γ)x=x0+γtan((y12)π)F1(x;x0,γ)=x0+γtan((x12)π)\begin{aligned} y&=\frac{1}{\pi}\arctan\left(\frac{x-x_0}{\gamma}\right)+\frac{1}{2} \cr \pi\left(y-\frac{1}{2}\right)&=\arctan\left(\frac{x-x_0}{\gamma}\right) \cr x&=x_0+\gamma\cdot\tan\left(\left(y-\frac{1}{2}\right)\pi\right) \cr F^{-1}(x;x_0,\gamma)&=x_0+\gamma\cdot \tan\left(\left(x-\frac{1}{2}\right)\pi\right) \end{aligned}
于是,只要在 Unif(0.5,0.5)\mathrm{Unif}(-0.5,0.5) 里观测一个点 xx,接着带入即可。

指数分布

我们用 Exp\mathrm{Exp} 表示指数分布。Exp(λ)\mathrm{Exp}(\lambda) 的概率密度函数被定义为 f(x)=λeλxf(x)=\lambda e^{-\lambda x}。容易发现,
F(x)=0xf(x)dx=1eλxF1(x)=1λln(1x)\begin{aligned} F(x)&=\int_{0}^xf(x)\mathrm dx=1-e^{-\lambda x} \cr F^{-1}(x)&=-\frac{1}{\lambda}\ln(1-x) \end{aligned}
于是只要观测一个 Unif(0,1)\mathrm{Unif}(0,1) 的随机变量 xx,然后计算 F1(x)F^{-1}(x) 即可。

韦布尔分布

韦布尔分布的概率密度函数和累计密度函数,分别为:
f(x;λ,k)={kλ(xλ)k1e(x/λ)k,x00,x<0F(x;λ,k)={1e(x/λ)k,x00,x<0\begin{aligned} f(x;\lambda,k)&= \begin{cases} \frac{k}{\lambda}(\frac{x}{\lambda})^{k-1}e^{-(x/\lambda)^k}&,x\ge 0\cr 0 &, x<0 \end{cases} \cr F(x;\lambda,k)&= \begin{cases} 1-e^{-(x/\lambda)^k} &,x\ge 0 \cr 0 & ,x<0 \end{cases} \end{aligned}
同样地,我们可以计算出累计密度函数的逆函数:
F1(x;λ,k)=λln(1y)kF^{-1}(x;\lambda,k)=\lambda\cdot\sqrt[k]{-\ln(1-y)}
于是只要观测一个 Unif(0,1)\mathrm{Unif}(0,1) 的随机变量 xx,然后计算 F1(x)F^{-1}(x) 即可。

极值分布

极值分布(这里是指其中的极小值分布)的概率密度函数和累计密度函数,分别为:
f(x)=1σexμσexμσF(x)=1eexμσ\begin{aligned} f(x)&=\frac{1}{\sigma}e^{\frac{x-\mu}{\sigma}-e^{\frac{x-\mu}{\sigma}}}\cr F(x)&=1-e^{-e^{\frac{x-\mu}{\sigma}}} \end{aligned}
可能有点不大清楚。用 exp(x)\operatorname {exp}(x) 替代 exe^x
f(x)=1σexp(xμσexp(xμσ))F(x)=1exp(exp(xμσ))\def\exp#1{\operatorname{exp}\left(#1\right)} \begin{aligned} f(x)&=\frac{1}{\sigma}\exp{\frac{x-\mu}{\sigma}-\exp{\frac{x-\mu}{\sigma}}}\cr F(x)&=1-\exp{-\exp{\frac{x-\mu}{\sigma}}} \end{aligned}
同样地,我们可以求出它的逆函数:
F1(x)=μ+σln(ln(1x))F^{-1}(x)=\mu+\sigma \ln(-\ln(1-x))
于是只要观测一个 Unif(0,1)\mathrm{Unif}(0,1) 的随机变量 xx,然后计算 F1(x)F^{-1}(x) 即可。

伯努利分布

伯努利分布是一个非常简单的离散型概率分布。
P(X=0)=pP(X=1)=1p\begin{aligned} P(X=0)&=p \cr P(X=1)&=1-p \end{aligned}
那么我们观测一个 Unif(0,1)\mathrm{Unif}(0,1) 的随机变量 xx。如果 x<px<p,那就返回 00;否则返回 11

几何分布

假如随机变量 XX 服从几何分布 G(p)G(p),那么:
P(X=k)=p(1p)k.k=0,1,2,P(X=k)=p\cdot (1-p)^k.k=0,1,2,\cdots
考虑使用类似伯努利分布的方法,把每个 kk 的概率头尾相接画在数轴上。此时数轴上形成了一条长度为 11 的线段 [0,1)[0,1)。那么生成 00 的概率对应 [0,p)[0,p),生成 11 的概率对应 [p,p+p(1p))[p,p+p(1-p)),以此类推。在这条线段上任取一个点 x0x_0,然后判断它所在的线段对应哪个 kk。我们设:
g(k)=i=0kP(X=i)=p(1p)k+11p=1(1p)k+1g(k)=\sum_{i=0}^kP(X=i)=p\cdot \frac{(1-p)^{k+1}-1}{-p}=1-(1-p)^{k+1}
根据最后一个式子,我们把 g(x)g(x) 的定义域从自然数集扩域到非负实数集。然后考虑从 1(1p)k+11-(1-p)^{k+1} 反推出 kk 的值。
y=1(1p)k+1(1p)k+1=1yk+1=ln(1y)ln(1p)k=ln(1y)ln(1p)1\begin{aligned} y&=1-(1-p)^{k+1} \cr (1-p)^{k+1}&=1-y \cr k+1&=\frac{\ln(1-y)}{\ln(1-p)} \cr k&=\frac{\ln(1-y)}{\ln(1-p)}-1 \end{aligned}
然后上取整,就能得到 x0x_0 所在点对应线段的值。这其实仍然是逆变换法的应用。

到这里为止,我们计算的都是可以顺利计算出 F1(x)F^{-1}(x) 的分布。但有时候,并不那么容易计算出 F1(x)F^{-1}(x)

正态分布

对于一个符合正态分布的随机变量 ζ\zeta 而言,若 ζN(μ,σ2)\zeta\sim\mathcal N(\mu,\sigma^2),设 ξ=ζμσ\xi=\frac{\zeta-\mu}{\sigma},那么就有 ζN(0,1)\zeta\sim\mathcal N(0,1)。因此,只要能够生成服从标准正态分布的随机数,就可以得到服从 N(μ,σ2)\mathcal N(\mu,\sigma^2) 的随机数。
现在假设有两个变量 X,YX,Y,它们都服从标准正态分布。于是有:
XN(0,1)f(x)=12πex22YN(0,1)f(y)=12πey22X\sim\mathcal N(0,1)\Rightarrow f(x)=\frac{1}{\sqrt{2\pi}}e^{-\frac{x^2}{2}}\\ Y\sim\mathcal N(0,1)\Rightarrow f(y)=\frac{1}{\sqrt{2\pi}}e^{-\frac{y^2}{2}}\\
(x,y)(x,y) 对应了平面直角坐标系上的一个点。于是,
f(x,y)=f(x)f(y)=12πex2212πey22=12πex2+y22f(x,y)=f(x)f(y)=\frac{1}{\sqrt{2\pi}}e^{-\frac{x^2}{2}}\cdot \frac{1}{\sqrt{2\pi}}e^{-\frac{y^2}{2}}=\frac{1}{2\pi}e^{-\frac{x^2+y^2}{2}}
现在我们用极角坐标的方式描述点 (x,y)(x,y)。也就是用角度 θ\theta 和该点到原点的距离 rr 描述 (x,y)(x,y)。那么:
r=x2+y2r2=x2+y2θ=arctan(x÷y)\begin{aligned} r&=\sqrt{x^2+y^2} \Rightarrow r^2=x^2+y^2 \cr \theta&=\arctan(x\div y) \end{aligned}
并且我们硬点一下 rrθ\theta 对应的随机变量服从的分布:
r2Exp(12)θUnif(0,2π)=2πUnif(0,1)\begin{aligned} r^2 &\sim \mathrm{Exp}\left(\frac{1}{2}\right) \cr \theta & \sim \mathrm{Unif}(0,2\pi)=2\pi\mathrm{Unif}(0,1) \end{aligned}
UUnif(0,1)U\sim \mathrm{Unif}(0,1),于是 R2=2ln(1U)R^2=-2\ln(1-U)。这时候我们得到了一个非常简单地做法:
  • 观测服从 Unif(0,1)\mathrm{Unif}(0,1) 的随机变量,得到实数 u1,u2u_1,u_2
  • 计算 r=2ln(1u1),θ=2πu2r=\sqrt{-2\ln(1-u_1)},\theta =2\pi u_2。根据 rrθ\theta 作为极角坐标 (r,θ)(r,\theta)
  • (r,θ)(r,\theta) 转换为平面直角坐标系上的坐标 (x,y)=(rcosθ,rsinθ)(x,y)=(r\cos\theta,r\sin \theta)。根据上面的反推,可以得到 x,yx,y 分别服从标准正态分布 N(0,1)\mathcal N(0,1)
这种做法有个效率上的缺陷,就是计算 sin\sincos\cos 带来的时间开销。考虑换个角度,设 XUnif(1,1),YUnif(1,1)X\sim \mathrm{Unif}(-1,1),Y\sim\mathrm{Unif}(-1,1),并分别进行观测,得到两个实数 x,yx,y。如果 (x,y)(x,y) 不在单位圆里(或者在单位圆的圆心),那就直接排除并重新生成,直到符合要求。容易发现,此时的 (x,y)(x,y) 是在单位圆里均匀随机生成的一个点,设 t=x2+y2t=x^2+y^2。考察由 (x,y)(x,y) 转换到极坐标的形式为 (θ,r)(\theta',r'),那么显然 θ\theta' 对应的随机变量服从 2πUnif(0,1)2\pi\mathrm{Unif}(0,1)。此外,tt 对应的随机变量服从 Unif(0,1)\mathrm{Unif}(0,1)。又容易发现,
cosθ=xrsinθ=yr\begin{aligned} \cos \theta&=\frac{x}{r} \cr \sin \theta&=\frac{y}{r} \end{aligned}
因此可以得到最终的式子:
x0=x2lntty0=y2lntt\begin{aligned} x_0&=x\sqrt{\frac{-2\ln t}{t}}\cr y_0&=y\sqrt{\frac{-2\ln t}{t}} \end{aligned}
x0,y0x_0,y_0 都是标准正态分布的观测值。优化后的做法称为 Marsaglia polar method\textsf{Marsaglia polar method}。这里给出一个简单的实现:
CPP
#include<iostream>
#include<cmath>
#include<random>
std::mt19937 MT;
double uni(){
    return std::generate_canonical<double, 10>(MT);
}
double nod(){
    static double rem; static bool flag = false;
    if(flag){
        flag = false; return rem;
    }
    double x, y, R;
    do{
        x = 2.0 * uni() - 1.0;
        y = 2.0 * uni() - 1.0;
        R = x * x + y * y;
    }while(R == 0.0 || R > 1.0);
    R = sqrt(-2.0 * std::log(R) / R);
    rem = R * y, flag = true; return R * x;
}
const int MAXN = 20 + 3;
int C[MAXN], s, h = 20;
int main(){
    for(int i = 0;i < 1000000; ++i){
        double x = nod();
        if(x < -2.5 || x > 2.5) continue;
        ++C[(int) round(x / 0.4) + 10];
    }
    for(int i = 0;i < 20; ++i)
        s = std::max(s, C[i]);
    for(int i = h - 1;i >= 0; --i){
        for(int j = 0;j < 20; ++j)
            std::cout <<(1.0 * C[j] / s * h >= 0.5 + i ? '*' : ' ');
        std::cout << '\n';
    }
    return 0;
}

对数正态分布

根据定义,若 XX 服从对数正态分布,那么:
lnXN(μ,σ2)\ln X\sim \mathcal N(\mu,\sigma^2)
因此得出一个服从 N(μ,σ2)\mathcal N(\mu,\sigma^2) 的随机变量的观测值 x0x_0,然后返回 ex0e^{x_0} 即可。

伽马分布

在讲伽马分布前,我们讲一下拒绝采样法。
已知 XX 的概率密度函数 f(x)f(x)。考虑画出 f(x)f(x) 的图像。容易发现,我们要做的实际就是在这条曲线下方的图形里均匀随机地任取一个点,然后取它的横坐标,这就是 XX 的一个观测值。
不过,有些函数图像下面的图形是不大容易随机取点的。考虑采取一种妥协的方法。对于概率密度函数 f(x)f(x),我们找到另外一个随机变量 YY,它的概率密度函数 g(x)g(x),累计密度函数为 G(x)G(x)。同时,存在一个常数 cc,满足对于任意的 xx 都有 cg(x)f(x)cg(x)\ge f(x)。现在我们观测 YY 得到一个实数 yy。但要注意的是,g(y)f(x)g(y)\ge f(x),因此我们不一定接受这个 yy。接受它的概率为: f(y)cg(y)\frac{f(y)}{cg(y)} 如果被拒绝了,那就重新选定一个 yy。直到接受了它,记 xx 为这个最终结果。按照这种做法最终获取的 xx 就是 XX 的一个合法的观测值。为了提高拒绝采样的效率,g(x)g(x) 应当是容易计算的。并且 f(x)cg(x)\frac{f(x)}{cg(x)} 的值越接近 11 越好
Marsaglia\textsf{Marsaglia} 写的一篇期刊论文里,提出了一种 exact-approximation\text{exact-approximation} 方法。大致如下:
有时候我们无法快速计算 F1(x)F^{-1}(x) 的值。但是我们同样可以折中:选择一个 g(x)g(x),它非常接近 F1(x)F^{-1}(x)。但是 g(U),UUnif(0,1)g(U),U\sim\textrm{Unif}(0,1) 并不等于我们所需要的 XX。但是可以考虑不使用 UU,然是使用另外一个随机变量 YY 来替代。那么,YY 的概率密度函数应当为: h(x)=f(g(x))g(x)h(x)=f(g(x))g'(x) 这是容易证明的。因为, H(x)=0xf(g(t))g(t)dt=F(g(x))H1(x)=g1(F1(x))\begin{aligned}H(x)&=\int_0^x f(g(t))g'(t)\mathrm dt=F(g(x))\cr H^{-1}(x)&=g^{-1}(F^{-1}(x))\end{aligned} 因而对于一个随机变量 UUnif(0,1)U\sim \mathrm{Unif}(0,1),有: g(H1(U))=g(g1(F1(U)))=F1(U)g(H^{-1}(U))=g(g^{-1}(F^{-1}(U)))=F^{-1}(U)
讲完了前置芝士,现在来探讨一下伽马分布 Γ(α,β)\Gamma(\alpha,\beta)。它的概率密度函数如下:
f(x;α,β)=βαΓ(α)xα1eβxf(x;\alpha,\beta)=\frac{\beta^{\alpha}}{\Gamma(\alpha)}\cdot x^{\alpha-1}e^{-\beta x}
有个性质:若 XΓ(α,β),X0Γ(α,1)X\sim \Gamma(\alpha,\beta),X_0\sim \Gamma(\alpha,1),那么 X=βX0X=\beta X_0。于是只要能计算出 X0X_0 的一个观测值,就能求出 XX 的一个观测值了。因此下文中,β=1\beta=1
然后我们着重讲一下 A Simple Method for Generating Gamma Variables\textrm{A Simple Method for Generating Gamma Variables}。它构造了这样一个函数:
h(x)=d(1+cx)3d=α13,c=19dh(x)=d(1+cx)^3 \\ d=\alpha-\frac{1}{3},c=\frac{1}{\sqrt{9d}}
根据 exact-approximation\text{exact-approximation} 方法,如果我们能找到一个 XX,它的概率密度函数为
g0(x)=h(x)α1eh(x)h(x)Γ(α)g_0(x)=\frac{h(x)^{\alpha-1}\cdot e^{-h(x)}\cdot h'(x)}{\Gamma(\alpha)}
并且设法得到了 XX 的观测值 xx,就能通过 Y=h(X)Y=h(X) 得到服从 Γ(α,1)\Gamma(\alpha,1)YY 的一个观测值 h(x)h(x) 啦。现在考虑使用拒绝采样法观测 XX。大胆设 g(x)=ln(g0(x))g(x)=\ln(g_0(x)),这么做的好处是 g(x)g(x) 的所有常数项都可以忽略,因为 g(x)g(x) 上的一个常数因子在 eg(x)e^{g(x)} 里都会变成一个乘积,而拒绝采样法是可以忽略掉所乘上的数字的。
lnh(x)α1eh(x)h(x)Γ(α)(α1)lnh(x)h(x)+lnh(x)(α1)(3ln(1+cx))+2cln(1+cx)h(x)(α1/3)(3ln(1+cx))h(x)d(ln(1+cx)3)h(x)+d\begin{aligned} &\ln \frac{h(x)^{\alpha-1}\cdot e^{-h(x)}\cdot h'(x)}{\Gamma(\alpha)}\cr \Rightarrow& (\alpha-1)\ln h(x)-h(x)+\ln h'(x)\cr \Rightarrow& (\alpha-1)(3\ln(1+cx))+2c\ln(1+cx)-h(x)\cr \Rightarrow& (\alpha-1/3)(3\ln(1+cx))-h(x)\cr \Rightarrow& d(\ln(1+cx)^3)-h(x)+d \end{aligned}
g(x)=d(ln(1+cx)3)h(x)+dg(x)=d(\ln(1+cx)^3)-h(x)+d,设 g0(x)=peg(x)g_0(x)=pe^{g(x)}。同时,标准正态分布的概率密度函数也可以设为 qe0.5x2qe^{-0.5x^2}。拒绝采样法的那个常数设为 p/qp/q。那么接受一个用正态分布生成的 x0x_0,概率为:
peg(x0)pqqe0.5x2=eg(x0)e0.5x2\frac{pe^{g(x_0)}}{\frac{p}{q}\cdot qe^{-0.5x^2}}=\frac{e^{g(x_0)}}{e^{-0.5x^2}}
然后经过一番申必操作(论文里计算了一下 eg(x)e^{g(x)}e0.5x2e^{-0.5x^2} 的比例),可以发现 eg(x)e^{g(x)}e0.5x2e^{-0.5x^2} 非常接近!同时,可以证明 eg(x)e0.5x2e^{g(x)}\le e^{-0.5x^2}。因而使用拒绝采样法可以得到一个效率很高的算法。基本框架如下:
  • 根据标准正态分布生成随机数 xx
  • 根据 Unif(0,1)\mathrm{Unif}(0,1) 生成随机数 uu,用来判断是否要接受这个 xx。也就是计算 uueg(x)+0.5x2e^{g(x)+0.5x^2} 哪个更大。若前者更大,则拒绝;否则接受。
实际实现的时候,比较 uueg(x)+0.5x2e^{g(x)+0.5x^2} 可以先取对数变成 lnu\ln ug(x)+0.5x2g(x)+0.5x^2。把后面的式子拆开来,并且设 v=(1+cx)3v=(1+cx)^3,于是等价于比较 lnu\ln udlnvdv+dd\ln v-dv+d。尽管这种做法已经很快,它仍需要取两次对数。但有一种方法是,找到一个适合的函数 s(x)s(x),满足 s(x)eg(x)e0.5x2s(x)\le e^{g(x)}\le e^{-0.5x^2}。那么只要 ue0.5x2s(x)ue^{-0.5x^2}\le s(x) 时,就有 us(x)e0.5x2eg(x)e0.5x2u\le \frac{s(x)}{e^{-0.5x^2}}\le \frac{e^{g(x)}}{e^{-0.5x^2}},就接受 xx。 论文里提到的 s(x)s(x) 为:
s(x)=(10.0331x4)e0.5x2s(x)=(1-0.0331x^4)e^{-0.5x^2}
指数部分是会被消去的。剩下来的部分容易计算。于是得到最终算法:
  • 计算服从标准正态分布和 Unif(0,1)\text{Unif}(0,1) 的两个随机数 x,ux,u。同时计算 vv,若 v0v\le 0 就重新生成。
  • 如果 u(10.0331x4)u\le (1-0.0331x^4),那就直接返回 xx。否则:
  • 如果 lnudlnvdv+d\ln u\le d\ln v-dv+d,那就返回 xx。否则重新生成 x,ux,u,直到符合要求。
然而并没有完全解决。这种做法在 α<1\alpha<1 时效率较低,因此需要利用 α+1\alpha+1 生成的随机数得到 α\alpha 的。将生成的随机数乘上 uα+1\sqrt[\alpha+1]{u} 即可。

卡方分布

卡方分布 χ2(n)\chi^2(n) 的概率密度函数如下:
f(x;n)=xn/21ex/22n/2Γ(n/2)f(x;n)=\frac{x^{n/2-1}e^{-x/2}}{2^{n/2}\cdot \Gamma(n/2)}
容易发现,这就是 α=n2,β=12\alpha=\frac{n}{2},\beta =\frac{1}{2} 的伽马分布概率密度函数。于是直接套伽马分布就行了。

学生 t 分布

根据定义,假设 XN(0,1),Yχ2(n)X\sim\mathcal N(0,1),Y\sim \chi^2(n),那么服从学生 tt 分布 t(n)t(n)ZZ 为:
Z=XY/nZ=\frac{X}{\sqrt{Y/n}}
于是观测 X,YX,Y 得到 x,yx,y,然后返回 z=xy/nz=\frac{x}{\sqrt{y/n}} 即可。

费舍 F 分布

根据定义,假设 Xχ2(n),Yχ2(m)X\sim\chi^2(n),Y\sim \chi^2(m),那么服从费舍 FF 分布 F(n,m)F(n,m)ZZ 为:
Z=Xn÷Ym=XmYnZ=\frac{X}{n}\div \frac{Y}{m}=\frac{X\cdot m}{Y\cdot n}
同样地,观测 X,YX,Y 得到 x,yx,y,然后返回 xmyn\frac{xm}{yn}

二项分布

服从二项分布 B(n,p)B(n,p) 的随机变量 XX,满足:
P(X=k)=(nk)pk(1p)nkP(X=k)=\binom{n}{k}p^k(1-p)^{n-k}
random\verb!random! 库里使用的是拒绝采样法(和另外一个解决小范围二项分布的做法)。但是我没能获取到相关论文以及里面的常数的来源(库里实现非常复杂),因此没办法讲解如何生成服从二项分布的随机数了。库里提供了这样的注释:
CPP
/**
* A rejection algorithm when t * p >= 8 and a simple waiting time
* method - the second in the referenced book - otherwise.
* NB: The former is available only if _GLIBCXX_USE_C99_MATH_TR1
* is defined.
*
* Reference:
* Devroye, L. Non-Uniform Random Variates Generation. Springer-Verlag,
* New York, 1986, Ch. X, Sect. 4 (+ Errata!).
*/

泊松分布

服从泊松分布 B(λ)B(\lambda) 的随机变量 XX,满足:
P(X=k)=λkk!eλP(X=k)=\frac{\lambda^k}{k!}e^{-\lambda}
random\verb!random! 库里,使用的同样是拒绝采样法。然而我同样没能获取到相关函数及常数来源(与二项分布疑似出自同一本书),我就只能给出库中的注释了:
CPP
/**
* A rejection algorithm when mean >= 12 and a simple method based
* upon the multiplication of uniform random variates otherwise.
* NB: The former is available only if _GLIBCXX_USE_C99_MATH_TR1
* is defined.
*
* Reference:
* Devroye, L. Non-Uniform Random Variates Generation. Springer-Verlag,
* New York, 1986, Ch. X, Sects. 3.3 & 3.4 (+ Errata!).
*/

负二项分布

服从负二项分布的随机变量 XX,满足:
P(X=k)=(k+r1r1)(1p)kprP(X=k)=\left(\begin{array}{c}k+r-1\\r-1\end{array}\right)\cdot(1-p)^{k}p^{r}
标准库里写到了这样一条注释:
CPP
// This is Leger's algorithm, also in Devroye, Ch. X, Example 1.5.
同样因为过于生僻了,这里不做讲解。

采样分布

在经历了一大堆数学公式的洗礼后,本文将会以较为简单的采样分布作为结尾。

离散分布

给定 nn 个实数 w0,w1,wn1w_{0},w_1,\cdots w_{n-1},其中 wiw_i 表示数字 ii 的权重。定义服从该离散分布的随机变量 XX,满足:
P(X=i)=wij=0n1wj.i=0,1,(n1)P(X=i)=\frac{w_i}{\sum_{j=0}^{n-1} w_j}.i=0,1,\cdots (n-1)
实现起来比较简单。计算 S=w0+w1++wn1S=w_0+w_1+\cdots+w_{n-1},然后让 wiw_i 除去 SS。此时每个数字的权重和就是 11 了。对其求前缀和。在 Unif(0,1)\textrm{Unif}(0,1) 内生成一个随机整数 uu,然后在前缀和数组里二分出 uu 的位置并返回即可。

分段常值分布

给定 nn 个实数 b0,b1,bn1b_0,b_1,\cdots b_{n-1} 作为分段函数的端点,同时给出 n1n-1 个实数 w0,w1,w2wn2w_0,w_1,w_2\cdots w_{n-2},其中 wiw_i 表示线段 [bi,bi+1)[b_i,b_{i+1}) 的权重。该权重决定了最终生成的随机数应该在哪个线段上。概率密度函数为:
f(x)=wij=0n2wi(bi+1bi).i=0,1,(n1)f(x)=\frac{w_i}{\sum_{j=0}^{n-2}w_i(b_{i+1}-b_i)}.i=0,1,\cdots (n-1)
更直观地,我们给出一个例子:
同样地,我们计算 S=j=0n2wjS=\sum_{j=0}^{n-2} w_j 并将每个 wiw_i 除去 SS,接着求 wiw_i 的前缀和。同样地生成一个Unif(0,1)\textrm{Unif}(0,1) 内的随机整数 uu,并使用二分查找找到它所对应的那条线段。与离散分布不同的是,我们根据该条线段 [wx,wx+1)[w_x,w_{x+1})uu 所在的具体位置,就可以得到随机出来的随机数,应该为:
x=bx+uwxwx+1wx(bx+1bx)x=b_x+\frac{u-w_x}{w_{x+1}-w_x}\cdot(b_{x+1}-b_x)

分段线性分布

分段线性分布的定义要略微复杂一点。给定 nn 个实数 b0,b1,bn1b_0,b_1,\cdots b_{n-1} 作为分段函数的端点,同时给出 nn 个实数 w0,w1,w2wn1w_0,w_1,w_2\cdots w_{n-1},作为每个端点的权重。权重决定了该分布的概率密度函数在 bib_i 处的取值。它的概率密度函数并不算复杂,与分段常数分布有些许类似。以下是一个例子:
首先得计算出每个端点处的概率密度函数取值,使得整个图形的面积(粉红色部分)为 11。那么就得首先计算出原来的面积:
S=i=0n1(bi+1bi)(wi+1+wi)2S=\sum_{i=0}^{n-1}\frac{(b_{i+1}-b_i)\cdot (w_{i+1}+w_i)}{2}
然后让每个 wiw_i 除以 SS,这样得到每个端点处的概率密度函数取值。接着我们就能计算出概率密度函数 F(x)F(x)bib_i 处的取值了。与前两个分布一样,我们在 Unif(0,1)\mathrm{Unif}(0,1) 里随机出一个 uu,二分 uu 应该在哪一段里面(设为 [bx,bx+1)[b_x,b_{x+1})),接着计算出使得 F(x)=uF(x)=uxx

参考文献

评论

50 条评论,欢迎与作者交流。

正在加载评论...