2022年05期 v.43 13-16+29页
林家荣 郑凯文 甘兆明 谢仕宇 郑君锐
(广东工业大学,广东 广州 510006)
摘要:针对基于双向门控循环单元(BiGRU)的心肺音分离模型存在的参数量大、算法繁杂、训练成本高、硬件算力要求高等问题,提出基于知识蒸馏的心肺音分离模型。该模型基于BiGRU,教师网络、学生网络分别采用3层、1层BiGRU网络。实验结果表明,采用“教师—学生”知识蒸馏方法可提高学生网络性能,且蒸馏后的“学生+KD”心肺音分离网络相比教师网络,模型更小、算法更简单、训练成本更低,为部署到边缘设备以及资源受限的设备提供了理论依据。
关键词:心肺音分离;双向门控循环单元;知识蒸馏;教师网络;学生网络
中图分类号:R318; TN912.3; TP183 文献标识码:A 文章编号:1674-2605(2022)05-0003-05
DOI:10.3969/j.issn.1674-2605.2022.05.003