[發(fā)明專利]一種機器學習中規(guī)則模型的建立方法

有效

• 申請?zhí)枺?/td>: 202110005459.0
• 申請日: 2021-01-05
• 公開（公告）號: CN112348202B
• 公開（公告）日: 2021-03-30
• 發(fā)明（設計）人: 傅濤;王力
• 申請（專利權）人: 博智安全科技股份有限公司
• 主分類號: G06N20/00
• 分類號: G06N20/00
• 代理公司: 北京元周律知識產權代理有限公司 11540
• 代理人: 校麗麗

1.一種機器學習中規(guī)則模型的建立方法，其特征在于，所述建立方法包括：

對第一工控協(xié)議通訊行為樣本進行解析，提取第一協(xié)議樣本；

將所述第一協(xié)議樣本輸入機器學習模塊中進行訓練，確定通訊正常庫；

對第二工控協(xié)議通訊行為樣本進行解析，提取第二協(xié)議樣本；

將所述第二協(xié)議樣本輸入所述機器學習模塊中進行訓練；

獲取所述第二協(xié)議樣本的權重，并根據(jù)所述第二協(xié)議樣本的權重更新通訊正常庫和建立通訊異常庫；

所述將所述第一協(xié)議樣本輸入機器學習模塊中進行訓練，確定通訊正常庫，具體為：

將所述第一協(xié)議樣本的權重標記為1，并將所述第一協(xié)議樣本添加至通訊正常庫中；

所述獲取所述第二協(xié)議樣本的權重，并根據(jù)所述第二協(xié)議樣本的權重更新通訊正常庫和建立通訊異常庫，具體包括：

若所述第二協(xié)議樣本存在于所述通訊正常庫中，且其權重不為1，則更新所有權重不為1的第二協(xié)議樣本的權重；并根據(jù)更新后所述第二協(xié)議樣本的權重將所述第二協(xié)議樣本添加至所述通訊正常庫中或所述通訊異常庫中；

若所述第二協(xié)議樣本不存在于所述通訊正常庫中，則將所述第二協(xié)議樣本的權重設置為預設權重，更新所有權重不為1的第二協(xié)議樣本的權重，并將所述第二協(xié)議樣本轉入通訊異常庫中；其中，所述預設權重小于0.5。

2.根據(jù)權利要求1所述的建立方法，其特征在于，所述預設權重為0.1。

3.根據(jù)權利要求1所述的建立方法，其特征在于，所述根據(jù)更新后所述第二協(xié)議樣本的權重將所述第二協(xié)議樣本添加至所述通訊正常庫中或所述通訊異常庫中，具體包括：

若更新后的第二協(xié)議樣本的權重小于閾值權重，則將所述第二協(xié)議樣本轉入所述通訊異常庫中；并且，若所述第二協(xié)議樣本存在于所述通訊正常庫中，將其從所述通訊正常庫中刪除；

若更新后的第二協(xié)議樣本的權重大于或等于所述閾值權重，則將所述第二協(xié)議樣本添加至所述通訊正常庫中；并且，若所述第二協(xié)議樣本存在于所述通訊異常庫中，將其從所述通訊異常庫中刪除。

4.根據(jù)權利要求3所述的建立方法，其特征在于，所述閾值權重為0.5。

5.根據(jù)權利要求1所述的建立方法，其特征在于，所述更新所有權重不為1的第二協(xié)議樣本的權重具體為：

采用信息量權重法重新計算所有權重不為1的第二協(xié)議樣本的權重。

6.根據(jù)權利要求1所述的建立方法，其特征在于，所述對第一工控協(xié)議通訊行為樣本進行解析，提取出第一協(xié)議樣本具體包括：

對第一工控協(xié)議通訊行為樣本的工控協(xié)議通訊報文進行解析，并提取出源IP、目的IP、協(xié)議名稱、協(xié)議控制命令、協(xié)議控制點位和協(xié)議控制值；

將所述源IP、所述目的IP、所述協(xié)議名稱、所述協(xié)議控制命令、所述協(xié)議控制點位和所述協(xié)議控制值打包構成第一協(xié)議樣本。

7.根據(jù)權利要求1所述的建立方法，其特征在于，所述對第二工控協(xié)議通訊行為樣本進行解析，提取出第二協(xié)議樣本具體包括：

對第二工控協(xié)議通訊行為樣本的工控協(xié)議通訊報文進行解析，并提取出源IP、目的IP、協(xié)議名稱、協(xié)議控制命令、協(xié)議控制點位和協(xié)議控制值；

將所述源IP、所述目的IP、所述協(xié)議名稱、所述協(xié)議控制命令、所述協(xié)議控制點位和所述協(xié)議控制值打包構成第二協(xié)議樣本。