# Removing Neural Signal Artifacts with Autoencoder-Targeted Adversarial Transformers (AT-AT)

arXiv: https://arxiv.org/abs/2502.05332 · HTML: https://mlsanity.com/paper/2502.05332/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- Builds on 15 of its 32 references that are in the map.
- First posted on arXiv 7 Feb 2025.

## Builds on (most cited first, 15 of 15)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [Generative Adversarial Networks: An Overview](https://mlsanity.com/paper/1710.07035/) (2017)
- [Generative Adversarial Transformers](https://mlsanity.com/paper/2103.01209/) (2021)
- [EEGdenoiseNet: A benchmark dataset for end-to-end deep learning solutions of EEG denoising](https://mlsanity.com/paper/2009.11662/) (2020)
- [Characterizing Complex Networks with Forman-Ricci Curvature and Associated Geometric Flows](https://mlsanity.com/paper/1607.08654/) (2016)
- [An Accurate EEGNet-based Motor-Imagery Brain-Computer Interface for Low-Power Edge Computing](https://mlsanity.com/paper/2004.00077/) (2020)
- [EEGFuseNet: Hybrid Unsupervised Deep Feature Characterization and Fusion for High-Dimensional EEG with An Application to Emotion Recognition](https://mlsanity.com/paper/2102.03777/) (2021)
- [Curve Your Attention: Mixed-Curvature Transformers for Graph Representation Learning](https://mlsanity.com/paper/2309.04082/) (2023)
- [Curvature-based Clustering on Graphs](https://mlsanity.com/paper/2307.10155/) (2023)
- [A novel convolutional neural network model to remove muscle artifacts from EEG](https://mlsanity.com/paper/2010.11709/) (2020)
- [EEGDnet: Fusing Non-Local and Local Self-Similarity for 1-D EEG Signal Denoising with 2-D Transformer](https://mlsanity.com/paper/2109.04235/) (2021)
- [MindBigData 2022 A Large Dataset of Brain Signals](https://mlsanity.com/paper/2212.14746/) (2022)
- [Building Brain Invaders: EEG data of an experimental validation](https://mlsanity.com/paper/1905.05182/) (2019)
- [Targeted Adversarial Denoising Autoencoders (TADA) for Neural Time Series Filtration](https://mlsanity.com/paper/2501.04967/) (2025)
- [EEGDiR: Electroencephalogram denoising network for temporal information storage and global modeling through Retentive Network](https://mlsanity.com/paper/2404.15289/) (2024)

## Built on by (most cited first, 1 of 1)

- [A Statistical Mixture-of-Experts Framework for EMG Artifact Removal in EEG: Empirical Insights and a Proof-of-Concept Application](https://mlsanity.com/paper/2509.19385/) (2025)

## Abstract (arXiv)

Electromyogenic (EMG) noise is a major contamination source in EEG data that can impede accurate analysis of brain-specific neural activity. Recent literature on EMG artifact removal has moved beyond traditional linear algorithms in favor of machine learning-based systems. However, existing deep learning-based filtration methods often have large compute footprints and prohibitively long training times. In this study, we present a new machine learning-based system for filtering EMG interference from EEG data using an autoencoder-targeted adversarial transformer (AT-AT). By leveraging the lightweight expressivity of an autoencoder to determine optimal time-series transformer application sites, our AT-AT architecture achieves a >90% model size reduction compared to published artifact removal models. The addition of adversarial training ensures that filtered signals adhere to the fundamental characteristics of EEG data. We trained AT-AT using published neural data from 67 subjects and found that the system was able to achieve comparable test performance to larger models; AT-AT posted a mean reconstructive correlation coefficient above 0.95 at an initial signal-to-noise ratio (SNR) of 2 dB and 0.70 at -7 dB SNR. Further research generalizing these results to broader sample sizes beyond these isolated test cases will be crucial; while outside the scope of this study, we also include results from a real-world deployment of AT-AT in the Appendix.
