mlsanityData to 14 Oct 2026

ArtifactGen: Benchmarking WGAN-GP vs Diffusion for Label-Aware EEG Artifact Synthesis

arXiv 2509.08188 ↗First posted 9 Sep 2025cs.LG cs.NE q-bio.NCcited by 0

Not on the map. Why?

Abstract (arXiv)

Artifacts in electroencephalography (EEG) -- muscle, eye movement, electrode, chewing, and shiver -- confound automated analysis yet are costly to label at scale. We study whether modern generative models can synthesize realistic, label-aware artifact segments suitable for augmentation and stress-testing. Using the TUH EEG Artifact (TUAR) corpus, we curate subject-wise splits and fixed-length multi-channel windows (e.g., 250 samples) with preprocessing tailored to each model (per-window min-max for adversarial training; per-recording/channel z-score for diffusion). We compare a conditional WGAN-GP with a projection discriminator to a 1D denoising diffusion model with classifier-free guidance, and evaluate along three axes: (i) fidelity via Welch band-power deltas (Deltaδ, Deltaθ, Deltaα, Deltaβ), channel-covariance Frobenius distance, autocorrelation L2, and distributional metrics (MMD/PRD); (ii) specificity via class-conditional recovery with lightweight kNN/classifiers; and (iii) utility via augmentation effects on artifact recognition. In our setting, WGAN-GP achieves closer spectral alignment and lower MMD to real data, while both models exhibit weak class-conditional recovery, limiting immediate augmentation gains and revealing opportunities for stronger conditioning and coverage. We release a reproducible pipeline -- data manifests, training configurations, and evaluation scripts -- to establish a baseline for EEG artifact synthesis and to surface actionable failure modes for future work.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 48 of its 80 references that are in the map. First posted on arXiv 9 Sep 2025.

Builds on

48 of 80 references.

References · 48

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  4. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  5. 2020Denoising Diffusion Implicit Modelscited by 9,781
  6. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  7. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
  8. 2022Scalable Diffusion Models with Transformerscited by 6,054
  9. 2022Classifier-Free Diffusion Guidancecited by 5,847
  10. 2022Flow Matching for Generative Modelingcited by 5,702
  11. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
  12. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
36 more
  1. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  2. 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
  3. 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,898
  4. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
  5. 2017Improved Training of Wasserstein GANscited by 2,235
  6. 2023Consistency Modelscited by 1,822
  7. 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
  8. 2018Demystifying MMD GANscited by 1,293
  9. 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
  10. 2019AugMix: A Simple Data Processing Method to Improve Robustness and Uncertaintycited by 901
  11. 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 864
  12. 2023Extracting Training Data from Diffusion Modelscited by 732
  13. 2016EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfacescited by 628
  14. 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 622
  15. 2018Assessing Generative Models via Precision and Recallcited by 406
  16. 2019Parallel WaveGAN: A fast waveform generation model based on generative adversarial networks with multi-resolution spectrogramcited by 340
  17. 2017Real-valued (Medical) Time Series Generation with Recurrent Conditional GANscited by 333
  18. 2016Revisiting Classifier Two-Sample Testscited by 268
  19. 2018cGANs with Projection Discriminatorcited by 254
  20. 2018Adversarial Audio Synthesiscited by 225
  21. 2018Fr'echet Audio Distance: A Metric for Evaluating Music Enhancement Algorithmscited by 225
  22. 2019Deep learning-based electroencephalography analysis: a systematic reviewcited by 156
  23. 2023Are Diffusion Models Vulnerable to Membership Inference Attacks?cited by 136
  24. 2022Differentially Private Diffusion Modelscited by 119
  25. 2020Uncovering the structure of clinical EEG signals with self-supervised learningcited by 109
  26. 2023Membership Inference Attacks against Diffusion Modelscited by 86
  27. 2018The Temple University Hospital Seizure Detection Corpuscited by 83
  28. 2023Diffusion Models for Time Series Applications: A Surveycited by 72
  29. 2022Transformer Convolutional Neural Networks for Automated Artifact Detection in Scalp EEGcited by 55
  30. 2018EEG-GAN: Generative adversarial networks for electroencephalograhic (EEG) brain signalscited by 48
  31. 2023Reliable Generation of Privacy-preserving Synthetic Electronic Health Record Time Series via Diffusion Modelscited by 24
  32. 2023EEG Synthetic Data Generation Using Probabilistic Diffusion Modelscited by 8
  33. 2022Energy-Efficient Tree-Based EEG Artifact Detectioncited by 6
  34. 2024Synthesizing EEG Signals from Event-Related Potential Paradigms with Conditional Diffusion Modelscited by 6
  35. 2024Improving EEG Classification Through Randomly Reassembling Original and Generated Data with Transformer-based Diffusion Modelscited by 4
  36. 2023Image-based Data Representations of Time Series: A Comparative Analysis in EEG Artifact Detectioncited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025D4PM: A Dual-branch Driven Denoising Diffusion Probabilistic Model with Joint Posterior Diffusion Sampling for EEG Artifacts Removal
  2. 2025A Survey on Bridging EEG Signals and Generative AI: From Image and Text to Beyond
  3. 2021High Frequency EEG Artifact Detection with Uncertainty via Early Exit Paradigm
  4. 2026PRISM: Exploring Heterogeneous Pretrained EEG Foundation Model Transfer to Clinical Differential Diagnosis
  5. 2024Enhancing EEG Signal Generation through a Hybrid Approach Integrating Reinforcement Learning and Diffusion Models
  6. 2026MontageFlow: Physically Valid High Fidelity EEG Generation
  7. 2025Synthetic EEG Generation using Diffusion Models for Motor Imagery Tasks
  8. 2024Guess What I Think: Streamlined EEG-to-Image Generation with Latent Diffusion Models