公开(公告)号：CN112434733A

公开(公告)日：2021-03-02

申请号：CN202011290978.8

申请日：2020-11-17

Applicant: 西安交通大学

Inventor： 董小社 ,  王宇菲 ,  王龙翔 ,  王强 ,  李博闻 ,  陈维多 ,  张兴军 ,  伍卫国

IPC: G06K9/62 ,  G06N3/04 ,  G06N3/08 ,  G06F11/22

Abstract: 本发明公开了一种小样本硬盘故障数据生成方法、存储介质及计算设备，基于长短期记忆网络作为生成对抗网络的生成网络，对硬盘SMART数据进行学习，对长短期记忆网络的梯度进行调整并生成用于对抗训练的虚拟硬盘故障数据；对生成的虚拟硬盘故障数据进行甄别，将真实的硬盘故障数据作为训练样本数据集；交替计算生成对抗网络中生成网络G和判别网络D的梯度并调整至收敛，完成小样本硬盘故障数据训练，训练完成后，加载模型中的生成网络G，将初始随机向量传给生成网络G，生成网络G生成虚拟硬盘故障数据。本发明能够有效地生成符合真实硬盘故障数据特点的虚拟硬盘故障数据，扩充硬盘故障数据样本集，提高机器学习算法识别硬盘故障的准确率。