用多层级残差量化提升音视频图文的自监督表示能力
BRIDLE: Generalized Self-supervised Learning with Quantization
- 引入残差量化机制,通过多层代码本实现更精细的离散化
- 在音频分类任务上达到顶尖性能,图像与视频任务也优于传统方法
- 适合需要高质量自监督表征的多模态研究者
自监督学习在多个领域中通过无标签数据学习有意义表征,减少对大规模标注数据的依赖。受BERT在自然语言处理中双向上下文建模成功的启发,类似框架已被拓展至音频等领域,如使用向量量化(VQ)的BEATs模型。然而,现有方法通常依赖单一代码本,难以捕捉信号的复杂多面性,且代码本利用率低,导致部分代码向量未被充分使用。为此,我们提出BRIDLE(双向残差量化交错离散学习编码器),将残差量化(RQ)融入双向训练过程,可泛化应用于音频、图像和视频的预训练。通过多层级代码本,RQ实现潜在空间中的细粒度离散化,提升表征质量。BRIDLE采用编码器与分词器交错训练机制。我们在音频理解任务的分类基准上评估,取得当前最优结果,并在图像与视频分类任务中表现具有竞争力,显著优于传统VQ方法。
原文摘要 · Abstract (English)
Self-supervised learning has been a powerful approach for learning meaningful representations from unlabeled data across various domains, reducing the reliance on large labeled datasets. Inspired by BERT's success in capturing deep bidirectional contexts in natural language processing, similar frameworks have been adapted to other modalities such as audio, with models like BEATs extending the bidirectional training paradigm to audio signals using vector quantization (VQ). However, these frameworks face challenges, notably their dependence on a single codebook for quantization, which may not capture the complex, multifaceted nature of signals. In addition, inefficiencies in codebook utilization lead to underutilized code vectors. To address these limitations, we introduce BRIDLE (Bidirectional Residual Quantization Interleaved Discrete Learning Encoder), a self-supervised encoder pretraining framework that incorporates residual quantization (RQ) into the bidirectional training process, and is generalized for pretraining with audio, image, and video. Using multiple hierarchical codebooks, RQ enables fine-grained discretization in the latent space, enhancing representation quality. BRIDLE involves an interleaved training procedure between the encoder and tokenizer. We evaluate BRIDLE on audio understanding tasks using classification benchmarks, achieving state-of-the-art results, and demonstrate competitive performance on image classification and video classification tasks, showing consistent improvements over traditional VQ methods in downstream performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。