Mar, 2022

RNN-Transducer 基于采样 softmax 的高效内存训练

TL;DR本文提出了使用采样 softmax 算法减少 RNN-Transducer 在训练过程中的内存消耗,并基于辅助 CTC losses 分布进行 sampling vocabulary 来提高模型准确性,在 LibriSpeech、AISHELL-1 和 CSJ-APS 三个数据集上进行实验,结果显示采样 softmax 显著降低了内存消耗,且仍然保持了基线模型的准确度。