1. 首页
  2. 编程语言
  3. C
  4. 知识蒸馏在异常检测中的应用完整代码分享

知识蒸馏在异常检测中的应用完整代码分享

上传者: 2023-03-08 12:37:50上传 ZIP文件 17.88KB 热度 8次

如何使用知识蒸馏技术,在不牺牲准确性的前提下,将复杂的源网络压缩成一个更简单的克隆网络,并用于异常检测任务。我们选择了VGG16作为源网络,并将其分为四个部分。通过知识蒸馏,我们将预训练的综合知识提取到克隆网络中,使用梯度实现了精确的异常定位热图。我们的克隆网络相对于源网络更加简单,不仅在准确率方面表现出色,而且具有更快的训练速度和更小的模型尺寸。

用户评论