학술논문

基于知识蒸馏的心肺音分离模型 / Cardiorespiratory Sound Separation Method Based on Knowledge Distillation
Document Type
Academic Journal
Source
自动化与信息工程 / Automation & Information Engineering. 43(5):13-29
Subject
心肺音分离
双向门控循环单元
知识蒸馏
教师网络
学生网络
Language
Chinese
ISSN
1674-2605
Abstract
针对基于双向门控循环单元(BiGRU)的心肺音分离模型存在的参数量大、算法繁杂、训练成本高、硬件算力要求高等问题,提出基于知识蒸馏的心肺音分离模型.该模型基于BiGRU,教师网络、学生网络分别采用3层、1层BiGRU网络.实验结果表明,采用"教师—学生"知识蒸馏方法可提高学生网络性能,且蒸馏后的"学生+KD"心肺音分离网络相比教师网络,模型更小、算法更简单、训练成本更低,为部署到边缘设备以及资源受限的设备提供了理论依据.