[發明專利]一種消息傳遞方法和裝置

有效

• 申請號: 202010988129.3
• 申請日: 2020-09-18
• 公開（公告）號: CN112231113B
• 公開（公告）日: 2023-01-06
• 發明（設計）人: 李龍翔;劉羽;楊振宇;于占樂;王倩;邊晴云
• 申請（專利權）人: 蘇州浪潮智能科技有限公司
• 主分類號: G06F9/54
• 分類號: G06F9/54;G06N20/00;H03M7/30
• 代理公司: 北京連和連知識產權代理有限公司 11278
• 代理人: 張濤;楊帆

1.一種消息傳遞方法，其特征在于，包括執行以下步驟：

從基本消息傳遞模塊中提取出消息傳遞接口函數及其使用的消息壓縮算法；

針對每個所述消息傳遞接口函數，以其使用的所述消息壓縮算法的向量作為狀態、以所述狀態的變化作為動作構建動作價值函數，其進一步包括：

使用所述消息傳遞接口函數和所述消息壓縮算法的所有可能的數值對的集合構成壓縮算法向量作為狀態；以及

使用所述消息傳遞接口函數和所述消息壓縮算法的所有可能的變化動作集作為動作；

針對每個所述動作價值函數，使用強化學習算法以所述消息傳遞接口函數在集群環境中的單步運行速度作為獎勵進行迭代訓練，直到所述動作價值函數收斂和/或迭代次數超過閾值，其進一步包括：

初始化所述動作價值函數并根據所述壓縮算法向量設置所述動作價值函數的當前狀態；

從所述變化動作集選擇一個動作對所述當前狀態執行，以獲得獎勵和下一狀態，根據所述獎勵和所述下一狀態更新所述動作價值函數并使用所述下一狀態覆蓋所述當前狀態；

重復執行上一步驟直到所述動作價值函數收斂和/或迭代次數超過閾值；

其中，根據所述獎勵和所述下一狀態更新所述動作價值函數包括：

使用下式進行更新：

Q(p_s,α)＝(1-β)·Q(p_s,α)+β[R+γmax_αQ(p_s′,α)]

其中p_s為當前狀態，α為動作，Q(p_s,a)為動作價值函數，β為學習率，R為獎勵，γ為折扣因子，p_s′為下一狀態，max_αQ(p_s′,α)為下一狀態所采用動作的最大收斂函數值；

根據對應的所述狀態、動作、和收斂的所述動作價值函數確定最優的所述消息傳遞接口函數和相應的所述消息壓縮算法以壓縮并傳遞消息；

其中，所述消息壓縮算法為以下之一：RLE、HUFF、RICE、LZO，或不使用壓縮算法；其中所述消息壓縮算法為RLE、HUFF、RICE或LZO時所述狀態還包括所述消息壓縮算法的壓縮參數。

2.根據權利要求1所述的方法，其特征在于，所述集群環境包括：集群的硬件設備、消息類型、消息的數據壓縮率、執行壓縮的時間、執行解壓的時間。

3.根據權利要求1所述的方法，其特征在于，所述消息傳遞接口函數附接在傳遞消息的進程接口上；所述消息傳遞接口函數在集群環境中的單步運行速度包括所述傳遞消息的進程的實際運行速度。

4.一種消息傳遞裝置，其特征在于，包括：

處理器；和

存儲器，存儲有處理器可運行的程序代碼，所述程序代碼在被運行時依次執行以下步驟：

從基本消息傳遞模塊中提取出消息傳遞接口函數及其使用的消息壓縮算法；

針對每個所述消息傳遞接口函數，以其使用的所述消息壓縮算法的向量作為狀態、以所述狀態的變化作為動作構建動作價值函數，其進一步包括：

使用所述消息傳遞接口函數和所述消息壓縮算法的所有可能的數值對的集合構成壓縮算法向量作為狀態；以及

使用所述消息傳遞接口函數和所述消息壓縮算法的所有可能的變化動作集作為動作；

針對每個所述動作價值函數，使用強化學習算法以所述消息傳遞接口函數在集群環境中的單步運行速度作為獎勵進行迭代訓練，直到所述動作價值函數收斂和/或迭代次數超過閾值，其進一步包括：

初始化所述動作價值函數并根據所述壓縮算法向量設置所述動作價值函數的當前狀態；

從所述變化動作集選擇一個動作對所述當前狀態執行，以獲得獎勵和下一狀態，根據所述獎勵和所述下一狀態更新所述動作價值函數并使用所述下一狀態覆蓋所述當前狀態；

重復執行上一步驟直到所述動作價值函數收斂和/或迭代次數超過閾值；

其中，根據所述獎勵和所述下一狀態更新所述動作價值函數包括：

使用下式進行更新：

Q(p_s,α)＝(1-β)·Q(p_s,α)+β[R+γmax_αQ(p_s′,α)]

其中p_s為當前狀態，α為動作，Q(p_s,a)為動作價值函數，β為學習率，R為獎勵，γ為折扣因子，p_s′為下一狀態，max_αQ(p_s′,α)為下一狀態所采用動作的最大收斂函數值；根據對應的所述狀態、動作、和收斂的所述動作價值函數確定最優的所述消息傳遞接口函數和相應的所述消息壓縮算法以壓縮并傳遞消息；

其中，所述消息壓縮算法為以下之一：RLE、HUFF、RICE、LZO，或不使用壓縮算法；其中所述消息壓縮算法為RLE、HUFF、RICE或LZO時所述狀態還包括所述消息壓縮算法的壓縮參數。