Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Knowledge Distillation with Graph Neural Networks for Epileptic Seizure Detection
 
conference paper

Knowledge Distillation with Graph Neural Networks for Epileptic Seizure Detection

Zheng, Qinyue
•
Venkitaraman, Arun  
•
Petravic, Simona
Show more
Morales, GD
•
Perlich, C
Show more
January 1, 2023
Machine Learning And Knowledge Discovery In Databases: Applied Data Science And Demo Track, Ecml Pkdd 2023, Pt Vi
European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML PKDD)

Wearable devices for seizure monitoring detection could significantly improve the quality of life of epileptic patients. However, existing solutions that mostly rely on full electrode set of electroencephalogram (EEG) measurements could be inconvenient for every day use. In this paper, we propose a novel knowledge distillation approach to transfer the knowledge from a sophisticated seizure detector (called the teacher) trained on data from the full set of electrodes to learn new detectors (called the student). They are both providing lightweight implementations and significantly reducing the number of electrodes needed for recording the EEG. We consider the case where the teacher and the student seizure detectors are graph neural networks (GNN), since these architectures actively use the connectivity information. We consider two cases (a) when a single student is learnt for all the patients using pre-selected channels; and (b) when personalized students are learnt for every individual patient, with personalized channel selection using a Gumbel-softmax approach. Our experiments on the publicly available Temple University Hospital EEG Seizure Data Corpus (TUSZ) show that both knowledge-distillation and personalization play significant roles in improving performance of seizure detection, particularly for patients with scarce EEG data. We observe that using as few as two channels, we are able to obtain competitive seizure detection performance. This, in turn, shows the potential of our approach in more realistic scenario of wearable devices for personalized monitoring of seizures, even with few recordings.

  • Details
  • Metrics
Type
conference paper
DOI
10.1007/978-3-031-43427-3_33
Web of Science ID

WOS:001156143700033

Author(s)
Zheng, Qinyue
Venkitaraman, Arun  
Petravic, Simona
Frossard, Pascal  
Editors
Morales, GD
•
Perlich, C
•
Ruchansky, N
•
Kourtellis, N
•
Baralis, E
•
Bonchi, F
Date Issued

2023-01-01

Publisher

Springer International Publishing Ag

Publisher place

Cham

Published in
Machine Learning And Knowledge Discovery In Databases: Applied Data Science And Demo Track, Ecml Pkdd 2023, Pt Vi
ISBN of the book

978-3-031-43426-6

978-3-031-43427-3

Volume

14174

Start page

547

End page

563

Subjects

Technology

•

Personalized Seizure Detection

•

Graph Neural Networks

•

Knowledge Distillation

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LTS4  
Event nameEvent placeEvent date
European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML PKDD)

Turin, ITALY

SEP 18-22, 2023

FunderGrant Number

PEDESITE project

193813

Available on Infoscience
February 23, 2024
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/205480
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés