Time-Frequency-Based Attention Cache Memory Model for Real-Time Speech Separation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Guo, Li, Kai, Yang, Runxuan, Hu, Xiaolin
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912382120361984
author Chen, Guo
Li, Kai
Yang, Runxuan
Hu, Xiaolin
author_facet Chen, Guo
Li, Kai
Yang, Runxuan
Hu, Xiaolin
contents Existing causal speech separation models often underperform compared to non-causal models due to difficulties in retaining historical information. To address this, we propose the Time-Frequency Attention Cache Memory (TFACM) model, which effectively captures spatio-temporal relationships through an attention mechanism and cache memory (CM) for historical information storage. In TFACM, an LSTM layer captures frequency-relative positions, while causal modeling is applied to the time dimension using local and global representations. The CM module stores past information, and the causal attention refinement (CAR) module further enhances time-based feature representations for finer granularity. Experimental results showed that TFACM achieveed comparable performance to the SOTA TF-GridNet-Causal model, with significantly lower complexity and fewer trainable parameters. For more details, visit the project page: https://cslikai.cn/TFACM/.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13094
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Time-Frequency-Based Attention Cache Memory Model for Real-Time Speech Separation
Chen, Guo
Li, Kai
Yang, Runxuan
Hu, Xiaolin
Sound
Artificial Intelligence
Audio and Speech Processing
Existing causal speech separation models often underperform compared to non-causal models due to difficulties in retaining historical information. To address this, we propose the Time-Frequency Attention Cache Memory (TFACM) model, which effectively captures spatio-temporal relationships through an attention mechanism and cache memory (CM) for historical information storage. In TFACM, an LSTM layer captures frequency-relative positions, while causal modeling is applied to the time dimension using local and global representations. The CM module stores past information, and the causal attention refinement (CAR) module further enhances time-based feature representations for finer granularity. Experimental results showed that TFACM achieveed comparable performance to the SOTA TF-GridNet-Causal model, with significantly lower complexity and fewer trainable parameters. For more details, visit the project page: https://cslikai.cn/TFACM/.
title Time-Frequency-Based Attention Cache Memory Model for Real-Time Speech Separation
topic Sound
Artificial Intelligence
Audio and Speech Processing
url https://arxiv.org/abs/2505.13094