题目/Title:Design of a Hybrid Competition-Cooperation Teacher-Students Model for Single Channel Based Sleep Staging
作者/Author:
Yiqiao Liao,Milin Zhang,Zhihua Wang,Xiang Xie
会议/Conference:ISCAS 2020
地点/Location:Sevilla, Spain
年份/Issue Date:2020.10-21 Oct.
页码/pages:pp. 1 - 5
摘要/Abstract:
A competition-cooperation teacher-student model is proposed for knowledge transfer between channels for sleep staging. The teacher model was trained simultaneously with competitor and cooperator student models. One extra 鈥渃ompetitor鈥?student model was added to the traditional teacher-student model. The performance of the teacher model can be optimized during the competition with the competitor model. The over-fitting of the student model is greatly reduced in cooperation with the teacher model. A three-step competition-cooperation training process was proposed to optimize the performance of both the teacher model and the student models. The experimental results demonstrate a 2.7% and a 1.6% improvement of the accuracy for the teacher model and the student model, respectively. A state-of-the-art sleep staging accuracy of 83.7% with single-channel input and 85.9% with multi-channel inputs was achieved by the proposed algorithm.