Learning Lane Graph Representations for Motion Forecasting
- 1.ActorNet
- 2.MapNet
- 3.FusionNet
- 4.Prediction Header
- 5.模型参数学习
自动驾驶论文阅读笔记2
Uber–ECCV2020–论文文章代码
模型的作用:Motion Forecasting (这个motion都包括什么呢?)
方法:模型由四个模块组成–ActorNet、MapNet、FusionNet、Header
1.ActorNet
作用–编码actor轨迹特征
- 输入:每一条actor的轨迹(3xT的输入向量)(actor包括所有运动者?)
- 操作:1D CNN + FPN(特征金字塔)(多尺度不断上采样融合特征)
- 输出:该轨迹的特征向量(多少维度呢?128维度么)
轨迹表示–位移差
{Δp−(T−1),...,Δp−1,Δp0}\{\Delta p_{-(T-1)},...,\Delta p_{-1},\Delta p_{0}\}{Δp−(T−1),...,Δp−1,Δp0}
Δpt=(xt,yt)−(xt−1,yt−1)\Delta p_{t} = (x_t,y_t)-(x_{t-1},y_{t-1})Δpt=(xt,yt)−(xt−1,yt−1)
长度为T,不足T的padding 0,下面2xT的位移向量 拼接 1xT的padding标志向量(1-表示该位置的位移是padding的)
ActorNet–3组1D卷积,每组包括两个残差块;特征金字塔融合多尺度的特征(卷积 卷下去,上采样,再和对应的尺度叠加)
输出-- The output of ActorNet is a temporal feature map,whose element at t = 0 is used as the actor feature.(不是很清楚这个时间序列是怎么回事?)
2.MapNet
作用–编码地图特征,主要是对车道的编码
- 构建lane graph(结点与连接的定义)
- 利用LaneGCN获取lane图的特征
2.1 构建lane Graph
- lane node --车道中心线的一段,结点位置为两端结点坐标的均值。==一条车?所有?==道线上的众多结点可以表示为向量V∈RN×2V\in\mathbb{R}^{N\times 2}V∈RN×2(结点的二维特征表示)
- 车道结点特征向量带有4个连接矩阵{Ai}i∈{pre,suc,left,right}\{A_i\}_{i\in \{pre,suc,left,right\}}{Ai}i∈{pre,suc,left,right},Ai∈RN×NA_i\in\mathbb{R}^{N\times N}Ai∈RN×N, Ai,jk=1A_{i,jk}=1Ai,jk=1表示结点jjj 存在一个类型iii的邻居结点kkk。
- LaneConv Operator
a). lane node 特征xix_ixi–编码形状(长度方向),位置(空间坐标)信息,经过全联接层处理后,输出lane node特征xix_ixi;构成结点特征矩阵XXX。
xi=MLPshape(viend−vistart)+MLPloc(vi)x_i=MLP_{shape}(v_i^{end}-v_i^{start})+MLP_{loc}(v_i)xi=MLPshape(viend−vistart)+MLPloc(vi)
b).LaneConv 为了获得lane graph大规模拓扑信息(四个AiA_iAi矩阵都用上)
Y=XW0+∑i∈{pre,suc,left,right}AiXWiY=XW_0+\sum_{i\in\{pre,suc,left,right\}}A_iXW_iY=XW0+i∈{pre,suc,left,right}∑AiXWi
c).Dilated LaneConv 为了让模型获得车道线方向的长时依赖关系(速度快的物体位移大,只用到ApreA_{pre}Apre和AsucA_{suc}Asuc)
Y=XW0+AprekXWpre,k+AsuckXWsuc,kY=XW_0 + A^k_{pre}XW_{pre,k} + A^k_{suc}XW_{suc,k}Y=XW0+AprekXWpre,k+AsuckXWsuc,k
d). LaneConv(k1,...,kc)LaneConv(k_1,...,k_c)LaneConv(k1,...,kc)=Dilated LaneConv + LaneConv - kck_ckc为第ccc个dilation 尺寸
Y=XW0+∑i∈{left,right}AiXWi+∑c=1C(AprekcXWpre,kc+AsuckcXWsuc,kc)Y=XW_0+\sum_{i\in\{left,right\}}A_iXW_i + \sum_{c=1}^C(A^{k_c}_{pre}XW_{pre,k_c}+A^{k_c}_{suc}XW_{suc,k_c})Y=XW0+i∈{left,right}∑AiXWi+c=1∑C(AprekcXWpre,kc+AsuckcXWsuc,kc)
2.2 LaneGCN操作
LaneConv(k1,...,kc)LaneConv(k_1,...,k_c)LaneConv(k1,...,kc) + Linear Layer 构成残差块, 4个残差块堆叠,构成LaneGCN。
注意点:
每个结点与邻居结点的连接关系,指明前后左右的结点。
a). 结点A的前驱结点、后续结点:同一条车道线上,能够到达A的结点和A能够到达的结点
b).结点A的左邻居结点、右邻居结点:邻居车道线上空间距离l2l_2l2最近的结点
对车道结点图不采用广泛使用的图卷积L=D−12(I+A)D−12(1)L=D^{-\frac{1}{2}}(I+A)D^{-\frac{1}{2}}(1)L=D−21(I+A)D−21(1)更新隐状态的原因:
a). 不知道结点特征会保存何种车道信息(可解释性不强?)
b). 该拉普拉斯矩阵操作没法获取结点的连接关系信息(没有对AiA_iAi矩阵的操作,(1)式中的A为结点的邻接矩阵,与AiA_iAi内涵不同)
c). 无法解决长时依赖性的问题
为了解决以上三个不足点,作者提出了LaneConv Operator操作。
Dilated LaneConv 参考了dilated convolution,说是能够沿着车道线传递k步的信息。
3.FusionNet
利用空间注意力机制(spatial attention)(用于构建A2L, L2A, A2A) 和LaneGCN(主要用来构建L2L网络的)融合actor结点和lane结点的信息,
四个信息融合模块:
- A2L-将实时交通信息传达给lane node
- L2L-依据实时交通信息,更新lane node 的特征
- L2A-将更新后lane node 的特征返回给actor
- A2A-解决actor之间的相互作用,并且actor特征给motion预测网络
L2L 网络结构-结点特征更新时的LaneGAN结构一致
A2L, L2A, A2A网络结构一致,采用空间注意力机制构成残差块。
yi=xiW0+∑jϕ(concat(xj,Δi,j,xj)W1)W2y_i=x_iW_0 + \sum_j\phi(concat(x_j,\Delta_{i,j},x_j)W_1)W_2yi=xiW0+j∑ϕ(concat(xj,Δi,j,xj)W1)W2
4.Prediction Header
Prediction Header包含两路分支
-
回归分支- 回归每一个actor的k种模态(速度?path?)的预测轨迹,每一条轨迹包括T个时间步长,pm,1kp_{m,1}^kpm,1k轨迹的二维特征。
Om,reg={(pm,1k,pm,2k,...,pm,Tk)}k∈[0,K−1]O_{m,reg}=\{(p_{m,1}^k,p_{m,2}^k,...,p_{m,T}^k)\}_{k\in[0,K-1]}Om,reg={(pm,1k,pm,2k,...,pm,Tk)}k∈[0,K−1] -
分类分支-计算每个actor k种模态的置信度, 输入特征:actor 特征 拼接 预测轨迹的偏移 embeding-pm,Tk−pm,1kp_{m,T}^k-p_{m,1}^kpm,Tk−pm,1k
5.模型参数学习
end-to-end 学些过程,总的损失函数=分类损失+回归损失
L=Lcls+αLregL=L_{cls} + \alpha L_{reg}L=Lcls+αLreg
分类损失:置信系数的max-margin loss
Lcls=1M(K−1)∑m=1M∑k≠k^max(0,cm,k+ϵ−cm,k^)L_{cls}=\frac{1}{M(K-1)}\sum_{m=1}^M\sum_{k\neq\hat{k}}\max(0, c_{m,k}+\epsilon-c_{m,\hat{k}})Lcls=M(K−1)1m=1∑Mk=k^∑max(0,cm,k+ϵ−cm,k^)
回归损失:正预测轨迹,逐结点smoth l1 loss
Lreg=1MT∑m=1M∑t=1Treg(pm,tk^−pm,t∗)L_{reg}=\frac{1}{MT}\sum_{m=1}^M\sum_{t=1}^Treg(p_{m,t}^{\hat{k}}-p_{m,t}^*)Lreg=MT1m=1∑Mt=1∑Treg(pm,tk^−pm,t∗)
注:正轨迹k^\hat{k}k^,在最后一个时间结点处拥有最小的偏移误差。
参考博文:
1.ECCV2020介绍Uber在GNN-based motion forecasting的两篇研究论文–还介绍了另一篇文章
2.2007-Learning Lane Graph Representations–提供代码开源信息,赞!
3.文献阅读报告-Learning Lane Graph Representations for Motion Forecasting–有和VectorNet的比较。