【えぬえすぴー】

NSP(次文予測) とは?

最終更新:
💡 この2つの文、本当につながってる?AIの読解チェック

BERTの事前学習で、2つのテキストが元の文書で続いていたかを二値分類するタスク。次の文を生成するタスクではない。

📌 このページのポイント
NSP:元の文章で続いていた? IsNext A:今日は晴れだ。 B:散歩に行こう。 元の文書の続き NotNext A:今日は晴れだ。 B:猫が寝ている。 Bをランダムに選ぶ BERTで二値分類 IsNext か NotNext か 学習例を作る確率は各50% 予測の確率ではない/例文は説明用
元の続きとランダムなテキストを区別する事前学習。ランダムな文でも話題が関連する場合があり、意味の関連性だけを判定するタスクではありません。
ひよこ ひよこ
次文予測って次の文を書くこと?
ペンギン先生 ペンギン先生
ここでは文章の生成ではなく、2つのテキストが元の文書で続いていたかを当てる分類だよ。BERTの事前学習でMLMと一緒に使われたんだ。
ひよこ ひよこ
何のために学習させるの?
ペンギン先生 ペンギン先生
質問応答や自然言語推論に役立つ、テキスト同士の関係を学ぶ狙いがあるよ。元のBERT論文では、NSPを除くと一部の評価タスクの性能が下がったと報告しているんだ。
ひよこ ひよこ
正解のラベルはどう作るの?
ペンギン先生 ペンギン先生
元の続きからBを取る場合はIsNext、コーパスからランダムにBを取る場合はNotNextとするよ。それぞれ50%の確率で作る。ランダムなBでも、内容が偶然関連することはあるんだ。
ひよこ ひよこ
その50%はAIの予測の確率なの?
ペンギン先生 ペンギン先生
学習例を作る割合だよ。モデルは入力ごとにIsNextかNotNextかを予測するので、予測がいつも半々になるわけではないんだ。
ひよこ ひよこ
NSPがないと文の関係は学べない?
ペンギン先生 ペンギン先生
そんなことはないよ。RoBERTaはNSPを外し、入力のまとめ方なども変更して良い性能を得たんだ。NSPが役立つかは学習方法や評価条件にもよるんだね。
もっと詳しく知りたい人へ

入力の「文A」「文B」はそれぞれ1文だけ?

説明では2文で例示しますが、BERTの入力セグメントは複数の自然な文を含む場合もあります。NSPは2つのセグメントの関係を分類します。

ペンギン
まとめ:ざっくりこれだけ覚えればOK!
「NSP」って出てきたら「元の文章で続いていたかを当てるBERTの学習」と思えばだいたいOK!
📖 おまけ:英語の意味
「Next Sentence Prediction」 = 次の文の予測
💬 next sentenceは次の文という意味で、2つの文が実際に隣り合っていたかを当てるタスクだからこの名前だよ

参考資料

← 用語集にもどる