Latent Iterative Refinement for Modular Source Separation
This work addresses computational efficiency for researchers and practitioners in audio source separation, though it appears incremental as it builds on existing modular architectures.
The paper tackles the resource inefficiency of traditional end-to-end source separation models by reformulating training and inference as iterative mappings of latent representations, resulting in reduced memory requirements and dynamic adjustment of processing blocks during inference.
Traditional source separation approaches train deep neural network models end-to-end with all the data available at once by minimizing the empirical risk on the whole training set. On the inference side, after training the model, the user fetches a static computation graph and runs the full model on some specified observed mixture signal to get the estimated source signals. Additionally, many of those models consist of several basic processing blocks which are applied sequentially. We argue that we can significantly increase resource efficiency during both training and inference stages by reformulating a model's training and inference procedures as iterative mappings of latent signal representations. First, we can apply the same processing block more than once on its output to refine the input signal and consequently improve parameter efficiency. During training, we can follow a block-wise procedure which enables a reduction on memory requirements. Thus, one can train a very complicated network structure using significantly less computation compared to end-to-end training. During inference, we can dynamically adjust how many processing blocks and iterations of a specific block an input signal needs using a gating module.