一种基于对比学习的电池故障诊断预训练大模型

本发明涉及电动车辆故障诊断方法领域，具体涉及一种基于对比学习的电池故障诊断预训练大模型，包括，获取车辆运行数据，并筛选充放电数据；选取预设类型的数据，并进行预处理；将每个数据根据时间维度进行切片，形成输入数据；对输入数随机抽取进行增强处理，得到成对不同增强方式的输入矩阵；将一对输入矩阵输入到编码器中，得到两个表征向量，传递至预测头，得到两个输出向量，计算相似度；根据相似度计算对比损失函数；针对下游故障诊断任务，固定上游模型权重参数，将全连接层进行单独读取，待后续训练；利用少量故障标签数据对下游故障诊断任务进行训练，与上游预训练大模型形成端到端的故障诊断预训练大模型。本发明让下游故障诊断精准识别。