FasTUSS: Faster Task-Aware Unified Source Separation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Paissan, Francesco, Wichern, Gordon, Masuyama, Yoshiki, Aihara, Ryo, Germain, François G., Saijo, Kohei, Roux, Jonathan Le
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918093117194240
author Paissan, Francesco
Wichern, Gordon
Masuyama, Yoshiki
Aihara, Ryo
Germain, François G.
Saijo, Kohei
Roux, Jonathan Le
author_facet Paissan, Francesco
Wichern, Gordon
Masuyama, Yoshiki
Aihara, Ryo
Germain, François G.
Saijo, Kohei
Roux, Jonathan Le
contents Time-Frequency (TF) dual-path models are currently among the best performing audio source separation network architectures, achieving state-of-the-art performance in speech enhancement, music source separation, and cinematic audio source separation. While they are characterized by a relatively low parameter count, they still require a considerable number of operations, implying a higher execution time. This problem is exacerbated by the trend towards bigger models trained on large amounts of data to solve more general tasks, such as the recently introduced task-aware unified source separation (TUSS) model. TUSS, which aims to solve audio source separation tasks using a single, conditional model, is built upon TF-Locoformer, a TF dual-path model combining convolution and attention layers. The task definition comes in the form of a sequence of prompts that specify the number and type of sources to be extracted. In this paper, we analyze the design choices of TUSS with the goal of optimizing its performance-complexity trade-off. We derive two more efficient models, FasTUSS-8.3G and FasTUSS-11.7G that reduce the original model's operations by 81\% and 73\% with minor performance drops of 1.2~dB and 0.4~dB averaged over all benchmarks, respectively. Additionally, we investigate the impact of prompt conditioning to derive a causal TUSS model.
format Preprint
id arxiv_https___arxiv_org_abs_2507_11435
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FasTUSS: Faster Task-Aware Unified Source Separation
Paissan, Francesco
Wichern, Gordon
Masuyama, Yoshiki
Aihara, Ryo
Germain, François G.
Saijo, Kohei
Roux, Jonathan Le
Sound
Audio and Speech Processing
Signal Processing
Time-Frequency (TF) dual-path models are currently among the best performing audio source separation network architectures, achieving state-of-the-art performance in speech enhancement, music source separation, and cinematic audio source separation. While they are characterized by a relatively low parameter count, they still require a considerable number of operations, implying a higher execution time. This problem is exacerbated by the trend towards bigger models trained on large amounts of data to solve more general tasks, such as the recently introduced task-aware unified source separation (TUSS) model. TUSS, which aims to solve audio source separation tasks using a single, conditional model, is built upon TF-Locoformer, a TF dual-path model combining convolution and attention layers. The task definition comes in the form of a sequence of prompts that specify the number and type of sources to be extracted. In this paper, we analyze the design choices of TUSS with the goal of optimizing its performance-complexity trade-off. We derive two more efficient models, FasTUSS-8.3G and FasTUSS-11.7G that reduce the original model's operations by 81\% and 73\% with minor performance drops of 1.2~dB and 0.4~dB averaged over all benchmarks, respectively. Additionally, we investigate the impact of prompt conditioning to derive a causal TUSS model.
title FasTUSS: Faster Task-Aware Unified Source Separation
topic Sound
Audio and Speech Processing
Signal Processing
url https://arxiv.org/abs/2507.11435