arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.05910eess.AScs.CLcs.LGcs.SD
收藏

自监督语音Transformer中的属性神经元

Property Neurons in Self-Supervised Speech Transformers

National Taiwan University(台湾大学);University of Edinburgh(爱丁堡大学)

Tzu-Quan Lin, Guan-Ting Lin, Hung-yi Lee, Hao Tang

AI总结:

本文在自监督语音Transformer的前馈层中识别属性神经元,并证明在剪枝时保护这些神经元比基于范数的剪枝更有效,从而提升下游性能。

摘要:

已有许多研究分析自监督语音Transformer,尤其是逐层分析。然而,期望有一种方法能够精确定位负责特定语音属性的神经元子集,从而便于模型剪枝和模型编辑。在这项工作中,我们在Transformer的前馈层中识别了一组属性神经元,以研究语音相关属性(如音素、性别和音高)是如何存储的。当移除特定属性的神经元(一种简单的模型编辑形式)时,相应的下游性能显著下降,表明属性神经元的重要性。我们将此方法应用于Transformer中前馈层的剪枝,这些层包含了大部分模型参数。我们表明,在剪枝过程中保护属性神经元比基于范数的剪枝显著更有效。用于识别属性神经元的代码可在 https://github.com/nervjack2/PropertyNeurons 获取。

英文摘要:

There have been many studies on analyzing self-supervised speech Transformers, in particular, with layer-wise analysis. It is, however, desirable to have an approach that can pinpoint exactly a subset of neurons that is responsible for a particular property of speech, being amenable to model pruning and model editing. In this work, we identify a set of property neurons in the feedforward layers of Transformers to study how speech-related properties, such as phones, gender, and pitch, are stored. When removing neurons of a particular property (a simple form of model editing), the respective downstream performance significantly degrades, showing the importance of the property neurons. We apply this approach to pruning the feedforward layers in Transformers, where most of the model parameters are. We show that protecting property neurons during pruning is significantly more effective than norm-based pruning. The code for identifying property neurons is available at https://github.com/nervjack2/PropertyNeurons.

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

↑