ICRA 2020 | 行人軌跡預測競賽冠軍方案總結
在實際操作過程中,由于數據集中缺乏場景信息,我們對模型做了適當的調整。在世界模型中(對應上圖的Interaction Net),我們僅使用了現有數據集,以及模型能夠提供的位置信息和跟蹤信息LSTM隱狀態信息。最終得到的模型結構設計如下圖3所示:

圖3 競賽使用的基于世界模型的預測算法
整個模型基于Seq2Seq結構,主要包含歷史軌跡編碼模塊(Encoder)、世界模型(Interaction Module)和解碼預測模塊(Decoder)三個部分。其中,編碼器的功能在于對行人歷史軌跡進行編碼,主要提取行人在動態環境中的運動模式;解碼器則是利用編碼器得到的行人運動模式特征,來預測他們未來的運動軌跡分布。
需要強調一下,在整個編碼與解碼的過程中,都需要對世界模型進行實時更新(Update)與查詢(Query)兩種操作。更新操作主要根據時序的推進,將行人的運動信息實時編入世界模型中;查詢操作則是根據全局的世界地圖以及行人的自身位置,來獲取行人當前鄰域內的環境特征。

圖4 編碼階段
在圖4中,展示了我們模型在歷史軌跡編碼階段的計算流程。編碼階段共有9個時刻,對應9個歷史觀測時間點,每個時刻都執行相同的操作。以 t 時刻為例。
首先,將 t 時刻的所有行人坐標數據,包含:

位置集合

速度集合

所有行人跟蹤信息(上時刻編碼得到的LSTM隱狀態)
將以上信息輸入到世界模型中更新地圖信息,即Update操作。整個Update操作經過MLP、MaxPooling以及GRU等模塊獲得一個全局的時空地圖特征R;然后,每個LSTM(對應一個行人),使用其當前觀測時刻的坐標信息:

然后與R進行Attention操作,得到個人領域內的時空特征,最后與他的坐標信息、上時刻隱狀態信息一并輸入到LSTM并更新LSTM內置狀態。
解碼預測階段的流程與歷史軌跡編碼階段基本一致,但存在兩個細微的不同點:
區別1:編碼階段每個行人對應的LSTM隱狀態的初始化為0;而解碼階段,LSTM由編碼階段的LSTM隱狀態和噪聲共同初始化。
區別2:編碼階段行人對應的LSTM和世界模型使用的是行人歷史觀測坐標;而解碼階段使用的是上時刻預測的行人坐標。

圖5 解碼預測階段
四、數據預處理與后處理
為了對數據有更好的理解,便于使用更適合的模型,我們對訓練數據做了一些預處理操作。首先,數據集給出了各個行人的行為標簽,這些標簽是根據規則得到的,由于我們采用了交互預測的方法,希望模型能自動學習行人與周圍主體之間的位置關系、速度關系等,所以我們就不直接使用標注中的“類型”信息;然后這次比賽的數據采集自馬路、校園等不同場景中行人的運動軌跡,場景之間的差異性非常大,訓練集和測試集數據分布不太一致。
于是,我們做了數據的可視化工作,將所有軌跡數據的起點放置于坐標軸的原點處,根據歷史觀測軌跡(前9個時刻)終點的位置朝向,將所有軌跡分為4類:沿左上方運動(top-left moving)、沿右上方運動(top-right moving)、沿左下方運動(bottom-left moving)和沿右下方運動(bottom-right moving)。分布的結果如圖6所示,可以發現,訓練集和測試集的數據分布存在一定的差距。

圖6 訓練集與測試集歷史觀測軌跡中行人運動方向分布
請輸入評論內容...
請輸入評論/評論長度6~500個字
最新活動更多
- 1 AI狂歡遇上油價破百,全球股市還能漲多久? | 產聯看全球
- 2 OpenAI深夜王炸!ChatGPT Images 2.0實測:中文穩、細節炸,設計師慌了
- 3 6000億美元估值錨定:字節跳動的“去單一化”突圍與估值重構
- 4 Tesla AI5芯片最新進展總結
- 5 連夜測了一波DeepSeek-V4,我發現它可能只剩“審美”這個短板了
- 6 熱點丨AI“瑜亮之爭”:既生OpenClaw,何生Hermes?
- 7 AI界的殺豬盤:9秒刪庫跑路,全員被封號,還繼續扣錢!
- 8 2026,人形機器人只贏了面子
- 9 DeepSeek降價90%:價格屠夫不是身份,是戰略
- 10 AI Infra產業鏈卡在哪里了?


分享













