基于预训练模型的AI文本检测方法

本申请的实施例涉及AI文本检测技术领域，特别涉及一种基于预训练模型的AI文本检测方法，包括：基于获取到的人类文本数据、AI文本数据和AI润色文本数据，构建原始数据集，并划分为训练数据集和验证数据集，原始数据集中的每个样本数据上均标注有表征文本类型的第一标签和表征使用的AI大语言模型的第二标签；在预训练模型的末尾添加分类层得到分类模型；将训练数据集输入至分类模型中，利用第一标签和分类模型的输出对分类模型进行训练，得到文本分类模型；将训练数据集输入至文本分类模型中，利用第二标签和文本分类模型的输出对文本分类模型进行训练，得到细粒度文本分类模型，该模型能够很好地进行细粒度、低成本、高效的文本检测。