[發明專利]模型訓練方法、裝置、設備和存儲介質

在審

• 申請號: 202011637010.8
• 申請日: 2020-12-31
• 公開（公告）號: CN112734033A
• 公開（公告）日: 2021-04-30
• 發明（設計）人: 朱星華;王健宗
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06N3/08
• 分類號: G06N3/08;G06N3/04
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 王明磊

1.一種模型訓練方法，其特征在于，應用于模型訓練系統中的中央服務器，所述模型訓練系統包括所述中央服務器和至少兩個客戶端，所述方法包括：

向各客戶端發送待訓練模型中共享模塊的結構信息和參數信息；所述待訓練模型包括共享模塊和本地模塊；所述共享模塊用于獲取輸入所述待訓練模型的點云數據的特征值；所述本地模塊用于根據所述特征值得到所述待訓練模型的模型結果；

接收各客戶端返回的共享模塊的參數，所述共享模塊的參數為各所述客戶端根據本地存儲的點云數據對所述待訓練模型進行訓練得到的；

根據各所述共享模塊的參數更新所述待訓練模型。

2.根據權利要求1所述的方法，其特征在于，所述根據各所述共享模塊的參數更新所述待訓練模型，包括：

對各所述共享模塊的參數進行加權平均，得到共享模塊更新參數；

根據所述共享模塊更新參數更新所述待訓練模型。

3.根據權利要求2所述的方法，其特征在于，所述根據所述共享模塊更新參數更新所述待訓練模型之后，所述方法還包括：

向各所述客戶端返回所述共享模塊更新參數。

4.根據權利要求1-3任一項所述的方法，其特征在于，所述向各客戶端發送待訓練模型中共享模塊的結構信息和參數信息，包括：

根據網絡傳輸速度，按照各所述客戶端對應的批次，分批次向各所述客戶端發送所述待訓練模型中共享模塊的結構信息和參數信息。

5.根據權利要求1-3任一項所述的方法，其特征在于，所述本地模塊設置在對應的客戶端上；

所述本地模塊的參數為各所述客戶端根據所述共享模塊的結構信息、所述共享模塊的參數信息、預設在所述客戶端上本地模塊的結構信息和本地模塊的參數信息，以及，所述客戶端上存儲的點云數據對所述待訓練模型進行訓練得到的。

6.根據權利要求5所述的方法，其特征在于，所述方法還包括：

向各所述客戶端發送更新指令，以使各所述客戶端根據所述本地模塊的參數更新所述本地模塊。

7.根據權利要求5所述的方法，其特征在于，所述本地模塊用于根據所述特征值獲取點云數據的分類結果或分割結果。

8.一種模型訓練裝置，其特征在于，應用于模型訓練系統中的中央服務器，所述模型訓練系統包括所述中央服務器和至少兩個客戶端，所述裝置包括：

發送模塊，用于向各客戶端發送待訓練模型中共享模塊的結構信息和參數信息；所述待訓練模型包括共享模塊和本地模塊；所述共享模塊用于獲取輸入所述待訓練模型的點云數據的特征值；所述本地模塊用于根據所述特征值得到所述待訓練模型的模型結果；

接收模塊，用于接收各客戶端返回的共享模塊的參數，所述共享模塊的參數為各所述客戶端根據本地存儲的點云數據對所述待訓練模型進行訓練得到的；

更新模塊，用于根據各所述共享模塊的參數更新所述待訓練模型。

9.一種電子設備，其特征在于，包括存儲器和處理器，所述存儲器存儲有計算機程序，所述處理器執行所述計算機程序時實現如權利要求1至7任一項所述的方法。

10.一種計算機可讀存儲介質，其特征在于，其上存儲有計算機程序，所述計算機程序被處理器執行時實現如權利要求1至7中任一項所述的方法。