arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.01708cs.CV

AVESFormer:面向实时音视频分割的高效Transformer设计

AVESFormer: Efficient Transformer Design for Real-Time Audio-Visual Segmentation

School of Artificial Intelligence, University of Chinese Academy of Sciences (UCAS)(中国科学院大学人工智能学院);Institute of Automation, Chinese Academy of Sciences (CASIA)(中国科学院自动化研究所);China Tower Corporation Limited(中国铁塔股份有限公司)

Zili Wang, Qi Yang, Linsu Shi, Jiazhong Yu, Qinghua Liang, Fei Li, Shiming Xiang

AI总结:

本文提出首个实时音视频分割Transformer模型AVESFormer,通过高效提示查询生成器和ELF解码器解决注意力耗散与解码器低效问题,在多个基准上取得领先性能并实现速度与精度的良好平衡。

摘要:

近年来,基于Transformer的模型在音视频分割(AVS)任务上表现出卓越的性能。然而,其高昂的计算成本使得实时推理难以实现。通过刻画网络的注意力图,我们识别出AVS模型中的两个关键障碍:1)注意力耗散,对应于Softmax在受限帧内过度集中的注意力权重;2)低效且负担沉重的Transformer解码器,由早期阶段狭窄的关注模式所导致。在本文中,我们提出AVESFormer,这是首个实时音视频高效分割Transformer,能够同时实现快速、高效和轻量化。我们的模型利用高效的提示查询生成器来纠正交叉注意力的行为。此外,我们提出ELF解码器,通过引入适合局部特征的卷积来降低计算负担,从而带来更高的效率。大量实验表明,我们的AVESFormer显著提升了模型性能,在S4上达到79.9%,在MS3上达到57.9%,在AVSS上达到31.2%,超越了此前的领先方法,并在性能与速度之间取得了优异的权衡。代码可在https://github.com/MarkXCloud/AVESFormer.git获取。

英文摘要:

Recently, transformer-based models have demonstrated remarkable performance on audio-visual segmentation (AVS) tasks. However, their expensive computational cost makes real-time inference impractical. By characterizing attention maps of the network, we identify two key obstacles in AVS models: 1) attention dissipation, corresponding to the over-concentrated attention weights by Softmax within restricted frames, and 2) inefficient, burdensome transformer decoder, caused by narrow focus patterns in early stages. In this paper, we introduce AVESFormer, the first real-time Audio-Visual Efficient Segmentation transformer that achieves fast, efficient and light-weight simultaneously. Our model leverages an efficient prompt query generator to correct the behaviour of cross-attention. Additionally, we propose ELF decoder to bring greater efficiency by facilitating convolutions suitable for local features to reduce computational burdens. Extensive experiments demonstrate that our AVESFormer significantly enhances model performance, achieving 79.9% on S4, 57.9% on MS3 and 31.2% on AVSS, outperforming previous state-of-the-art and achieving an excellent trade-off between performance and speed. Code can be found at https://github.com/MarkXCloud/AVESFormer.git.

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

↑