Gumbel-Max trick

Gumbel蒸馏提升并行文本生成质量

研究人员开发了一种名为Gumbel蒸馏的新技术，以提高并行文本模型的生成质量。该方法利用Gumbel-Max技巧在潜在噪声空间和教师模型的输出token之间创建确定性连接。Gumbel蒸馏是模型无关的，可以与各种并行解码架构集成。在LM1B和OpenWebText数据集上的实验表明，与标准的MDLM训练相比，MAUVE分数显著提高了30.0%，生成困惑度降低了10.5%。