轻量级网络提升脑电听觉注意力检测精度与泛化能力
ListenNet: A Lightweight Spatio-Temporal Enhancement Nested Network for Auditory Attention Detection
- 设计嵌套结构捕捉脑电信号时空依赖关系
- 多尺度时序增强与动态注意力机制提升性能
- 参数量减少7倍,适合资源受限场景
听觉注意力检测(AAD)旨在从脑电信号(如EEG)中识别多说话人环境中关注说话人的方向。现有基于EEG的AAD方法忽视了脑电信号的时空依赖性,限制了其解码与泛化能力。为此,本文提出轻量级时空增强嵌套网络ListenNet,包含三个核心组件:时空依赖编码器(STDE)、多尺度时序增强(MSTE)和跨嵌套注意力(CNA)。STDE重建通道间连续时间窗口的依赖关系,提升动态模式提取鲁棒性;MSTE在多尺度上捕获时序特征,表征细粒度与长程时序模式;CNA通过新型动态注意力机制更有效整合层级特征,捕获深层时空关联。在三个公开数据集上的实验表明,ListenNet在个体内与极具挑战性的个体间设置下均优于现有最先进方法,同时可将可训练参数减少约7倍。代码已开源。
原文摘要 · Abstract (English)
Auditory attention detection (AAD) aims to identify the direction of the attended speaker in multi-speaker environments from brain signals, such as Electroencephalography (EEG) signals. However, existing EEG-based AAD methods overlook the spatio-temporal dependencies of EEG signals, limiting their decoding and generalization abilities. To address these issues, this paper proposes a Lightweight Spatio-Temporal Enhancement Nested Network (ListenNet) for AAD. The ListenNet has three key components: Spatio-temporal Dependency Encoder (STDE), Multi-scale Temporal Enhancement (MSTE), and Cross-Nested Attention (CNA). The STDE reconstructs dependencies between consecutive time windows across channels, improving the robustness of dynamic pattern extraction. The MSTE captures temporal features at multiple scales to represent both fine-grained and long-range temporal patterns. In addition, the CNA integrates hierarchical features more effectively through novel dynamic attention mechanisms to capture deep spatio-temporal correlations. Experimental results on three public datasets demonstrate the superiority of ListenNet over state-of-the-art methods in both subject-dependent and challenging subject-independent settings, while reducing the trainable parameter count by approximately 7 times. Code is available at:https://github.com/fchest/ListenNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。