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

基本信息

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

摘要

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