Search papers, labs, and topics across Lattice.
This paper investigates distillation loss functions for speech VAEs, focusing on improving unified reconstruction, understanding, and generation. It compares different alignment approaches between VAE latent representations and SSL features, evaluating their impact across the three tasks. The study finds that a joint-marginal alignment approach with adaptive weighting yields the best overall performance and allows for controllable task balance.
Aligning speech VAEs with SSL features isn't a one-size-fits-all game: joint-marginal alignment with adaptive weighting unlocks superior performance across reconstruction, understanding, and generation.
Continuous speech representations based on Variational Autoencoders (VAEs) have emerged as a promising alternative to traditional spectrogram or discrete token based features for speech generation and reconstruction. Recent research has tried to enrich the structural information in VAE latent representations by aligning with self-supervised learning (SSL) features, aiming for better generation performance. However, it remains unclear whether the widely-used alignment approach based on time-axis distillation is optimal when considering more tasks. To address this problem, this paper systematically explores different alignment approaches and analyzes their impact on the performances over three axes: reconstruction, understanding, and generation. We investigate various design choices in the distillation loss. Extensive experiments show that the joint-marginal alignment approach with adaptive weighting can achieve the best overall performance while allowing for a controllable balance.
