首页 | 本学科首页   官方微博 | 高级检索  
检索        

基于深度贝叶斯蒸馏网络的皮肤病变识别研究
引用本文:董春序,欧译丹,李雪,陈思光.基于深度贝叶斯蒸馏网络的皮肤病变识别研究[J].中国数字医学,2023(4):48-56+89.
作者姓名:董春序  欧译丹  李雪  陈思光
作者单位:1. 南京邮电大学物联网学院;2. 南京医科大学附属妇产医院(南京市妇幼保健院)皮肤科
基金项目:国家自然科学基金(61971235);;中国博士后科学基金(面上一等资助)(2018M630590);;江苏省“333高层次人才培养工程”;;江苏省博士后科研资助计划(2021K501C);
摘    要:目的:解决卷积神经网络无法量化模型的偶然不确定性与认知不确定性问题,优化皮肤病变类型识别机制,降低误诊率。方法:在构建多尺度网络时,基于贝叶斯深度学习网络构建深度贝叶斯蒸馏网络,通过多次采样数据分布方式拟合训练数据模型,量化模型的偶然不确定性与认知不确定性。进一步引入了知识蒸馏对模型进行压缩,构建学生网络模型拟合教师网络的输出,使用教师网络的参数和真实值标签训练学生网络,从而实现对模型参数量与时间的优化。结果:识别准确率与现有相关方案相比提高3.00%~8.00%,达到83.90%,同时参数量减少14.12%,运行时间节约8.70%。结论:基于深度贝叶斯蒸馏网络的皮肤病变识别机制能够显著提高识别准确率,同时减少模型参数量与运行时间。

关 键 词:皮肤病变  深度贝叶斯网络  知识蒸馏  模型压缩
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号