当分類器は、従来の言語モデルアーキテクチャを採用しています。入力テキストを受け取り、それをトークン化します。その後、モデルは各トークンを埋め込み(各トークンの意味を表す数値のベクトル)に変換します。
入力はニューラルネットワークを通過し、出力エンベディングが生成されます。分類器ヘッドは、この出力エンベディングを0または1の予測値に変換します。ここで、0は人間のラベル、1はAIのラベルを表します。
初期のモデルはすでにかなり効果的でしたが、精度を最大限に高め、誤検知(人間が作成した文書をAI生成と誤って判定すること)の可能性を最小限に抑えたいと考えました。そのため、AI検出モデル専用のアルゴリズムを開発しました。
初期のデータセットでは、モデルの精度を99%から99.999%に引き上げるのに十分な特徴情報が得られませんでした。モデルはデータ内の初期パターンを素早く学習しますが、人間が書いたテキストとAIが生成したテキストを正確に区別するためには、困難な境界事例を学習する必要があります。
この問題を解決するために、モデルを用いて大規模なデータセットから誤検知を抽出し、再学習を行う前に、これらの追加された「難しい例」を用いて初期の学習データを拡張します。このプロセスを数回繰り返すことで、結果として得られるモデルは、誤検知率をほぼゼロに抑えるだけでなく、保留評価セットにおける全体的な性能も向上させます。