Learning Lane Graph Representations for Motion Forecasting

• • 1.ActorNet
  • 2.MapNet
  • 3.FusionNet
  • 4.Prediction Header
  • 5.模型參數(shù)學(xué)習(xí)

自動(dòng)駕駛論文閱讀筆記2
Uber–ECCV2020–論文文章代碼

模型的作用：Motion Forecasting （這個(gè)motion都包括什么呢？）
方法：模型由四個(gè)模塊組成–ActorNet、MapNet、FusionNet、Header

1.ActorNet

作用–編碼actor軌跡特征

輸入：每一條actor的軌跡（3xT的輸入向量）(actor包括所有運(yùn)動(dòng)者？)

操作：1D CNN + FPN(特征金字塔)（多尺度不斷上采樣融合特征）

輸出：該軌跡的特征向量(多少維度呢？128維度么)

軌跡表示–位移差
$$\{\Delta p_{?(T?1)},...,\Delta p_{?1},\Delta p_0\}$$

$$\Delta p_t=(x_t,y_t)?(x_{t?1},y_{t?1})$$

長度為T，不足T的padding 0，下面2xT的位移向量 拼接 1xT的padding標(biāo)志向量(1-表示該位置的位移是padding的)

ActorNet–3組1D卷積，每組包括兩個(gè)殘差塊；特征金字塔融合多尺度的特征（卷積 卷下去，上采樣，再和對(duì)應(yīng)的尺度疊加）

輸出-- The output of ActorNet is a temporal feature map,whose element at t = 0 is used as the actor feature.（不是很清楚這個(gè)時(shí)間序列是怎么回事？）

2.MapNet

作用–編碼地圖特征，主要是對(duì)車道的編碼

構(gòu)建lane graph(結(jié)點(diǎn)與連接的定義)

利用LaneGCN獲取lane圖的特征

2.1 構(gòu)建lane Graph

lane node --車道中心線的一段，結(jié)點(diǎn)位置為兩端結(jié)點(diǎn)坐標(biāo)的均值。==一條車？所有？==道線上的眾多結(jié)點(diǎn)可以表示為向量$V\in {\mathbb{R}}^{N\times 2}$（結(jié)點(diǎn)的二維特征表示）

車道結(jié)點(diǎn)特征向量帶有4個(gè)連接矩陣$\{A_i{\}}_{i\in \{pre,suc,left,right\}}$,$A_i\in {\mathbb{R}}^{N\times N}$, $A_{i,jk}=1$表示結(jié)點(diǎn)$j$ 存在一個(gè)類型$i$的鄰居結(jié)點(diǎn)$k$。

LaneConv Operator
a). lane node 特征$x_i$–編碼形狀(長度方向)，位置(空間坐標(biāo))信息，經(jīng)過全聯(lián)接層處理后，輸出lane node特征$x_i$；構(gòu)成結(jié)點(diǎn)特征矩陣$X$。
$$x_i=ML{P}_{shape}(v_i^{end}?v_i^{start})+ML{P}_{loc}(v_i)$$
b).LaneConv 為了獲得lane graph大規(guī)模拓?fù)湫畔?#xff08;四個(gè)$A_i$矩陣都用上）
$$Y=X{W}_0+\sum _{i\in \{pre,suc,left,right\}}{A}_{i}X{W}_i$$
c).Dilated LaneConv 為了讓模型獲得車道線方向的長時(shí)依賴關(guān)系(速度快的物體位移大，只用到$A_{pre}$和$A_{suc}$)
$$Y=X{W}_0+A_{pre}^{k}X{W}_{pre,k}+A_{suc}^{k}X{W}_{suc,k}$$
d). $LaneConv(k_1,...,k_c)$=Dilated LaneConv + LaneConv - $k_c$為第$c$個(gè)dilation 尺寸
$$Y=X{W}_0+\sum _{i\in \{left,right\}}{A}_{i}X{W}_i+\sum _{c=1}^C(A_{pre}^{k_c}X{W}_{pre,k_c}+A_{suc}^{k_c}X{W}_{suc,k_c})$$

2.2 LaneGCN操作
$LaneConv(k_1,...,k_c)$ + Linear Layer 構(gòu)成殘差塊， 4個(gè)殘差塊堆疊，構(gòu)成LaneGCN。

注意點(diǎn)：
每個(gè)結(jié)點(diǎn)與鄰居結(jié)點(diǎn)的連接關(guān)系，指明前后左右的結(jié)點(diǎn)。
a). 結(jié)點(diǎn)A的前驅(qū)結(jié)點(diǎn)、后續(xù)結(jié)點(diǎn)：同一條車道線上，能夠到達(dá)A的結(jié)點(diǎn)和A能夠到達(dá)的結(jié)點(diǎn)
b).結(jié)點(diǎn)A的左鄰居結(jié)點(diǎn)、右鄰居結(jié)點(diǎn)：鄰居車道線上空間距離$l_2$最近的結(jié)點(diǎn)

對(duì)車道結(jié)點(diǎn)圖不采用廣泛使用的圖卷積$L=D^{?\frac{1}{2}}(I+A)D^{?\frac{1}{2}}(1)$更新隱狀態(tài)的原因：
a). 不知道結(jié)點(diǎn)特征會(huì)保存何種車道信息（可解釋性不強(qiáng)？）
b). 該拉普拉斯矩陣操作沒法獲取結(jié)點(diǎn)的連接關(guān)系信息(沒有對(duì)$A_i$矩陣的操作，(1)式中的A為結(jié)點(diǎn)的鄰接矩陣，與$A_i$內(nèi)涵不同)
c). 無法解決長時(shí)依賴性的問題
為了解決以上三個(gè)不足點(diǎn)，作者提出了LaneConv Operator操作。

Dilated LaneConv 參考了dilated convolution，說是能夠沿著車道線傳遞k步的信息。

3.FusionNet

利用空間注意力機(jī)制(spatial attention)（用于構(gòu)建A2L, L2A, A2A） 和LaneGCN(主要用來構(gòu)建L2L網(wǎng)絡(luò)的)融合actor結(jié)點(diǎn)和lane結(jié)點(diǎn)的信息，
四個(gè)信息融合模塊：

A2L-將實(shí)時(shí)交通信息傳達(dá)給lane node

L2L-依據(jù)實(shí)時(shí)交通信息，更新lane node 的特征

L2A-將更新后lane node 的特征返回給actor

A2A-解決actor之間的相互作用，并且actor特征給motion預(yù)測網(wǎng)絡(luò)

L2L 網(wǎng)絡(luò)結(jié)構(gòu)-結(jié)點(diǎn)特征更新時(shí)的LaneGAN結(jié)構(gòu)一致
A2L, L2A, A2A網(wǎng)絡(luò)結(jié)構(gòu)一致，采用空間注意力機(jī)制構(gòu)成殘差塊。
$$y_i=x_i{W}_0+\sum _j?(concat(x_j,{\Delta }_{i,j},x_j)W_1)W_2$$

4.Prediction Header

Prediction Header包含兩路分支

回歸分支- 回歸每一個(gè)actor的k種模態(tài)(速度？path？)的預(yù)測軌跡，每一條軌跡包括T個(gè)時(shí)間步長，$p_{m,1}^k$軌跡的二維特征。
$$O_{m,reg}=\{(p_{m,1}^k,p_{m,2}^k,...,p_{m,T}^k){\}}_{k\in [0,K?1]}$$

分類分支-計(jì)算每個(gè)actor k種模態(tài)的置信度, 輸入特征：actor 特征 拼接 預(yù)測軌跡的偏移 embeding-$p_{m,T}^k?p_{m,1}^k$

5.模型參數(shù)學(xué)習(xí)

end-to-end 學(xué)些過程，總的損失函數(shù)=分類損失+回歸損失
$$L=L_{cls}+\alpha L_{reg}$$
分類損失：置信系數(shù)的max-margin loss
$$L_{cls}=\frac{1}{M(K?1)}\sum _{m=1}^M\sum _{k=\hat{k}}\max (0,c_{m,k}+??c_{m,\hat{k}})$$
回歸損失：正預(yù)測軌跡，逐結(jié)點(diǎn)smoth l1 loss
$$L_{reg}=\frac{1}{MT}\sum _{m=1}^M\sum _{t=1}^{T}reg(p_{m,t}^{\hat{k}}?p_{m,t}^{?})$$

注：正軌跡$\hat{k}$,在最后一個(gè)時(shí)間結(jié)點(diǎn)處擁有最小的偏移誤差。

---

參考博文：
1.ECCV2020介紹Uber在GNN-based motion forecasting的兩篇研究論文–還介紹了另一篇文章
2.2007-Learning Lane Graph Representations–提供代碼開源信息，贊！
3.文獻(xiàn)閱讀報(bào)告-Learning Lane Graph Representations for Motion Forecasting–有和VectorNet的比較。

總結(jié)

以上是生活随笔為你收集整理的PaperNotes(19)-Learning Lane Graph Representations for Motion Forecasting的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。