Machine Unlearning for Seizure Prediction

被引:0
|
作者
Shao, Chenghao [1 ,2 ]
Li, Chang [1 ,2 ]
Song, Rencheng [1 ,2 ]
Liu, Xiang [3 ]
Qian, Ruobing [3 ]
Chen, Xun [4 ,5 ]
机构
[1] Hefei Univ Technol, Dept Biomed Engn, Hefei 230009, Peoples R China
[2] Hefei Univ Technol, Sch Instrument Sci & Optoelect Engn, Anhui Prov Key Lab Measuring Theory & Precis Instr, Hefei 230009, Anhui, Peoples R China
[3] Univ Sci & Technol China, Affiliated Hosp USTC 1, Dept Neurosurg, Div Life Sci & Med,Epilepsy Ctr, Hefei 230001, Anhui, Peoples R China
[4] Univ Sci & Technol China, Affiliated Hosp USTC 1, Dept Neurosurg, Div Life Sci & Med, Hefei 230001, Anhui, Peoples R China
[5] Univ Sci & Technol China, Dept Elect Engn & Informat Sci, Hefei 230001, Anhui, Peoples R China
基金
中国国家自然科学基金;
关键词
Data models; Brain modeling; Databases; Electroencephalography; Dogs; Task analysis; Epilepsy; Electroencephalogram (EEG); knowledge distillation (KD); machine unlearning; seizure prediction;
D O I
10.1109/TCDS.2024.3395663
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, companies and organizations have been required to provide individuals with the right to be forgotten to alleviate privacy concerns. In machine learning, this requires researchers not only to delete data from databases but also to remove data information from trained models. Thus, machine unlearning is becoming an emerging research problem. In seizure prediction field, prediction applications are established most on private electroencephalogram (EEG) signals. To provide the right to be forgotten, we propose a machine unlearning method for seizure prediction. Our proposed unlearning method is based on knowledge distillation using two teacher models to guide the student model toward achieving model-level unlearning objective. One teacher model is used to induce the student model to forget data information of patients with unlearning request (forgetting patients), while the other teacher model is used to enable the student model to retain data information of other patients (remaining patients). Experiments were conducted on CHBMIT and Kaggle databases. Results show that our proposed unlearning method can effectively make trained ML models forget the information of forgetting patients and maintain satisfactory performance on remaining patients. To the best of our knowledge, it is the first work of machine unlearning in seizure prediction field.
引用
收藏
页码:1969 / 1981
页数:13
相关论文
共 50 条
  • [1] Machine unlearning
    Agarwal, Shubham
    NEW SCIENTIST, 2023, 246 (3463) : 40 - 43
  • [2] Coded Machine Unlearning
    Aldaghri, Nasser
    Mahdavifar, Hessam
    Beirami, Ahmad
    IEEE ACCESS, 2021, 9 : 88137 - 88150
  • [3] Adaptive Machine Unlearning
    Gupta, Varun
    Jung, Christopher
    Neel, Seth
    Roth, Aaron
    Sharifi-Malvajerdi, Saeed
    Waites, Chris
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 34 (NEURIPS 2021), 2021, 34
  • [4] A Review on Machine Unlearning
    Zhang H.
    Nakamura T.
    Isohara T.
    Sakurai K.
    SN Computer Science, 4 (4)
  • [5] Machine Unlearning: A Survey
    Xu, Heng
    Zhu, Tianqing
    Zhang, Lefeng
    Zhou, Wanlei
    Yu, Philip S.
    ACM COMPUTING SURVEYS, 2024, 56 (01)
  • [6] Machine Learning for Seizure Prediction: A Revamped Approach
    Kumar, Sai A.
    Nigam, Lavi
    Karnam, Deepthi
    Murthy, Sreerama K.
    Fedorovych, Petro
    Kalidindi, Vasu
    2015 INTERNATIONAL CONFERENCE ON ADVANCES IN COMPUTING, COMMUNICATIONS AND INFORMATICS (ICACCI), 2015, : 1159 - 1164
  • [7] Machine Unlearning for Document Classification
    Kang, Lei
    Ali Souibgui, Mohamed
    Yang, Fei
    Gomez, Lluis
    Valveny, Ernest
    Karatzas, Dimosthenis
    DOCUMENT ANALYSIS AND RECOGNITION-ICDAR 2024, PT IV, 2024, 14807 : 90 - 102
  • [8] Backdoor Defense with Machine Unlearning
    Liu, Yang
    Fan, Mingyuan
    Chen, Cen
    Liu, Ximeng
    Ma, Zhuo
    Wang, Li
    Ma, Jianfeng
    IEEE CONFERENCE ON COMPUTER COMMUNICATIONS (IEEE INFOCOM 2022), 2022, : 280 - 289
  • [9] Unlearning during Learning: An Efficient Federated Machine Unlearning Method
    Gul, Hanlin
    Zhu, Gongxi
    Zhang, Jie
    Zhao, Xinyuan
    Han, Yuxing
    Fan, Lixin
    Yang, Qiang
    PROCEEDINGS OF THE THIRTY-THIRD INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2024, 2024, : 4035 - 4043
  • [10] Machine Unlearning, A Comparative Analysis
    Doughan, Ziad
    Itani, Sari
    ENGINEERING APPLICATIONS OF NEURAL NETWORKS, EANN 2024, 2024, 2141 : 558 - 568