基于GWO-LSTM模型的道路交通事故量預(yù)測(cè)

孔維麟,李文棟,楊立柱,張魯玉,王慶斌

1.山東建筑大學(xué)交通工程學(xué)院,山東 濟(jì)南 250101;2.中國(guó)市政工程西北設(shè)計(jì)研究院有限公司,甘肅 蘭州 730030;3.云南省設(shè)計(jì)院集團(tuán)有限公司,云南 昆明 650118

0 引言

隨建設(shè)交通強(qiáng)國(guó)的推進(jìn)及智能交通技術(shù)的迅速發(fā)展,人們?cè)絹?lái)越重視道路交通安全。道路交通事故預(yù)測(cè)可降低交通事故潛在發(fā)生的影響程度,同時(shí)為事故管理部門(mén)提供相關(guān)信息,依據(jù)預(yù)測(cè)結(jié)果及時(shí)規(guī)劃交通方案,對(duì)提高道路交通安全水平有重要意義[1]。

傳統(tǒng)的交通事故預(yù)測(cè)模型有自回歸移動(dòng)平均(autoregressive moving average,ARMA)模型、差分自回歸移動(dòng)平均(autoregressive integrated moving average,ARIMA)模型和灰色預(yù)測(cè)模型等:謝華為[2]根據(jù)平穩(wěn)時(shí)間序列建立ARMA道路交通事故預(yù)測(cè)模型;李心儀等[3]采用ARMA模型預(yù)測(cè)浙江省道路交通事故死亡人數(shù);Li等[4]將ARMA模型與Apriori算法結(jié)合,預(yù)測(cè)交通狀態(tài)趨勢(shì);Luo等[5]改進(jìn)季節(jié)性差分自回歸移動(dòng)平均(seasonal autoregressive integrated moving average,SARMIA)模型,并與遺傳算法相結(jié)合預(yù)測(cè)交通流;胡哨剛等[6]采用灰色預(yù)測(cè)方法預(yù)測(cè)鐵路交通事故。ARMA、ARIMA模型能預(yù)測(cè)有明顯趨勢(shì)的事故數(shù)據(jù),但無(wú)法預(yù)測(cè)非平穩(wěn)性序列,灰色預(yù)測(cè)模型在處理有較少特征數(shù)據(jù)時(shí)能得到規(guī)律較強(qiáng)的生成序列,但預(yù)測(cè)波動(dòng)變化明顯的序列時(shí)誤差較大。

與傳統(tǒng)預(yù)測(cè)模型相比,深度學(xué)習(xí)預(yù)測(cè)模型有較強(qiáng)的非線性映射能力和自適應(yīng)能力,能提取數(shù)據(jù)隱藏的自然結(jié)構(gòu)和固有的抽象特征:Mo[7]基于反向傳播(back propagation,BP)神經(jīng)網(wǎng)絡(luò)改進(jìn)ARIMA模型,解決ARIMA模型不能較好擬合高維度噪聲時(shí)間序列問(wèn)題;Wang等[8]提出粒子群優(yōu)化算法(particle swarm optimization,PSO)聯(lián)合BP神經(jīng)網(wǎng)絡(luò)的PSO-BP駕駛員反應(yīng)時(shí)間預(yù)測(cè)模型;吳芮[9]將BP神經(jīng)網(wǎng)絡(luò)與遺傳算法相結(jié)合預(yù)測(cè)交通事故嚴(yán)重程度;李博豪[10]通過(guò)PSO優(yōu)化BP神經(jīng)網(wǎng)絡(luò)建立多影響因素下高速公路交通事故嚴(yán)重程度實(shí)時(shí)預(yù)測(cè)模型;Shen等[11]采用BP神經(jīng)網(wǎng)絡(luò)預(yù)測(cè)進(jìn)入高速公路休息區(qū)的車(chē)輛百分比。BP神經(jīng)網(wǎng)絡(luò)存在收斂速度慢且易陷入局部最小值等缺點(diǎn)。長(zhǎng)短期記憶(long short-term memory,LSTM)神經(jīng)網(wǎng)絡(luò)通過(guò)引入門(mén)控單元機(jī)制,有效解決處理長(zhǎng)序列數(shù)據(jù)時(shí)梯度消失和梯度爆炸的問(wèn)題:Bhandari等[12]通過(guò)試驗(yàn)證明LSTM神經(jīng)網(wǎng)絡(luò)有優(yōu)異的擬合性和較高的預(yù)測(cè)準(zhǔn)確度;Zhang等[13]將LSTM神經(jīng)網(wǎng)絡(luò)與梯度增強(qiáng)回歸樹(shù)(gradient boosting regression tree,GBRT)算法結(jié)合,建立LSTM-GBRT的交通事故預(yù)測(cè)模型;熊曉夏等[14]將貝葉斯濾波(Bayesian filter,BF)與LSTM神經(jīng)網(wǎng)絡(luò)結(jié)合,評(píng)估高速公路事故風(fēng)險(xiǎn)狀態(tài);Oliveira等[15]將多層感知器與LSTM神經(jīng)網(wǎng)絡(luò)結(jié)合,預(yù)測(cè)交通流量。網(wǎng)絡(luò)結(jié)構(gòu)參數(shù)的設(shè)置對(duì)預(yù)測(cè)性能有較大影響,以上研究多基于BP神經(jīng)網(wǎng)絡(luò)確定LSTM神經(jīng)網(wǎng)絡(luò)的網(wǎng)絡(luò)參數(shù),計(jì)算量大、復(fù)雜度高且易陷于局部最優(yōu)。

灰狼優(yōu)化(grey wolf optimizer,GWO)算法是一種模擬灰狼群種族習(xí)性的啟發(fā)式算法,具有全局遍歷性和收斂速度快的特點(diǎn)[16-18]。本文采用GWO算法對(duì)LSTM神經(jīng)網(wǎng)絡(luò)中的初始學(xué)習(xí)率、隱藏層節(jié)點(diǎn)個(gè)數(shù)、正則化系數(shù)參數(shù)進(jìn)行優(yōu)化訓(xùn)練,提出GWO-LSTM道路交通事故量預(yù)測(cè)模型,與ARMA模型、BP神經(jīng)網(wǎng)絡(luò)和LSTM神經(jīng)網(wǎng)絡(luò)預(yù)測(cè)結(jié)果對(duì)比,檢驗(yàn)其性能與預(yù)測(cè)精度,為交通事故量預(yù)測(cè)提供參考和依據(jù)。

1 GWO-LSTM

1.1 LSTM神經(jīng)網(wǎng)絡(luò)

LSTM神經(jīng)網(wǎng)絡(luò)在隱藏層構(gòu)建專門(mén)存儲(chǔ)記憶的線性自循環(huán)存儲(chǔ)單元,改進(jìn)循環(huán)神經(jīng)網(wǎng)絡(luò)(recurrent neural network,RNN)的記憶暫時(shí)存儲(chǔ)功能,解決預(yù)測(cè)長(zhǎng)時(shí)間序列數(shù)據(jù)時(shí)的梯度消失問(wèn)題,在神經(jīng)元設(shè)置遺忘門(mén)、輸入門(mén)、輸出門(mén)3個(gè)控制門(mén)和1個(gè)記憶單元[19-21]。

遺忘門(mén)負(fù)責(zé)接收前一單元的記憶ct-1并對(duì)其各部分決定保留或遺忘,通過(guò)激活函數(shù)Sigmoid將t時(shí)刻的輸入xt與前一時(shí)刻細(xì)胞狀態(tài)ht-1轉(zhuǎn)換為輸出ft,ft中的每個(gè)元素都位于[0,1],表示ct-1中保留或遺忘的信息量,0為完全舍棄,1為完全保留,公式為:

ft=σ(Wf[ht-1xt]+bf),

式中:σ為激活函數(shù)Sigmoid,Wf為遺忘門(mén)的連接權(quán)值矩陣,篩選xt和ht-1中舍棄部分的信息;bf為遺忘門(mén)的偏置矩陣。

輸入門(mén)負(fù)責(zé)將信息記錄到單元模塊中,由激活函數(shù)Sigmoid和雙曲正切激活函數(shù)tanh共同控制,輸出分別為it、ct,公式為:

it=σ(Wi[ht-1xt]+bi),

ct=ftct-1+itct′,

式中:Wi為輸入門(mén)的連接權(quán)值矩陣,篩選xt和ht-1中記憶學(xué)習(xí)部分的信息;bi為遺忘門(mén)的偏置矩陣;ct-1為記憶單元前一時(shí)刻的輸出;ct′為臨時(shí)記憶單元,ct′=tanh(Wc[ht-1xt]+bc),其中,Wc為記憶單元的權(quán)值矩陣,bc為記憶單元的偏置矩陣。

通過(guò)激活函數(shù)Sigmoid將輸出門(mén)當(dāng)前時(shí)刻的狀態(tài)ot中的每個(gè)元素轉(zhuǎn)換為[0,1]上的數(shù)值,公式為:

ot=σ(Wo[ht-1xt]+bo),

式中:Wo為輸出門(mén)的連接權(quán)值矩陣,bo為輸出門(mén)的偏置矩陣。

輸出門(mén)最終決定當(dāng)前狀態(tài)輸出哪些信息,將ot與經(jīng)函數(shù)tanh處理的ct相乘,得到輸出門(mén)的輸出ht=ottanh(ct)。

圖1 LSTM神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)示意圖

LSTM神經(jīng)網(wǎng)絡(luò)的結(jié)構(gòu)如圖1所示。

1.2 GWO算法

GWO算法模擬灰狼群的種群階級(jí)和狩獵習(xí)性,在灰狼包圍、捕捉和攻擊獵物的過(guò)程中優(yōu)化目標(biāo)。根據(jù)等級(jí)制度將灰狼從高到低分為α、β、δ和ω4個(gè)階層:α層級(jí)灰狼決策權(quán)力最高,其他個(gè)體灰狼均服從其指揮;β層級(jí)灰狼是候選決策領(lǐng)導(dǎo)者,協(xié)助α層級(jí)灰狼管理狼群;δ層級(jí)灰狼須遵從α、β層級(jí)灰狼的決策,可管理ω層級(jí)灰狼;最底層級(jí)的ω層級(jí)灰狼須服從其他層級(jí)的灰狼[22-23]?；依堑尼鳙C過(guò)程分3個(gè)階段。

1)包圍獵物?；依桥c目標(biāo)獵物的距離

D=|C⊙Xp(k)-X(k)|,

式中:C為包圍獵物時(shí)的協(xié)同向量,k為迭代次數(shù),Xp(k)為獵物的位置,X(k)為灰狼的位置,⊙為矩陣的哈達(dá)瑪積。

灰狼位置更新

X(k+1)=Xp(k)-A⊙D,

式中:A為位置更新時(shí)的協(xié)同向量。

2)捕捉獵物。灰狼包圍獵物后,設(shè)α、β、δ層級(jí)灰狼距獵物距離依次增大,灰狼的位置隨迭代不斷更新,帶領(lǐng)其他灰狼不斷向獵物逼近,不同灰狼間的距離計(jì)算公式為:

Di=|Cj⊙Xi(k)-X(k)|,i=α,β,δ,j=1,2,3,

式中:Di分別為α、β、δ層級(jí)灰狼與其他灰狼的距離,Xi(k)分別為α、β、δ層級(jí)灰狼的位置,Cj為捕捉獵物時(shí)協(xié)同向量。

迭代k+1次后,灰狼的位置

式中Aj為灰狼對(duì)獵物的攻擊系數(shù)。

3)攻擊獵物。當(dāng)獵物無(wú)法移動(dòng)逃走時(shí),灰狼發(fā)起攻擊。逼近獵物的過(guò)程中,A在[-a,a]中變化,其中a為收斂因子,a在迭代過(guò)程中由2線性減小為0。GWO算法根據(jù)A確定搜尋范圍:|A|<1時(shí),灰狼群逼近獵物,進(jìn)行局部搜尋;|A|>1時(shí),灰狼群遠(yuǎn)離獵物,進(jìn)行全局搜尋;|A|=1時(shí),灰狼群停止搜尋。

2 交通事故預(yù)測(cè)模型

2.1 模型參數(shù)優(yōu)化

道路交通事故預(yù)測(cè)過(guò)程包括數(shù)據(jù)處理、模型參數(shù)優(yōu)化、模型訓(xùn)練、模型預(yù)測(cè)與評(píng)價(jià)。通過(guò)GWO算法對(duì)LSTM神經(jīng)網(wǎng)絡(luò)中的參數(shù)進(jìn)行調(diào)整優(yōu)化,獲取最優(yōu)參數(shù),提高模型預(yù)測(cè)準(zhǔn)確度。

確定初始學(xué)習(xí)率、隱藏層節(jié)點(diǎn)數(shù)、GWO算法的迭代次數(shù)及灰狼群數(shù)等參數(shù)。采用隨機(jī)分布函數(shù)初始化灰狼群,確定其位置向量并分解為L(zhǎng)STM神經(jīng)網(wǎng)絡(luò)的權(quán)值矩陣和偏置矩陣。以均方根誤差為適應(yīng)度函數(shù),將參數(shù)輸入LSTM神經(jīng)網(wǎng)絡(luò),計(jì)算輸出及灰狼個(gè)體的適應(yīng)度,將適應(yīng)度最佳的3個(gè)位置依次標(biāo)記為Xα、Xβ、Xδ,隨迭代次數(shù)的增加,此3個(gè)位置將被適應(yīng)度更優(yōu)的個(gè)體位置更新替換,直至迭代結(jié)束。

Xα即為L(zhǎng)STM神經(jīng)網(wǎng)絡(luò)參數(shù)的最優(yōu)解,將測(cè)試集數(shù)據(jù)通過(guò)輸入層→隱藏層→輸出層正向傳遞輸入LSTM神經(jīng)網(wǎng)絡(luò)進(jìn)行預(yù)測(cè),采用均方誤差(mean squared error,MSE)損失函數(shù)計(jì)算偏差,將其反向傳遞,神經(jīng)元中相關(guān)參數(shù)得到更新并采用Adam優(yōu)化函數(shù)調(diào)整網(wǎng)絡(luò)梯度。迭代更新直至滿足收斂條件終止后,輸出GWO-LSTM道路交通事故預(yù)測(cè)模型參數(shù),進(jìn)行預(yù)測(cè)與評(píng)估。

GWO-LSTM交通事故預(yù)測(cè)模型框架如圖2所示,圖中kmax為最大迭代次數(shù),x1～xn、y1～yn分別為第1～n個(gè)網(wǎng)絡(luò)單元輸入的樣本數(shù)據(jù)和訓(xùn)練輸出的數(shù)據(jù),L1～Ln為第1～n個(gè)LSTM神經(jīng)網(wǎng)絡(luò)單元。

圖2 GWO-LSTM交通事故預(yù)測(cè)模型框架

2.2 試驗(yàn)數(shù)據(jù)與預(yù)處理

采用2000—2019年美國(guó)道路交通致死事故數(shù)據(jù)為樣本數(shù)據(jù)源,樣本數(shù)據(jù)記錄了每起致死事故的完整信息,主要字段包括:STATE、ST_CASE、MONTH、DAY_WEEK、HOURNAME、ROUTE、FATALS,對(duì)應(yīng)的內(nèi)容分別為:事故發(fā)生所在州編號(hào)、事故唯一標(biāo)識(shí)編號(hào)、事故發(fā)生月份編號(hào)、事故發(fā)生日期、事故發(fā)生時(shí)刻、路線編碼、死亡人數(shù)。

采用平均值修正法修正樣本數(shù)據(jù)中的異常值,時(shí)間粒度事故統(tǒng)計(jì)樣本數(shù)據(jù)的特征受各因素影響而上下波動(dòng),為平滑時(shí)間序列,采用離差標(biāo)準(zhǔn)化歸一化法將數(shù)據(jù)線性轉(zhuǎn)換到[0,1],縮放后值域長(zhǎng)度較小,模型收斂速度和預(yù)測(cè)準(zhǔn)確度提高。歸一化后的樣本數(shù)據(jù)

xi′=(xi-xmin)/(xmax-xmin),

式中:xi為原始樣本數(shù)據(jù),xmax、xmin分別為樣本數(shù)據(jù)的最大值、最小值。

2.3 時(shí)間粒度及滑動(dòng)窗口劃分

時(shí)間粒度即劃分?jǐn)?shù)據(jù)單元的度量單位,如年、月、周、時(shí)等,每個(gè)絕對(duì)時(shí)間區(qū)間由有限個(gè)時(shí)間粒度單元組成,受樣本數(shù)據(jù)特征影響,適當(dāng)?shù)臅r(shí)間粒度可反映事故特性[24-25]。根據(jù)道路交通事故歷史數(shù)據(jù)序列的特征和周期確定以月粒度、周粒度、時(shí)粒度3種時(shí)間粒度統(tǒng)計(jì)樣本數(shù)據(jù)。

圖3 滑動(dòng)窗口示意圖

采用滑動(dòng)窗口預(yù)測(cè)法預(yù)測(cè)不同時(shí)間粒度的事故樣本數(shù)據(jù),將特征值按時(shí)間序列排列,根據(jù)設(shè)定的時(shí)間粒度劃分為不同窗口單元,窗口以1個(gè)單位步長(zhǎng)向前滑動(dòng),反復(fù)向前,得到特征值的變化趨勢(shì),原理如圖3所示。由圖3可知:n1～n5包含4個(gè)窗口單元,由時(shí)間粒度決定滑動(dòng)窗口的長(zhǎng)度,即ni-1與ni間的時(shí)間。以月粒度、周粒度、時(shí)粒度3種時(shí)間粒度劃分?jǐn)?shù)據(jù)后,可確保特征值向后滾動(dòng)預(yù)測(cè)時(shí)的滑動(dòng)窗口長(zhǎng)度,也可保留數(shù)據(jù)間的相關(guān)性。

2.4 模型性能評(píng)價(jià)指標(biāo)

選取平均絕對(duì)百分比誤差(mean absolute percentage error,MAPE)EMAPE、均方根誤差(root mean square error,RMSE)ERMSE為評(píng)價(jià)GWO-LSTM交通事故預(yù)測(cè)模型性能的指標(biāo),公式分別為:

式中:ei′為預(yù)測(cè)值,ei為實(shí)際值,n為樣本容量。

3 模型訓(xùn)練與評(píng)價(jià)

試驗(yàn)平臺(tái)采用Windows10 64位操作系統(tǒng),試驗(yàn)環(huán)境為AMD Ryzen 7 5800HS Creator Edition 3.20 GHz,RAM為16 GB,采用軟件MATLAB R2019a編程。

3.1 模型參數(shù)設(shè)置

在GWO-LSTM交通事故預(yù)測(cè)模型訓(xùn)練過(guò)程中輸入事故數(shù)據(jù)字段為:事故發(fā)生月份編號(hào)、事故發(fā)生日期、事故發(fā)生時(shí)刻、死亡人數(shù),結(jié)構(gòu)由輸入層、1層隱藏層和輸出層構(gòu)成,輸出維數(shù)為1;在GWO算法中設(shè)置灰狼群個(gè)體數(shù)為30,迭代次數(shù)為20,初始學(xué)習(xí)率搜索范圍為[0.001,0.500],隱藏層節(jié)點(diǎn)數(shù)的搜索范圍GWO算法參數(shù)尋優(yōu)時(shí)適應(yīng)度函數(shù)隨迭代次數(shù)的變化曲線如圖4所示。由圖4可知:月粒度劃分的事故樣本數(shù)據(jù)的適應(yīng)度函數(shù)收斂速度最快,適應(yīng)度函數(shù)值最大,周粒度劃分的事故樣本數(shù)據(jù)的適應(yīng)度函數(shù)值最小。GWO-LSTM模型訓(xùn)練過(guò)程中的ERMSE隨訓(xùn)練次數(shù)的變化曲線如圖5所示。由圖5可知周粒度劃分的事故樣本數(shù)據(jù)的ERMSE最小。

表1 不同時(shí)間粒度下LSTM神經(jīng)網(wǎng)絡(luò)的最優(yōu)參數(shù)

為[1,30],正則化系數(shù)的搜索范圍為[0.001,0.050]。為達(dá)到收斂并防止出現(xiàn)過(guò)擬合現(xiàn)象,將LSTM神經(jīng)網(wǎng)絡(luò)的訓(xùn)練次數(shù)設(shè)為200,學(xué)習(xí)率下降因子為0.1,激活函數(shù)為Relu,采用優(yōu)化函數(shù)Adam調(diào)節(jié)梯度下降。采用GWO算法對(duì)LSTM神經(jīng)網(wǎng)絡(luò)3種時(shí)間粒度的樣本數(shù)據(jù)進(jìn)行參數(shù)優(yōu)化,得到不同時(shí)間粒度下的最優(yōu)參數(shù)如表1所示。

圖4 適應(yīng)度函數(shù)隨迭代次數(shù)的變化曲線 圖5 ERMSE隨訓(xùn)練次數(shù)的變化曲線

將事故樣本數(shù)據(jù)序列按8：2劃分為訓(xùn)練集和測(cè)試集,即將預(yù)處理后的樣本序列的前80%作為訓(xùn)練集訓(xùn)練優(yōu)化模型并調(diào)試參數(shù),后20%作為測(cè)試集。取不同時(shí)間粒度滑動(dòng)窗口進(jìn)行采樣,采用訓(xùn)練好的GWO-LSTM模型進(jìn)行事故預(yù)測(cè)。

3.2 預(yù)測(cè)結(jié)果

將ARMA模型、BP神經(jīng)網(wǎng)絡(luò)和LSTM神經(jīng)網(wǎng)絡(luò)作對(duì)比,驗(yàn)證GWO-LSTM交通事故預(yù)測(cè)模型的性能。GWO-LSTM模型、ARMA模型、BP神經(jīng)網(wǎng)絡(luò)和LSTM神經(jīng)網(wǎng)絡(luò)對(duì)3種時(shí)間粒度下交通致死事故量的預(yù)測(cè)結(jié)果如圖6所示。

a)月粒度 b)周粒度 c)時(shí)粒度圖6 不同模型在不同時(shí)間粒度下交通致死事故量的預(yù)測(cè)結(jié)果

由圖6可知:相比月粒度預(yù)測(cè)結(jié)果,4種模型對(duì)周粒度和時(shí)粒度的交通致死事故量預(yù)測(cè)結(jié)果更準(zhǔn)確,這與時(shí)間粒度劃分更細(xì),模型數(shù)據(jù)起伏較大,周期性特征明顯,更易被模型捕捉有關(guān)。ARMA模型和BP神經(jīng)網(wǎng)絡(luò)預(yù)測(cè)數(shù)據(jù)的峰值趨勢(shì)時(shí)與實(shí)際數(shù)據(jù)存在一定偏差,LSTM神經(jīng)網(wǎng)絡(luò)和GWO-LSTM模型捕捉事故數(shù)據(jù)峰值的能力較強(qiáng)。

根據(jù)不同模型在不同時(shí)間粒度下交通致死事故量的預(yù)測(cè)結(jié)果,計(jì)算ERMSE和EMAPE,評(píng)價(jià)對(duì)比模型的預(yù)測(cè)準(zhǔn)確度,結(jié)果如表2所示。

表2 不同模型的性能評(píng)價(jià)指標(biāo)

由表2可知:不同時(shí)間粒度下GWO-LSTM模型的ERMSE和EMAPE均最小,模型預(yù)測(cè)準(zhǔn)確度最高,LSTM神經(jīng)網(wǎng)絡(luò)次之,BP神經(jīng)網(wǎng)絡(luò)和ARMA模型預(yù)測(cè)準(zhǔn)確度較低。ARMA模型捕捉平穩(wěn)時(shí)間序列數(shù)據(jù)間線性關(guān)系的能力較強(qiáng),但捕捉非線性時(shí)間序列數(shù)據(jù)峰值的靈敏性較差;BP神經(jīng)網(wǎng)絡(luò)采用誤差逆?zhèn)鞑ニ惴ㄟM(jìn)行網(wǎng)絡(luò)訓(xùn)練,模型預(yù)測(cè)時(shí)忽略輸入與輸出間的滯后和延遲性,導(dǎo)致預(yù)測(cè)誤差較大;相比于傳統(tǒng)模型,GWO-LSTM模型和LSTM神經(jīng)網(wǎng)絡(luò)考慮到預(yù)測(cè)的滯后效應(yīng),能更好解決長(zhǎng)期依賴問(wèn)題,捕捉數(shù)據(jù)信息間的規(guī)律和關(guān)系的能力更佳,在處理序列數(shù)據(jù)和非線性問(wèn)題中優(yōu)勢(shì)更顯著,與LSTM網(wǎng)絡(luò)參數(shù)調(diào)節(jié)相比,GWO-LSTM優(yōu)化參數(shù)尋優(yōu)方式,預(yù)測(cè)準(zhǔn)確度較好。

4 結(jié)論

采用GWO算法替代標(biāo)準(zhǔn)LSTM神經(jīng)網(wǎng)絡(luò)中的BP神經(jīng)網(wǎng)絡(luò)算法,優(yōu)化初始學(xué)習(xí)率、隱藏層節(jié)點(diǎn)數(shù)、正則化系數(shù)等參數(shù),提出基于GWO-LSTM的道路交通事故量預(yù)測(cè)模型。采用月粒度、周粒度和時(shí)粒度3種時(shí)間粒度劃分交通事故樣本數(shù)據(jù),分別采用傳統(tǒng)ARMA模型、BP神經(jīng)網(wǎng)絡(luò)、標(biāo)準(zhǔn)LSTM神經(jīng)網(wǎng)絡(luò)和GWO-LSTM模型預(yù)測(cè)交通事故量,并與實(shí)際數(shù)據(jù)對(duì)比。結(jié)果表明:在3種時(shí)間粒度下,與ARMA模型、BP神經(jīng)網(wǎng)絡(luò)和LSTM神經(jīng)網(wǎng)絡(luò)相比,GWO-LSTM模型的全局遍歷性和收斂性速度最快,預(yù)測(cè)結(jié)果的平均絕對(duì)百分比誤差和均方根誤差均較小,預(yù)測(cè)結(jié)果更準(zhǔn)確。

道路交通事故的發(fā)生具有隨機(jī)性等特點(diǎn)且受多種環(huán)境因素影響,參數(shù)優(yōu)化和訓(xùn)練中缺乏表征環(huán)境因素的相關(guān)數(shù)據(jù),未來(lái)考慮在環(huán)境多因素條件下預(yù)測(cè)交通事故量,進(jìn)一步提高模型的預(yù)測(cè)性能。