泊松隨機數的生成演算法:數學推導和程式實現
泊松過程是一個常見的隨機過程,泊松隨機數也是一種常見的隨機數。有時候需要模擬生成泊松隨機數。Knuth曾經提出過這樣的一個演算法用來生成泊松隨機數
[1]
:
algorithm poisson random number (Knuth):
init:
Let L ← e^{−λ}, k ← 0 and p ← 1。
do:
k ← k + 1。
Generate uniform random number u in [0,1] and let p ← p × u。
while p > L。
return k − 1。
維基百科的詞條裡面沒有解釋為什麼這個演算法可以生成泊松隨機數,我在此給出證明。這其實是一個停時問題。
第一節:演算法描述
上面的這個演算法可以描述為:
第一步:給定一個引數
, 生成一系列的隨機數,這些隨機數服從
分佈,也就是這些隨機數在開區間 (0, 1) 之間均勻分佈。
第二步:求這些隨機數的乘積,當乘積小於或者等於
時,程式停止。記下此時參與求乘積的隨機數的個數。這是一個停時特徵。
第三步:程式終止時參與乘積的隨機數的個數減一服從引數為
的泊松分佈。
第二節:演算法的數學表達
為了證明這個演算法確實可以生成泊松隨機數,我們記
這就等價於
已知隨機變數
的機率密度為
,令
。
所以變數
的機率密度為
已知
所以
這是一個指數分佈。
因為隨機變數
,所以我們要計算一系列服從指數分佈的隨機變數的和。已知,對於獨立隨機變數
,它們的和
的機率密度為
這是兩個機率密度函式的卷積。做傅立葉變換,得到
的機率分佈的特徵函式是
兩個隨機變數的機率密度分佈的特徵函式的乘積。為了計算
的機率分佈,我們先要計算指數分佈的特徵函式。根據特徵函式的定義,我們有
所以變數
的機率密度的特徵函式為
。
第三節:根據機率密度的特徵函式計算所對應的機率密度
現在我們已經知道了機率密度的特徵函式,接下來我們要根據這個特徵函式計算所對應的機率密度。做傅立葉逆變換可以得到所對應的機率密度分佈:
因為變數
是一系列負數的求和,所以上面的積分中,
。
選擇如下圖所示的一個積分圍道:
計算在這個圍道上的積分:
這個積分可以分為兩部分,第一部分是沿著橫軸求積分,第二部分是沿著外面的大圓求積分。可以證明沿著大圓的積分為零,因為
當大圓半徑為無窮大的時候該積分趨近於零,這是因為當
時,
以指數速度衰減到零。
所以我們就有
根據柯西積分定理,左邊的積分為
上面式子最右邊的積分為
所以圍道積分為
最終我們得到隨機變數
所服從的機率密度函式為
這個分佈的名字叫做
分佈。顯然,根據上式,當
的時候,上面的分佈退化為指數分佈。
第四節:計算隨機變數
#FormatImgID_42#
的機率密度函式
已經知道了
服從
分佈,那麼計算
的分佈也很簡單了。已知
所以
第五節:計算
#FormatImgID_48#
的機率
我們已經知道了變數
的分佈函式,那麼就可以計算
的機率為
因為這個機率依賴於
,所以可以將這個機率重新寫作
利用分部積分,可以得到機率的遞迴關係為
因為
,所以我們有
第六節:根據對機率的兩種等價解釋得到泊松分佈
現在我們已經算出來了當我們用
個均勻分佈在
的隨機數
連乘時,所得到的乘積小於
的機率。嚴格意義上,這個連乘的結果應該記做
前面已經知道這個機率為
這是
個
連乘後乘積小於
的機率。記
為可以讓
連乘後乘積小於
所需
的個數,則有
但是我們要計算的是剛好
個
連乘後的乘積首次小於
的機率,這個機率為
這就是泊松分佈。
第七節:程式實現
我已經寫了一個程式來實現這個演算法,並且得到了測試結果。程式GitHub地址為
PrimerLi/Poisson
第八節:實驗結果
圖中顯示了
所對應的泊松分佈的機率曲線。橫軸為
,縱軸為
。 對於每一個
,我都畫了兩條線,一條線是解析解,另一條是蒙特卡洛模擬的結果。可以看出,對於不同的引數
,理論計算出來的結果和用Monte Carlo模擬出來的結果相差不大。
參考
^
Generating Poisson-distributed random variables
https://en。wikipedia。org/wiki/Poisson_distribution