【论文阅读】REVISITING DEEP AUDIO-TEXT RETRIEVAL THROUGH THE LENS OF TRANSPORTATION

REVISITING DEEP AUDIO-TEXT RETRIEVAL THROUGH THE LENS OF TRANSPORTATION

• 原文摘要

  • 研究背景：

    • Learning-to-match (LTM)框架是一种有效的逆最优传输方法，用于学习两个数据源之间的基础度量，以促进后续匹配。

      • LTM框架属于逆最优传输问题（inverse OT）

      • 核心功能是学习ground metric（基础度量）

      • 应用场景是跨模态数据匹配

    • 存在问题：

      • 传统LTM框架面临可扩展性挑战，每次更新基础度量参数时都需要使用整个数据集。"

        • 计算效率瓶颈

        • 批处理方式导致内存问题

    • 方法创新：

      • 关键技术创新

        • 将深度学习中的mini-batch采样引入LTM–m-LTM

        • 采用马氏距离族（Mahalanobis-enhanced metrics）增强度量学习

      • 针对实际中训练数据不对齐的问题

        • 提出使用部分最优传输（partial OT）的变体

        • 解决错位数据对的影响

    • 实验验证：

      • 数据集：AudioCaps, Clotho, ESC-50三个音频-文本数据集

      • 实验结果：

        • 学习到富有表现力的联合嵌入空间

        • 达到SOTA性能

        • 在zero-shot声音事件检测任务中：

          • 跨模态间隙（modality gap）小于triplet/contrastive loss

        • 噪声鲁棒性：

          • partial OT+m-LTM比contrastive loss更抗噪
          • 特别在AudioCaps数据集不同噪声比例下验证

1. Introduction

• 研究背景与应用价值

  • 核心任务：音频-文本匹配（audio-text matching）

  • 应用场景：

    • 音频检索
    • 音频描述生成
    • 文本到音频生成

  • 技术目标：构建跨模态共享嵌入空间，需满足：

    • 高表达性
    • 最小化模态间隙
    • 增强下游任务迁移性

• 现有方法局限

  • 主流方法缺陷

    • 对比学习和三元组损失存在两大问题：

      • 几何结构忽视：平等对待所有负样本，忽略嵌入空间的几何特性
      • 模态间隙保留：对比学习会保持模态差异，损害迁移性

    • 数据噪声挑战

      • 错位数据对普遍存在：
        • 数据对源于网络数据收集
        • 现有解决方案依赖特定网络架构，泛化性差

• 创新方案

  • 核心框架：m-LTM

    • 理论基础：最优传输视角重构跨模态匹配

    • 关键技术突破：

      • 可扩展性：引入mini-batch训练策略
      • 度量增强：用马氏距离替代欧氏距离
      • 噪声鲁棒：首次在音频-文本匹配中采用部分最优传输

• 三大贡献

  1. 深度学习适配：

     • 将LTM框架改造为mini-batch训练模式
     • 结合熵正则化最优传输与马氏距离
     • 在AudioCaps/Clotho数据集达到SOTA

  2. 模态间隙消除：

     • 通过跨数据集实验（含ESC-50）验证

     • 在zero-shot声音事件检测任务中超越对比学习

  3. 噪声处理创新：

     • POT方案在AudioCaps数据集上展示更强噪声容忍度

2. Preliminary

2.1 DEEP AUDIO-TEXT RETRIEVAL

2.1.1 概述

• 任务定义

  • 音频-文本检索的目标是学习音频与其对应文本描述之间的跨模态对齐关系
  • 系统要能理解某段音频应该对应哪段文本描述（反之亦然）。

• 对齐方式

  • 学习一个联合嵌入空间，能将音频和文本映射到同一个空间中，使得它们之间的相似性可以直接比较。

• 主流方法

  • 当前最有效的方法是对比学习，能训练出具有良好表现力的跨模态嵌入空间。

2.1.2 训练

• 数据集：设训练集为 $D=\{(x_i,y_i){\}}_{i=1}^n$，其中 $x_i$ 是音频，$y_i$ 是对应的文本描述。

• 目标函数

  • 定义两个编码器：

    • $f_{\theta }(x_i)$：将音频编码为嵌入向量

    • $g_{\varphi }(y_i)$：将文本编码为嵌入向量

  • 目标函数如下（对比学习损失）：
    $$\underset{\theta ,\varphi }{\min }{\mathbb{E}}_{(X^b,Y^b)\sim D}[-\sum _{i=1}^b\log \frac{\exp (s(f_{\theta }(x_i),g_{\varphi }(y_i))/\tau )}{{\sum }_{k=1}^b\exp (s(f_{\theta }(x_i),g_{\varphi }(y_k))/\tau )}-\sum _{i=1}^b\log \frac{\exp (s(f_{\theta }(x_i),g_{\varphi }(y_i))/\tau )}{{\sum }_{k=1}^b\exp (s(f_{\theta }(x_k),g_{\varphi }(y_i))/\tau )}]$$

    • $s(\cdot ,\cdot )$ 是余弦相似度

    • $\tau$ 是温度参数

    • $(X^b,Y^b)$ 表示一个大小为 $b$ 的 mini-batch

  • 这是一种双向对比损失（Audio → Text & Text → Audio）。

2.1.3 检索方法

• 对于测试集中的音频集合 Xtest={xi}i=1n′X_{\text{test}} = \{x_i\}_{i=1}^{n'}Xtest​={xi​}i=1n′​，文本集合 Ytest={yj}j=1m′Y_{\text{test}} = \{y_j\}_{j=1}^{m'}Ytest​={yj​}j=1m′​，我们通过计算嵌入向量之间的余弦相似度生成一个排名矩阵 $R(X_{\text{test}},Y_{\text{test}})$。

• 给定音频 $x_i$，其对应文本为：$\hat{y}=\arg {\max }_{y_j\in Y_{\text{test}}}\frac{\exp (s(f_{\theta }(x_i),g_{\varphi }(y_j)))}{{\sum }_{k=1}^{m^{\prime }}\exp (s(f_{\theta }(x_i),g_{\varphi }(y_k)))}$

• 这就是 softmax 排序概率最大值对应的文本。

2.2 LEARNING TO MATCH

2.2.1 最优传输 Optimal Transport

• 定义两个经验概率分布：

  • $P_X=\frac{1}{n}{\sum }_{i=1}^n{\delta }_{x_i}$

  • $P_Y=\frac{1}{m}{\sum }_{j=1}^m{\delta }_{y_j}$

• 带熵正则项的最优传输问题为：
  $${\pi }_{X,Y}^{\epsilon ,c}=\arg \underset{\pi \in \Pi (P_X,P_Y)}{\min }\sum _{i=1}^n\sum _{j=1}^m{\pi }_{ij}c(x_i,y_j)-\epsilon \sum _{i=1}^n\sum _{j=1}^m{\pi }_{ij}\log {\pi }_{ij}$$

  • $c(x_i,y_j)$：代价函数（地面距离）

  • $\Pi (P_X,P_Y)$：所有合法的传输计划矩阵集合（满足行列边缘和为均匀分布）

  • $\epsilon >0$：熵正则系数，用于控制平滑性

• 该问题可通过 Sinkhorn 算法 高效求解。

2.2.2 LTM目标：学习代价函数

• 概述

  • 学习一个可训练的 ground metric（地面代价函数） $c_{\varphi }(x,y)$
  • 使得训练集中匹配对的传输计划最接近于理想状态。

• LTM优化目标
  $$\underset{c\in \mathcal{C}}{\inf }\text{KL}(\hat{\pi }\Vert {\pi }_{\epsilon ,c}^{X,Y})$$

  • $\hat{\pi }$：理想情况下的“独热”匹配矩阵，只有对角线非零（完美一一对应）

  • $\mathcal{C}$：代价函数的候选空间

• LTM 的本质是逼近理想矩阵与最优传输计划之间的差异，使用 KL 散度度量。

  • 优化目标是让基于 $c_{\varphi }$ 得到的最优传输计划 ${\pi }_{\varphi }^{\epsilon ,c}$ 尽可能接近理想矩阵 $\hat{\pi }$。
  • 所以LTM学习的是$c$，而不是$\pi$

2.2.3 基于 ground metric 的检索

• 在测试时，先用已学习的代价函数 $c$ 构造代价矩阵，并通过熵正则最优传输解出最优匹配矩阵 ${\pi }_{X,Y}^{\epsilon ,c}$。

• 对于音频 $x_i$ 的预测文本是：$\hat{y}=\arg {\max }_{y_j\in Y_{\text{test}}}{\pi }_{x_i,y_j}^{\epsilon ,c}$

  • 其中 ${\pi }_{x_i,y_j}^{\epsilon ,c}$ 是最优传输矩阵中的某一项，代表 $x_i$ 与 $y_j$ 的匹配概率。

  • 这个过程也可以对称地用于文本找音频（text-to-audio retrieval）。

3. DEEP AUDIO-TEXT MATCHING VIA MINI-BATCH LEARNING TO MATCH

3.1 MINI-BATCH LEARNING TO MATCH

3.1.1 定义与目标函数

Definition 1: 给定训练数据 $D=\{(x_i,y_i){\}}_{i=1}^n$，m-LTM 目标是最小化 mini-batch 版本的 KL 散度：

• KaTeX parse error: \tag works only in display equations

  • $b$：mini-batch 大小

  • ${\hat{\pi }}^b$：理想匹配矩阵（对角线为 $1/b$，其余为 0）

  • ${\pi }_{X^b,Y^b}^{\epsilon ,c}$：基于熵正则最优传输得到的匹配计划（见公式3）

• 和 LTM 不同：这里的最小化不是 over 全数据集，而是在 mini-batch 上完成，提升了可扩展性。

3.1.2 神经网络参数化

• 度量函数 $c_{\theta ,\varphi }(x,y)$ 被参数化为两个编码器输出的距离：$c_{\theta ,\varphi }(x,y)=d(f_{\theta }(x),g_{\varphi }(y))$

  • $f_{\theta }$, $g_{\varphi }$：神经网络编码器

  • $d(\cdot ,\cdot )$：通常使用 L2 距离或其他度量函数

• 最终目标转化为神经网络参数空间上的优化问题：KaTeX parse error: \tag works only in display equations

3.1.3 随机梯度估计

• 为训练上述目标函数，采用 mini-batch SGD 方法：
  $${\nabla }_{(\theta ,\varphi )}{\mathbb{E}}_{(X^b,Y^b)}\left[\text{KL}({\hat{\pi }}^b\Vert {\pi }^{\epsilon ,c_{\theta ,\varphi }})\right]\approx \frac{1}{B}\sum _{i=1}^B{\nabla }_{(\theta ,\varphi )}\text{KL}({\hat{\pi }}^b\Vert {\pi }_{(X^b,Y^b{)}_i}^{\epsilon ,c})\text{\hspace{1em}(8)}$$

  • 通常 $B=1$ 即单个 mini-batch 训练，计算效率高。

3.1.4 检索阶段

• 检索阶段数据量远小于训练集，并且需要将每个查询样本与整个测试集匹配，因此 不再使用 mini-batch 配置，而是回到 2.2节所述的完整最优传输计算流程。

3.2 MAHALANOBIS-ENHANCED GROUND METRIC

• 动机：
  • 传统的距离函数如 L2 可能在不同维度上缩放不一致（编码器输出未对齐）。
  • 为了解决这个问题，引入了更泛化的度量形式：Mahalanobis 距离。

3.2.1 定义2：Mahalanobis 度量

$$c_{\theta ,\varphi ,M}(x,y)=\sqrt{(f_{\theta }(x_i)-g_{\varphi }(y_j){)}^{\top }M(f_{\theta }(x_i)-g_{\varphi }(y_j))}\text{\hspace{1em}(9)}$$

• $M$：对称正定矩阵，用于调节不同维度的缩放与方向

• 该距离具备更强的表征能力

3.2.2 m-LTM优化目标

$$\underset{(\theta ,\varphi ,M)\in \Theta \times \Phi \times \mathcal{M}}{\min }\mathbb{E}\left[\text{KL}({\hat{\pi }}^b\Vert {\pi }^{\epsilon ,c_{\theta ,\varphi ,M}})\right]\text{\hspace{1em}(10)}$$

3.2.3 Hybrid SGD 优化方法

• 核心思想

  • 对无约束变量 $\theta ,\varphi$：使用普通 SGD 或 Adam 更新
  • 对受约束变量 $M$：使用 Projected Gradient Descent (PGD)，即更新后将其投影回正定矩阵空间

• 因为 $\theta ,\varphi$ 无约束，可以直接用 SGD，但 $M$ 是正定矩阵，需要额外处理。

  • $M$必须保持正定，不能随意更新

  • 对 $M$ 使用 投影梯度下降（Projected Gradient Descent）：

    • 先更新 $M$： $M^{\prime }=F(M,{\nabla }_M)$
    • 再投影回正定空间：$M=\text{Proj}(M^{\prime })$

  • 投影方法：

    • 设 $M=US{V}^{\top }$（SVD分解）

    • 将所有奇异值非负化：$\bar{S}=\max (S,0)$

    • 构造 $\text{Proj}(M)=U\bar{S}{V}^{\top }$

3.3 PARTIAL OT FOR NOISY CORRESPONDENCE

3.3.1 问题设定：训练集有误配对

• 假设数据集 $D=\{(x_i,y_i){\}}_{i=1}^N$ 中有一部分样本 $N_{\text{cor}}<N$ 被打乱（shuffle），但我们并不知道哪些是错的。

• 因此，理想匹配矩阵 $\hat{\pi }$ 变成了 不完整匹配 $\bar{\pi }$，存在行中为 0 的情况。

• 传统 OT 的局限性

  • 最优传输要求行列边缘为固定分布，即每个样本都必须参与匹配，这对噪声数据来说是问题，因为错误样本也被强迫匹配。

3.3.2 Partial Optimal Transport（POT）

• **概述

  • 为解决上述问题，引入部分最优传输，它放宽了传输守恒约束，只要求部分质量进行传输。

  • 新的目标函数（使用 noisy 数据）为：
    KaTeX parse error: Can't use function '$' in math mode at position 180: …right] \tag{12}$̲

    • 其中：
      • ${\pi }^{s,\epsilon ,c}$：部分最优传输解，定义如下：KaTeX parse error: \tag works only in display equations
        • ${\Pi }_s$：POT 的可行集合，满足：$\pi 1\le P_X,{\pi }^{\top }1\le P_Y,1^{\top }\pi 1=s$

  • 即：总传输质量为 $s$，不强制每个样本都参与。

• 求解方法

  • 该部分最优传输问题可通过Bregman 投影进行求解，能有效筛除掉 mini-batch 中的错误匹配。

• 直观解释

  • POT 会优先保留真实匹配对的传输质量，而对明显错误（shuffle导致的）匹配对则不予匹配，从而提升鲁棒性。

4. Experiment

4.1 实验目的

评估 m-LTM 框架在音频-文本匹配任务中的以下几方面表现：

1. 联合嵌入空间的表达能力与迁移能力
2. 模态鸿沟的缩小效果
3. 在零样本声音事件检测中的泛化性能
4. 对训练数据中 noisy 对应（配对错误）的鲁棒性
5. 不同超参数配置下的消融分析

4.2 实验设置

• 数据集：
  • AudioCaps 和 Clotho：用于音频-文本检索（A2T/T2A）
  • ESC-50：用于零样本声音事件检测 + 模态鸿沟分析
• 评价指标：
  • Recall@k (R@1, R@5, R@10)：交叉模态检索标准指标，越高越好
  • Modality Gap Metric：衡量嵌入空间中音频与文本平均差距，越低越好
  • mAP（仅在 ESC-50 上用于事件检测任务）
• 对比方法：
  • 经典：Triplet loss、Contrastive loss
  • SOTA 模型：如 Mei et al. 2022, Wu et al. 2022b 等

4.3 实验结果