我们的分类器采用传统的语言模型架构。它接收输入文本并将其分词。随后,模型将每个词汇转换为一个嵌入向量,该向量由数字组成,用于表示每个词汇的含义。
输入数据经过神经网络处理,生成输出嵌入向量。分类器头将输出嵌入向量转换为 0 或 1 的预测结果,其中 0 代表人工标注,1 代表 AI 标注。
最初的模型已经相当有效,但我们希望进一步提高准确率,并尽可能减少误报(即错误地将人类撰写的文档判定为AI生成的)。为此,我们专门为AI检测模型开发了一种算法。
在初始数据集的基础上,我们的模型缺乏足够的特征信息,无法将准确率从99%提升至99.999%。虽然模型能够快速学习数据中的初始模式,但为了精确区分人类文本和AI生成的文本,它需要接触到一些棘手的边界案例。
我们通过以下方式解决这一问题:利用模型在大型数据集中搜索假阳性结果,并在重新训练前将这些额外的困难样本添加到初始训练集之中。经过数轮迭代后,最终生成的模型不仅实现了接近于零的假阳性率,在保留集上的整体性能也得到了提升。