公开(公告)号：CN114925777A

公开(公告)日：2022-08-19

申请号：CN202210653697.7

申请日：2022-06-09

Applicant: 上海交通大学

Inventor： 胡昊 ,  李润至 ,  张志鹏 ,  戴磊 ,  徐峰 ,  张诗雯 ,  伊涵 ,  贾博 ,  孙延进 ,  李福娟

IPC: G06K9/62 ,  G06F40/289 ,  G06F40/242 ,  G06N3/08

Abstract: 本发明涉及一种基于BERT模型的航空安全报告多原因分类方法，包括以下步骤：S1、构建航空不安全事故的历史数据集；S2、对历史数据集进行数据预处理；S3、构建BERT模型，包括模型微调、模型训练和模型评估；S4、对于训练好的BERT模型实时输入数据集并预测不同层次的事故原因类别。与现有技术相比，本发明能够有效提升事故原因分类表现，预测不同层次的事故原因类别。