Mar, 2024

基于 Transformer 表示的概率主题建模

TL;DR我们提出了 Transformer-Representation Neural Topic Model (TNTM),它将基于变压器的嵌入空间中的主题表示优势与概率建模相结合,利用变分自编码器 (VAE) 框架提高推理速度和建模灵活性,实验结果表明,我们提出的模型在嵌入连贯性方面达到了各种最先进方法的结果,并且保持了几乎完美的主题多样性。