Feb, 2024

利用变压器量化变分自编码器改进离散潜空间中的语义控制

TL;DR通过使用 Vector Quantized Variational AutoEncoders (VQVAEs) 内的离散潜空间来提高 Transformer-based VAE 中的语义控制和生成性能,我们提出了一种新型模型 T5VQVAE,利用 VQVAEs 的可控性来引导 T5 中的自注意机制,从而更好地保留语义信息并取得优于现有 VAE 模型(包括 Optimus)的性能,在自动编码、文本转换和推理等不同任务中。此外,T5VQVAE 还表现出了改进的推理能力,为下游自然语言和符号推理任务提供了潜在的应用。