[發(fā)明專利]基于多跳推理機制的代碼摘要生成方法及系統(tǒng)

有效

• 申請?zhí)枺?/td>: 202110426243.1
• 申請日: 2021-04-20
• 公開（公告）號: CN113342343B
• 公開（公告）日: 2022-05-27
• 發(fā)明（設計）人: 呂晨;馬正;李季;高曰秀;呂蕾;劉弘
• 申請（專利權(quán)）人: 山東師范大學
• 主分類號: G06F8/41
• 分類號: G06F8/41;G06F16/36;G06N5/04
• 代理公司: 濟南圣達知識產(chǎn)權(quán)代理有限公司 37221
• 代理人: 黃海麗

本發(fā)明公開了基于多跳推理機制的代碼摘要生成方法及系統(tǒng)，包括：獲取待生成摘要的代碼；對待生成摘要的代碼進行解析，生成抽象語法樹；對抽象語法樹進行特征提取，得到抽象語法樹特征向量；將抽象語法樹特征向量和自然語言表述特征向量，輸入到訓練后的transformer模型中，得到代碼的初始摘要；對代碼的初始摘要和已知的知識圖譜，利用多跳推理算法進行處理，得到代碼的推理摘要；將代碼的初始摘要和代碼的推理摘要進行加權(quán)融合，得到最終的代碼摘要。使得生成的代碼摘要更加通順和準確，方便代碼維護人員的理解。

技術(shù)領域

本發(fā)明涉及人工智能與軟件工程相關技術(shù)領域，特別是涉及基于多跳推理機制的代碼摘要生成方法及系統(tǒng)。

背景技術(shù)

本部分的陳述僅僅是提到了與本發(fā)明相關的背景技術(shù)，并不必然構(gòu)成現(xiàn)有技術(shù)。

隨著互聯(lián)網(wǎng)技術(shù)的快速發(fā)展，各種各樣的開發(fā)軟件層出不窮，并且軟件的功能變得越來越豐富，這隨之帶來的就是代碼實現(xiàn)的規(guī)模越來越龐大，結(jié)構(gòu)越來越復雜，在這種情況下程序員如何對代碼進行有效的維護成為了一個問題。

傳統(tǒng)的幫助代碼維護人員進行代碼理解的方法有兩種：第一種是代碼開發(fā)人員在開發(fā)時對代碼進行代碼注釋和摘要，由于現(xiàn)在的實現(xiàn)軟件功能的代碼越來越龐大，這對代碼開發(fā)人員無疑不是一個負擔；第二種是代碼摘要生成技術(shù)，可以自動對代碼片段進行注釋，幫助開發(fā)人員在開發(fā)過程中更加關注代碼編寫。

發(fā)明人發(fā)現(xiàn)，目前傳統(tǒng)的代碼摘要生成技術(shù)對代碼片段解析出來相應的自然語言描述表達不順暢，甚至有些描述會出現(xiàn)歧義，生成的效果較差，因而無法滿足維護人員對代碼的快速理解。

發(fā)明內(nèi)容

為了解決現(xiàn)有技術(shù)的不足，本發(fā)明提供了基于多跳推理機制的代碼摘要生成方法及系統(tǒng)；

第一方面，本發(fā)明提供了基于多跳推理機制的代碼摘要生成方法；

基于多跳推理機制的代碼摘要生成方法，包括：

獲取待生成摘要的代碼；

對待生成摘要的代碼進行解析，生成抽象語法樹；對抽象語法樹進行特征提取，得到抽象語法樹特征向量；

將抽象語法樹特征向量和自然語言表述特征向量，輸入到訓練后的transformer模型中，得到代碼的初始摘要；

對代碼的初始摘要和已知的知識圖譜，利用多跳推理算法進行處理，得到代碼的推理摘要；

將代碼的初始摘要和代碼的推理摘要進行加權(quán)融合，得到最終的代碼摘要。

第二方面，本發(fā)明提供了基于多跳推理機制的代碼摘要生成系統(tǒng)；

基于多跳推理機制的代碼摘要生成系統(tǒng)，包括：

獲取模塊，其被配置為：獲取待生成摘要的代碼；

特征提取模塊，其被配置為：對待生成摘要的代碼進行解析，生成抽象語法樹；對抽象語法樹進行特征提取，得到抽象語法樹特征向量；

初始摘要生成模塊，其被配置為：將抽象語法樹特征向量和自然語言表述特征向量，輸入到訓練后的transformer模型中，得到代碼的初始摘要；

推理摘要生成模塊，其被配置為：對代碼的初始摘要和已知的知識圖譜，利用多跳推理算法進行處理，得到代碼的推理摘要；

加權(quán)融合模塊，其被配置為：將代碼的初始摘要和代碼的推理摘要進行加權(quán)融合，得到最終的代碼摘要。

第三方面，本發(fā)明還提供了一種電子設備，包括：一個或多個處理器、一個或多個存儲器、以及一個或多個計算機程序；其中，處理器與存儲器連接，上述一個或多個計算機程序被存儲在存儲器中，當電子設備運行時，該處理器執(zhí)行該存儲器存儲的一個或多個計算機程序，以使電子設備執(zhí)行上述第一方面所述的方法。