P

Pegasus Aeslc

由 google 开发
PEGASUS是基于间隙句子提取的预训练模型,专门用于抽象文本摘要任务。
下载量 21
发布时间 : 3/2/2022

模型简介

PEGASUS是一种基于Transformer架构的预训练模型,通过间隙句子提取的方式进行训练,特别擅长生成高质量的文本摘要。

模型特点

混合与随机化训练
同时在C4和HugeNews数据集上训练,采用随机采样间隔句子比例和重要性分数添加噪声的方式提升模型性能。
多数据集支持
在多个摘要数据集上表现优异,包括xsum、cnn_dailymail、newsroom等。
改进的分词器
更新sentencepiece分词器以支持换行符编码,提升对特定格式文本的处理能力。

模型能力

文本摘要生成
多数据集适配
抽象摘要

使用案例

新闻摘要
新闻文章摘要
对长篇新闻文章生成简洁的摘要
在cnn_dailymail数据集上ROUGE-1得分为44.16
学术论文摘要
论文摘要生成
为学术论文生成结构化的摘要
在arxiv数据集上ROUGE-1得分为44.21
技术文档摘要
专利摘要
为专利文档生成技术摘要
在big_patent数据集上ROUGE-1得分为52.29
AIbase
智启未来,您的人工智能解决方案智库
简体中文