基于扩散模型的多样性可控文本生成方法和装置

本发明属于自然语言处理的技术领域，具体涉及一种基于扩散模型的多样性可控文本生成方法和装置。所述方法包括：构建包括自然语言提示编码器、文本重构自编码器、控制信息聚类模型和隐空间扩散模型的多样性文本生成模型；在训练阶段，给定参考文本及包含任务需求和控制需求的自然语言提示，并将其输入多样性文本生成模型以进行模型训练；在应用阶段，给定自然语言提示，并将其输入训练好的多样性文本生成模型中，以逐步迭代生成符合相应控制的多样性文本。本发明将自然语言提示编码在隐空间中聚类，将自然语言控制转化成潜在的指令控制类别，根据提示控制类别分布采样控制信息，引导文本生成过程，实现对生成文本的控制。