ICML 2023 Workshop | Siamese SIREN：隐式神经表征的音频压缩

用户1324186

发布于 2023-09-27 14:56:22

3390

发布于 2023-09-27 14:56:22

来源：ICML 2023 Workshop 论文链接：https://openreview.net/forum?id=AgkMFYcOmM 作者：Luca A Lanzendörfer, Roger Wattenhofer 内容整理：王妍

引言

隐式神经表征 (INRs) 已经成为一种很有前景的表示各种数据模式的方法，包括3D形状、图像和音频。虽然最近的研究已经证明了 INRs 在图像和 3D 形状压缩方面的成功应用，但它们在音频压缩方面的潜力仍未得到充分开发。基于此，本文提出了一项关于使用 INRs 进行音频压缩的初步研究。

Siamese SIREN

INRs

INRs 是一类函数，其中一组函数参数 p 描述一个数据样本 D。INR 是在 D 上训练近似函数 f （音频信号函数 T→R，其中 T 为时间输入域，R 为幅度输出域的神经网络。

尽管 INRs 还不能与视觉和音频领域的其他数据压缩方法竞争，它仍然值得进一步研究。INRs 具有一些有趣的属性，例如对输入数据具有分辨率不变性，这意味着存储大小不随输入大小而缩放，并且能够在推理期间使用任意分辨率重建数据。

SIREN

图1

正弦表示网络，简称 SIREN，是一类特殊的 INR 模型，它使用多层感知器 (MLP) 架构，正弦函数作为其激活函数:

ϕ_i : x_i → sin(ω_i · (W_ix_i + b_i))\quad(1)

其中，

ϕ_i

为网络的第 i 层，

W_i

和

b_i

分别为第 i 层的权重矩阵和偏置向量。频率标度超参数

w_i

有助于 SIREN 更快收敛。设置

w_i

= 30，在音频的情况下

w_0

= 3000。

SIREN INRs 在图像、音频和 3D 几何图形上的表现优于标准的 relu 激活 INRs 。

Siamese SIREN

然而，当使用 INRs（包括 SIREN INRs）重建音频时，会出现一个重大挑战。在图像中，噪声可能存在，但通常不太明显。然而，在音频数据中，由于人类听力的对数特性，即使相对较小的重建误差也会以平稳背景噪声的形式被清晰地感知到。

因此，本文提出了一种新的 SIREN 扩展，即 Siamese SIREN。为了去除重建信号的背景噪声， Siamese SIREN 使用了 Noise Reduce ，这是一种计算信号和噪声估计的频谱图的算法。信号和噪声估计用于计算每个频带的噪声阈值。噪声掩模是基于阈值计算的，然后用来去除噪声。

为了构造降噪的噪声估计，假设一个有噪声重构的信号可以线性分解为真信号 f 和噪声分量 ε。因为 ε 的抽样分布通常是未知的，所以必须估计。在相同的信号 f 上训练两个具有不同随机权值初始化的 INR，得到 f 的两个近似值

f_0

f_1

。使用以下规则来估计噪声信号

ε_•

。

ε_• = α（\hat{f_•} - \frac{\hat{f_0} + \hat{f_1}}{2}）\quad(2)

其中 α 是控制噪声估计幅度的超参数。调整 α 对结果有影响，本文所有实验采用 α = 2。

ε_•

可以为

ε_0

或

ε_1

，本文将

ε_0

作为噪声估计输入到降噪中。

本文提出的 Siamese SIREN 网络不是单纯地训练两个 INR 来估计信号的噪声，从而导致参数数量翻倍，而是合并了一个层的子集，减少了所需参数的数量，同时仍然允许学习信号

f_0

和

f_1

。这样，每个连体双胞胎既拥有共享的层，也拥有仅针对它的层(连体层)。换句话说，共享层形成 INR 网络的公共主干，而连体层充当两个独立的头，如下图所示。

图2

在减少参数内存占用方面，权值量化策略通过转换网络权值来降低 p 值并提高推理吞吐量。这些通常以 32 位浮点精度存储，但通常可以量化为更小的数据类型，例如 8 位整数。量化方案多种多样，常见的两种方法是:训练后量化 (Post-Training Quantization, PTQ) 和量化感知训练 (Quantization-Aware Training, QAT) ，量化后的 QAT 往往具有较低的重构误差。然而，QAT 需要在训练时成为网络的一部分，或者在训练非量化模型后进行微调。PTQ 可以在训练后应用，不需要再训练网络，代价是重建质量稍差。本文使用 PTQ，因为 PTQ 误差不会显著影响主观信号质量。