一種基于BERT預(yù)訓(xùn)練模型的文本摘要生成方法

基本信息

申請?zhí)?/td> CN202110287084.1 申請日 -
公開(公告)號 CN113128214A 公開(公告)日 2021-07-16
申請公布號 CN113128214A 申請公布日 2021-07-16
分類號 G06F40/284;G06F40/30;G06F40/253;G06N3/04;G06N3/08 分類 計算;推算;計數(shù);
發(fā)明人 文凱;周玲玉;楊航;王宗文 申請(專利權(quán))人 重慶信科設(shè)計有限公司
代理機構(gòu) 重慶市恒信知識產(chǎn)權(quán)代理有限公司 代理人 陳棟梁
地址 400065 重慶市南岸區(qū)南山街道崇文路2號
法律狀態(tài) -

摘要

摘要 本發(fā)明請求保護一種基于BERT預(yù)訓(xùn)練模型的文本摘要生成方法。該方法包括:將中文短本文數(shù)據(jù)集進行預(yù)處理;利用BERT雙向編碼特性能更好地獲取全局信息,將數(shù)據(jù)送入BERT預(yù)訓(xùn)練模型中進行訓(xùn)練;將實際需要獲得摘要的原文本輸入BERT預(yù)訓(xùn)練模型,使用訓(xùn)練好的參數(shù)進行訓(xùn)練,獲得最佳詞向量;將得到的高質(zhì)量字向量送入改進后的LeakGAN模型;文本在改進的LeakGAN中進行訓(xùn)練,最終得到摘要輸出。本發(fā)明使生成器生成更加準(zhǔn)確的摘要,提高了摘要的準(zhǔn)確性和流暢性。