arXiv:2411.09339cs.SDcs.CL2024-11被引 4

轻量Transformer通过重参数化技术提升性能,适配边缘设备语音情感识别。

Re-Parameterization of Lightweight Transformer for On-Device Speech Emotion Recognition

  • 训练时插入高秩分解层增强学习能力,推理时合并恢复原结构。
  • 在IEMOCAP等3个数据集上,轻量模型性能接近大模型。
  • 适合资源受限的物联网设备部署,无需牺牲精度。

随着机器学习模型在边缘或物联网(IoT)设备上的广泛应用,如何在计算能力与存储空间有限的设备上部署先进模型仍具挑战。当前主流的Transformer架构虽在多个领域取得成功,但其复杂性限制了在资源受限的IoT设备上的应用。尽管已有多种模型压缩方法,但常伴随显著性能下降。为此,本文提出一种新的轻量Transformer重参数化方法,包含训练阶段的高秩分解(HRF)和推理阶段的去高秩分解(deHRF)。训练时,在轻量Transformer的前馈网络(FFN)前加入额外线性层以增强模型学习能力;推理时,将辅助的HRF层与后续的FFN层合并为单一线性层,恢复原始结构。我们在三种主流Transformer变体(ConvTransformer、Conformer、SpeechFormer)上评估该方法在语音情感识别任务中的效果,使用IEMOCAP、M3ED和DAIC-WOZ三个数据集。实验结果表明,所提方法持续提升轻量Transformer性能,使其达到与大模型相当的水平。该重参数化策略使先进Transformer模型可在资源受限的IoT设备上高效部署。

原文摘要 · Abstract (English)

With the increasing implementation of machine learning models on edge or Internet-of-Things (IoT) devices, deploying advanced models on resource-constrained IoT devices remains challenging. Transformer models, a currently dominant neural architecture, have achieved great success in broad domains but their complexity hinders its deployment on IoT devices with limited computation capability and storage size. Although many model compression approaches have been explored, they often suffer from notorious performance degradation. To address this issue, we introduce a new method, namely Transformer Re-parameterization, to boost the performance of lightweight Transformer models. It consists of two processes: the High-Rank Factorization (HRF) process in the training stage and the deHigh-Rank Factorization (deHRF) process in the inference stage. In the former process, we insert an additional linear layer before the Feed-Forward Network (FFN) of the lightweight Transformer. It is supposed that the inserted HRF layers can enhance the model learning capability. In the later process, the auxiliary HRF layer will be merged together with the following FFN layer into one linear layer and thus recover the original structure of the lightweight model. To examine the effectiveness of the proposed method, we evaluate it on three widely used Transformer variants, i.e., ConvTransformer, Conformer, and SpeechFormer networks, in the application of speech emotion recognition on the IEMOCAP, M3ED and DAIC-WOZ datasets. Experimental results show that our proposed method consistently improves the performance of lightweight Transformers, even making them comparable to large models. The proposed re-parameterization approach enables advanced Transformer models to be deployed on resource-constrained IoT devices.

轻量模型Transformer边缘计算语音识别

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