一种基于多任务学习的硬盘故障预测方法及系统

本发明涉及一种基于多任务学习的硬盘故障预测方法及系统，方法包括：获取多个硬盘的硬盘信息，根据硬盘信息得到训练样本集；基于表征学习网络和度量学习网络，通过训练样本集训练得到分类和回归双任务硬盘故障预测模型；当需要对目标硬盘进行故障预测时，以目标硬盘的硬盘信息作为待预测样本，将待预测样本与训练样本集中的训练样本形成样本输入对；将样本输入对输入分类和回归双任务硬盘故障预测模型，得到目标硬盘的故障预测结果。结合了表征学习网络和度量学习网络，表征学习网络通过学习硬盘的表征特征，以便更好地捕获潜在故障迹象，利用度量学习网络学习度量同厂商、同型号硬盘之间的相似性和差异性，更准确地预测硬盘故障的风险。