arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2211.10636cs.CVcs.LG

EVEREST:通过移除冗余时空令牌实现高效掩码视频自编码器

EVEREST: Efficient Masked Video Autoencoder by Removing Redundant Spatiotemporal Tokens

Korea Military Academy(韩国军事学院);UNC Chapel Hill(北卡罗来纳大学教堂山分校);KAIST(韩国科学技术院);ETRI(韩国电子通信研究院);DeepAuto

Sunil Hwang, Jaehong Yoon, Youngwan Lee, Sung Ju Hwang

AI总结:

针对掩码视频自编码器因随机掩码导致算力与内存浪费的问题,提出EVEREST方法,通过筛选高信息密度的时空令牌与帧,在单台8GPU机器上即可实现性能比肩高算力基线的高效视频表示学习。

摘要:

掩码视频自编码器(Masked Video Autoencoder, MVA)方法已展现出巨大潜力,其性能显著超越了以往的视频表示学习方法。然而,由于采用随机掩码策略,这类方法在预测无信息价值的令牌/帧时浪费了大量计算资源与内存(例如,需使用超过16个节点、配备128块NVIDIA A100 GPU的算力)。为解决这一问题,我们利用视频块之间信息密度不均的特性,提出了EVEREST——一种效率极高的视频表示学习MVA方法,该方法在预训练和微调阶段均能筛选出包含丰富运动特征的令牌,并丢弃无信息价值的令牌。我们进一步提出了一种高信息密度帧选择策略,使模型能够聚焦于信息丰富且具有因果性的帧,同时将冗余度降至最低。我们的方法大幅降低了MVA的计算与内存需求,仅需单台配备8块GPU的机器即可完成预训练和微调,且在多个基准数据集以及未经筛选的Ego4D数据集上,性能与计算和内存开销巨大的基线方法相当。我们希望本研究能为降低视频理解领域进一步研究的门槛贡献力量。

英文摘要:

Masked Video Autoencoder (MVA) approaches have demonstrated their potential by significantly outperforming previous video representation learning methods. However, they waste an excessive amount of computations and memory in predicting uninformative tokens/frames due to random masking strategies. (e.g., over 16 nodes with 128 NVIDIA A100 GPUs). To resolve this issue, we exploit the unequal information density among the patches in videos and propose EVEREST, a surprisingly efficient MVA approach for video representation learning that finds tokens containing rich motion features and discards uninformative ones during both pre-training and fine-tuning. We further present an information-intensive frame selection strategy that allows the model to focus on informative and causal frames with minimal redundancy. Our method significantly reduces the computation and memory requirements of MVA, enabling the pre-training and fine-tuning on a single machine with 8 GPUs while achieving comparable performance to computation- and memory-heavy baselines on multiple benchmarks and the uncurated Ego4D dataset. We hope that our work contributes to reducing the barrier to further research on video understanding.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑