arXiv reaDer
SwinUNet3D-シフトされたウィンドウトランスフォーマーを使用したディープトラフィック予測のための階層アーキテクチャ
SwinUNet3D -- A Hierarchical Architecture for Deep Traffic Prediction using Shifted Window Transformers
トラフィック予測は、モビリティ管理の重要な要素であり、ロジスティクス業界を動かす重要な鍵です。長年にわたり、時系列を使用した交通予測や時空間動的予測で多くの作業が行われてきました。このホワイトペーパーでは、UNet設定でのビジョントランスフォーマーの使用について説明します。エンコーダーブランチとデコーダーブランチの両方で3Dシフトウィンドウトランスフォーマーを使用しながら、UNetのすべての畳み込みベースのビルディングブロックを完全に削除します。さらに、パッチエンコーディングの直前にフィーチャミキシングを使用して、時空間入力の深度ディメンションの縮小を回避しながら、フィーチャの相互関係を制御することを試みます。提案されたネットワークは、Neural Information Processing Systems(NeurIPS)の競技トラックで開催されたTraffic Map Movie Forecasting Challenge 2021(Traffic4cast2021)によって提供されたデータでテストされます。 Traffic4cast2021タスクは、特定のトラフィック状態の1時間(5分間の平均で12フレーム)から1時間(6フレーム)のトラフィック状態(ボリュームと平均速度)を予測することです。ソースコードはhttps://github.com/bojesomo/Traffic4Cast2021-SwinUNet3Dからオンラインで入手できます。
Traffic forecasting is an important element of mobility management, an important key that drives the logistics industry. Over the years, lots of work have been done in Traffic forecasting using time series as well as spatiotemporal dynamic forecasting. In this paper, we explore the use of vision transformer in a UNet setting. We completely remove all convolution-based building blocks in UNet, while using 3D shifted window transformer in both encoder and decoder branches. In addition, we experiment with the use of feature mixing just before patch encoding to control the inter-relationship of the feature while avoiding contraction of the depth dimension of our spatiotemporal input. The proposed network is tested on the data provided by Traffic Map Movie Forecasting Challenge 2021(Traffic4cast2021), held in the competition track of Neural Information Processing Systems (NeurIPS). Traffic4cast2021 task is to predict an hour (6 frames) of traffic conditions (volume and average speed)from one hour of given traffic state (12 frames averaged in 5 minutes time span). Source code is available online at https://github.com/bojesomo/Traffic4Cast2021-SwinUNet3D.
updated: Mon Jan 17 2022 12:58:45 GMT+0000 (UTC)
published: Mon Jan 17 2022 12:58:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト