arXiv reaDer
ETAD: ラップトップでエンド ツー エンドのトレーニング アクション検出
ETAD: Training Action Detection End to End on a Laptop
エンド ツー エンドのトレーニングを使用した時間的アクション検出 (TAD) は、ビデオの長さが長いため、コンピューティング リソースに対する膨大な需要という問題に悩まされることがよくあります。この作業では、非常に低い GPU メモリ消費でビデオ フレームから直接トレーニングできる効率的な一時的アクション検出器 (ETAD) を提案します。私たちの主なアイデアは、トレーニングの各反復における特徴と勾配の間の重い計算を最小限に抑え、バランスを取ることです。ビデオエンコーダーを介してスニペットフレームを順次転送し、エンコーダーを更新するためにグラデーションの必要な小さな部分のみを逆方向に転送することを提案します。トレーニングにおける計算の冗長性をさらに軽減するために、トレーニング中に提案の小さなサブセットのみを動的にサンプリングすることを提案します。さらに、エンコーダと検出器の両方について、さまざまなサンプリング戦略と比率が研究されています。 ETAD は、TAD ベンチマークで最先端のパフォーマンスを驚くべき効率で達成します。 ActivityNet-1.3 では、18 時間で ETAD をトレーニングすると、エンド ツー エンドのトレーニングでビデオあたりわずか 1.3 GB のメモリ消費で 38.25% の平均 mAP に達することができます。私たちのコードは公開されます。
Temporal action detection (TAD) with end-to-end training often suffers from the pain of huge demand for computing resources due to long video duration. In this work, we propose an efficient temporal action detector (ETAD) that can train directly from video frames with extremely low GPU memory consumption. Our main idea is to minimize and balance the heavy computation among features and gradients in each training iteration. We propose to sequentially forward the snippet frame through the video encoder, and backward only a small necessary portion of gradients to update the encoder. To further alleviate the computational redundancy in training, we propose to dynamically sample only a small subset of proposals during training. Moreover, various sampling strategies and ratios are studied for both the encoder and detector. ETAD achieves state-of-the-art performance on TAD benchmarks with remarkable efficiency. On ActivityNet-1.3, training ETAD in 18 hours can reach 38.25% average mAP with only 1.3 GB memory consumption per video under end-to-end training. Our code will be publicly released.
updated: Mon Nov 28 2022 11:43:24 GMT+0000 (UTC)
published: Sat May 14 2022 21:16:21 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アソシエイト