Hugging Face Blog·· 2022-11-08精选AI 评分62
Transformers 支持 Contrastive Search 解码:兼顾流畅度与连贯性的文本生成实践
Generating Human-level Text with Contrastive Search in Transformers 🤗
AI 导读
Hugging Face 在 transformers 4.24.0 中集成了对比搜索解码算法,支持 PyTorch 与 TensorFlow 双框架。该方法结合模型预测置信度与表征退化惩罚项,通过计算候选 token 与已有上下文的余弦相似度,避免贪婪搜索引起的文本循环重复以及核采样的语义漂移。
推荐理由
文章通过退化惩罚项平衡模型置信度与表征相似度,为解决大语言模型生成重复与语义不连贯问题提供了兼顾连贯与多样性的实用解码方案。
来源:Hugging Face Blog · huggingface.co