Skip to content

README.md

Latest commit

 

History

History
19 lines (14 loc) · 988 Bytes

README.md

File metadata and controls

19 lines (14 loc) · 988 Bytes
Raw

DARNet

Code for paper: DARNet: Dual Attention Refinement Network with Spatiotemporal Construction for Auditory Attention Detection

This paper introduces DARNet, a novel AAD network that constructs more expressive spatiotemporal feature representations, by capturing the spatial distribution characteristics of EEG signals.

Sheng Yan, Cunhang fan, Hongyu Zhang, Xiaoke Yang, Jianhua Tao, Zhao Lv. DARNet: Dual Attention Refinement Network with Spatiotemporal Construction for Auditory Attention Detection. In NeurIPS 2024.

Preprocess

  • Please download the AAD dataset for training.
  • The public KUL dataset, DTU dataset and MM-AAD(not yet open) are used in this paper.

Requirements

  • Python3.9
    pip install -r requirements.txt

Run

  • Modify the getData function in myutils.py to match the dataset
  • Using main.py to train and test the model