arXiv:2411.07118cs.CVcs.HC2024-11被引 10

用轻量卷积混洗替代注意力,提升手势识别效率

ConvMixFormer- A Resource-efficient Convolution Mixer for Transformer-based Dynamic Hand Gesture Recognition

  • 用卷积层代替自注意力,降低计算开销
  • 在NVidia与Briareo数据集上达最优性能
  • 参数量减半,适合边缘设备部署

Transformer模型在自然语言处理和计算机视觉领域表现卓越,现也被用于手势识别。然而其自注意力机制随序列长度呈二次增长,导致计算复杂度高。为此,本文提出ConvMixFormer,以简单卷积混合器替代自注意力模块,显著降低计算成本与参数量。该设计能有效捕捉局部空间特征,弥补传统注意力在序列处理中的不足。同时引入高效门控机制替代标准前馈网络,进一步减少可学习参数,接近原Transformer的一半,实现更快训练速度。模型在NVidia动态手势与Briareo数据集上均取得领先结果,且在单模态与多模态输入下表现优异。代码已开源:https://github.com/mallikagarg/ConvMixFormer。

原文摘要 · Abstract (English)

Transformer models have demonstrated remarkable success in many domains such as natural language processing (NLP) and computer vision. With the growing interest in transformer-based architectures, they are now utilized for gesture recognition. So, we also explore and devise a novel ConvMixFormer architecture for dynamic hand gestures. The transformers use quadratic scaling of the attention features with the sequential data, due to which these models are computationally complex and heavy. We have considered this drawback of the transformer and designed a resource-efficient model that replaces the self-attention in the transformer with the simple convolutional layer-based token mixer. The computational cost and the parameters used for the convolution-based mixer are comparatively less than the quadratic self-attention. Convolution-mixer helps the model capture the local spatial features that self-attention struggles to capture due to their sequential processing nature. Further, an efficient gate mechanism is employed instead of a conventional feed-forward network in the transformer to help the model control the flow of features within different stages of the proposed model. This design uses fewer learnable parameters which is nearly half the vanilla transformer that helps in fast and efficient training. The proposed method is evaluated on NVidia Dynamic Hand Gesture and Briareo datasets and our model has achieved state-of-the-art results on single and multimodal inputs. We have also shown the parameter efficiency of the proposed ConvMixFormer model compared to other methods. The source code is available at https://github.com/mallikagarg/ConvMixFormer.

手势识别轻量化模型卷积混合器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。