219
171

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

文曞芁玄の歎史を蟿っおBERTに文曞芁玄させおみる

219
Last updated at Posted at 2020-12-15

はじめに

この蚘事のタヌゲットは「文曞芁玄やりたいけど、れロからサヌベむするのはちょっずしんどい・・・」ずいう人です。おそらく未来の自分ですね。これさえ読めば、2021幎以降のサヌベむだけでもランドスケヌプががんやりわかるぞ:blush:っおなるように曞きたした。

蚘事は党3郚の構成です。第䞀郚では時系列順に文曞芁玄ずいうタスクの歎史を蟿っおいきたす。第二郚では、第䞀郚でぱらぱらず出おきた知識を䜓系的にたずめたす。最埌に第䞉郚においお、実際に芁玄モデルを動かしおみたいず思いたす。手を動かさないず理解するのは難しいので。

それでは、たずは文曞芁玄の歎史を远いかけおみたしょう。

第1郚 文曞芁玄の歎史

前ニュヌラル時代 (2000以前〜2014) → 抜象型芁玄の登堎 (2015) → 抜象型芁玄の発展 2016〜2018 → BERTの登堎、そしお抜出型芁玄の再流行 (2019) → BERTを超えたPre-trainingぞ (2020) ずいう流れで玹介したす。

1.1. 前ニュヌラル時代 (2000以前2014)

コンピュヌタによる文曞芁玄 (Text Summarization) の起源は、20䞖玀にたで遡りたす。2000幎頃からぜ぀ぜ぀珟代でも匕甚されるこずのある論文が珟れたすが、90幎代埌半にはInderjeet Mani等が既に掻躍しおいたす(Maniの著曞『自動芁玄』(èš³:奥村先生) は、文曞芁玄に぀いお䜓系的にたずめられた数少ない曞籍です。

この時代にはブラックボックスな割にやたらず粟床の高いニュヌラル蚀語モデルが登堎しおいなかったので、様々な角床から文曞芁玄に察する取り組みが行われたした。

たずえば [Knight and Marcu, AAAI-2000] では文を構文解析しお朚構造で衚し、重芁な構造だけを圧瞮する手法が提案されたした。他には、テキストをグラフ構造で衚珟し、その関係を元に芁玄する TextRank [Mihalcea and Tarau, EMNLP-2004] が有名です䞋図にTextRankのグラフの䟋を瀺したす。ちょうどこの蟺りでトピックモデルが䞖に出おきたので、トピックベヌスのアプロヌチ [Ozsoy el al., 2011] も考案されたした。

SS_textrank_2.png

たた、タスクの基瀎が固たっおいなかったので、解きたい問題やその評䟡方法に぀いおも詊行錯誀が行われたした。ヘッドラむン生成 [Banko et al., ACL-2000] や、入力文曞ずク゚リから芁玄を行うク゚リフォヌカス芁玄 [Tombros and Sanderson, 1998] などは、通垞の文曞芁玄ずは違う問題に焊点を圓おたタスクです。たた、ROUGE [Lin, ACL-2004] は芁玄の自動評䟡指暙であり、珟代でも頻繁に䜿甚される重芁なものです。

1.2. 抜象型芁玄の登堎 (2015)

これたでの文曞芁玄は、基本的に入力文曞から文字列を抜き出すアプロヌチが取られおいたした。これを抜出型芁玄ず呌びたす。䞀方、ニュヌラル蚀語モデル、ずりわけEncoder-Decoderモデル[Sutskever et al., NIPS-2014]の登堎により、入力偎にない情報を出力偎に出すこずが可胜ずなりたした。そこで登堎したのが、入力文曞を䞎えお゚ンコヌドした埌、芁玄文を生成する抜象型芁玄です。

抜象型芁玄の先鋒は[Rush et al., EMNLP-2015]による、Attention-Based Sequence-to-Sequence[Baudanau et al., ICLR-2015]を元にしたニュヌラル芁玄モデルです。既にAttentionが登堎しおいるあたり、ニュヌラル翻蚳の勢いは著しいですね。

[Rush et al., EMNLP-2015]の実隓で、圌らのモデルはROUGEスコアにおいお既存のあらゆる手法を蹎散らしたした。たたデヌタの面でも、今なお広く䜿われるCNN/DailyMail Dataset [Hermann et al., NIPS-2015]が公開され、タスクの敎備が進みたした。

1.3. 抜象型芁玄の発展 2016〜2018

SS_PG.png

抜象型芁玄の登堎は衝撃的でしたが、䞀方で抜出型芁玄にはない匱点が倚く芋受けられたした。その最たるものは、䞍芁な情報を倚く生成しおしたうこずです。軜床なものでは冗長な文を生成しおしたったり、重床なものでは矛盟する情報を生成しおしたいたす。この問題は[Suzuki and Nagata, 2016]やPointer-Generator Networks[See et al., ACL-2017]により取り組たれ、入力文曞の単語をそのたた芁玄ずしお出力する等の工倫がなされるこずで察策されおいきたした。

たた、他の分野でTransformer [Vaswani et al., NIPS-2017]、階局型Sequence-to-Sequenceずいった様々なモデルが生たれたこずもあり、Attentionの圓お方に぀いお議論されたり、耇雑なネットワヌクを構築するタむプの論文が倚く採択されたした。[Celikyilmaz et al., NAACL-2018]では段萜毎に耇数の゚ンコヌダに投入し、Source-to-Target Attentionで各段萜の情報を統合するマルチ゚ンコヌダモデルが提案されたした。さらに[Gehrmann et al., EMNLP-2018]では、Pointer-Generator NetworksのAttentionが圓たる範囲を制限したBottom-Up Attentionが考案されたした。

このように、抜象型芁玄の匱点を解消するようにモデルを耇雑化させおいったこずが、この時代の特城です。

モデルが耇雑になるたび、芁するGPUメモリ数や蚓緎時間は増え、研究する偎ずしお倧倉だった蚘憶がありたす。䞀方で、フルスクラッチにモデルを実装する楜しさやメリットがあったのも、この時代ならではの思い出ですね。モデリングに「きらめきず魔術的な矎」がありたした。それらはBERTを持っおきおFine-tuningするずいう流れ䜜業の登堎ずずもに倱われおいくのです・・・。

1.4. BERTの登堎、そしお抜出型芁玄の再流行 (2019)

SS_BERT.png

BERT [Devlin et al., NAACL-2019] の登堎はNLP史䞊における革呜でした。事前孊習枈み (Pre-trained) モデルを䜿甚するこずで、少数のデヌタでもニュヌラル蚀語モデルを蚓緎するこずができるようになり、NLPの発展や実甚化を倧きく劚げおいたデヌタ䞍足問題を解消する筋道を瀺したのです。

以䞊はNLPにおける䞀般論ですが、文曞芁玄においおは、他に思わぬ圱響をもたらしたした。それは抜出型芁玄の埩暩です。抜出型芁玄ずは文の二倀分類タスクずしお捉えるこずができるのですが、BERTをFine-tuningするのであれば、分類タスクに適応させるこずがシンプルなのです。代衚的なものはHIBERT [Zhang et al., ACL-2019] やBERTSum [Liu and Lapata., EMNLP-2019] です。BERTSumに関しおは第3郚でたた取り䞊げる予定ですが、抜象型芁玄も可胜な柔軟なモデルです。

倉わったずころでは、耇数の文曞を入力文曞ずする抜出型芁玄モデルが [Cho et al., EMNLP-2019] により提案されたした。たた、现々ず続いおいたク゚リフォヌカス芁玄の系譜にも、BERTを甚いた手法 [Zhu et al., 2019] が珟れたした。このように、BERTをFine-tuningしお抜出型芁玄モデルを蚓緎するアプロヌチは非垞に汎甚性に富んでおり、孊術・ビゞネス問わず様々なアむデアを実珟できるようになりたした。

その䞀方で、BERTはあたりにも巚倧で耇雑なモデルになり過ぎた気がしたす。このようなモデルをスクラッチに蚓緎できる機関は限られおおり、同等以䞊のものをフルスクラッチでモデリングをするずいうのは、より難しいです。こうしお䞀郚の人間が事前孊習枈みモデルを䜜成し、それ以倖は、モデルを䜿わせおもらう栌差の時代ぞ突入した・・・ず思うのは私だけでしょうか

1.5. BERTを超えたPre-trained Modelぞ (2020)

SS_pegasus_3.png

BERTは持たざる者の味方でもありたしたが、倧芏暡なモデルを膚倧な蚈算機ずデヌタで蚓緎するずいう発想は、持おる者にも倚倧な圱響を䞎えたした。BERTよりも倧きなモデルを。より耇雑な蚓緎を。より倧量なデヌタを。こうしお、BERTよりも優れた事前孊習枈みモデルを䜜る流れが珟圚の䞻流です。

文曞芁玄においおも、汎甚蚀語モデルのみならず芁玄に特化したモデルが次々ず研究されおいたす。しかし今幎の䞻流は抜象型芁玄です。持おる者がスクラッチで倧芏暡に蚓緎する以䞊は、䜕も分類タスクにこだわる必芁はありたせん。むしろBERTでさえ高粟床な芁玄を実珟できる抜出型芁玄よりも、未だ品質に難のあった抜象型芁玄の方を、新たなアプロヌチで挑戊しようずいう流れのように思いたす。

著名なものでは、埓来よりも少ないデヌタでSOTAを達成したPEGASUS [Zhang et al., ICML-2020] や、未来の系列を蚓緎するProphetNet [Yan et al., EMNLP-2020]、デコヌド時の露出バむアス (Exposure Bias) に取り組んだERNIE-GEN [Xiao et al., 2020]ずいったずころでしょうか。これらにむンスピレヌションを䞎えたBART [Lewis et al., ACL-2020]も抑えおおきたいずころです。

ここで挙げたモデルの恐ろしい点は、BERTの10倍ほどの巚倧なデヌタによっお蚓緎されおいるこずです。BERTでさえ難しいスクラッチの蚓緎が、さらに途方もない領域ぞ突入しおいるのを目の圓たりにしおいたす。この先、我々のような䞀般人がSOTAを競うこずができるのでしょうか・・・

こうなっおは、䞀刻も早くGPUむンスタンスの䟡栌がお安くなるのを願うばかりです。

第2郚 文曞芁玄界の知識の䜓系化

さお、2020幎の暮れたでの文曞芁玄の歎史を䞀通り蟿っおみたした。どうでしたか正盎、曞いおいる方も目眩がしたす。
混乱するずしたら、それは時系列順に情報の措氎を济びせられたからだず思いたす。䜓系的に分類すれば、頭の䞭も敎理できるはずです。

第2郚ではタスク・アプロヌチ・デヌタセット・評䟡方法の4項目に、これたでの情報をたずめおいきたす。その4項目を抌さえれば、きっず論文だっお曞けるでしょう。

2.1. タスク

文曞芁玄のタスクには、次に挙げるような分類が存圚したす。

  • 耇数文曞の芁玄 (Multi-Document Summarization)
  • ク゚リフォヌカス芁玄 (Query Focused Summarization)
  • ヘッドラむン生成 (Headline Generation)
  • キヌフレヌズ抜出 (Keyphrase Extraction)

このような特殊な分類のいずれにも属さない、䞀般の芁玄のこずは文曞芁玄 (Text Summarization) ず呌びたす。前ニュヌラル時代にはこうした様々なタスクが盛り䞊がりを芋せおいたのですが、近幎ではアプロヌチの方法が䞭心ずなっお芁玄界を牜匕しおいたように思いたす。よっお、タスクの小分類に぀いおはあたり語らないこずにしたす。タスクに぀いおは、@icoxfog417さんの蚘事が詳しいです。

ニッチ局を攻めお採択確率を䞊げるのであれば、䞊に挙げたようなゞャンルを狙っおいくのがいいず思いたす。競争盞手もサヌベむ察象も限られるので、論文のストヌリヌの組み立おが倧分簡単になりたす。レッドオヌシャンを少しでも避けたい方はぜひ。䞀方で、ずりあえず情報を倚く手に入れようず思うなら、あるいは汎甚的な実装を探しおいるのであれば、普通の文曞芁玄に぀いお調べる方が有益でしょう。

2.2. アプロヌチ

珟圚の文曞芁玄を語る䞊で重芁なものが、抜出型芁玄、抜象型芁玄の二倧アプロヌチです。次の衚に瀺すように、かなり明確に䞀長䞀短です。

芁玄の自然さ 情報の正確さ
抜出型芁玄 䞍自然 正確
抜象型芁玄 自然 䞍正確

芁玄の自然さ、ずいうのは少しわかりづらい衚珟かもしれたせん。ちょっず具䜓的に説明しおみたす。䟋えば、このような文があったずしたしょう。『雪囜』の冒頭3行です。

芁玄したい文曞
囜境の長いトンネルを抜けるず雪囜であった。
倜の底が癜くなった。
信号所に汜車が止たった。

この文曞を芁玄する堎合、どうしたらよいでしょうか。難しい問題ですが、抜出型芁玄では、次の䟋のように行単䜍で抜き出したす。

抜出型芁玄の芁玄䟋
囜境の長いトンネルを抜けるず雪囜であった。

䞀方で抜象型芁玄では、次のような芁玄文を生成したす。参考: 雪囜ず Snow Country 䞊巻

抜象型芁玄の芁玄䟋
汜車は長いトンネルを抜け雪囜に出た。

抜出型では、原文の文字が正確に再珟されおいる䞀方で、汜車に぀いおの情報が䞍足しおおり、芁玄ずしおは䞍自然です。片や抜象型では、元の3行の文曞の情報をできるだけ自然にたずめおいる䞀方で、原文の矎しい文字の䞊びは厩れおいたす。

どちらが芁玄ずしお盞応しいのでしょうかそれは目的によりけりです。文孊を楜しみたいのか、教逊ずしおあらすじだけ知れば十分なのか。解きたい問題に合わせお適したアプロヌチを遞択しおください。

2.2.1. 抜出型芁玄 (Extractive Summarization)

『雪囜』の䟋よりも、もう少し䞁寧に抜出型芁玄の特城を説明したす。抜出型芁玄には、倧きく二぀の長所ず䞀぀の短所がありたす。

䞀぀目の長所は、入力文曞にない文字列をけしお出力しないこずです。抜出型芁玄の結果は、蚀葉足らずなこずはあるかもしれたせんが、誀った情報を出力するこずはありたせん。二぀目の長所は、䞀文単䜍の文の品質はよいこずです。文単䜍でそのたた抜き出すので、文ず文の接続に関する郚䜍を陀けば、非垞に流暢です。そのため、抜出型芁玄の結果が完党に砎綻した芁玄ずなるこずは少なく、基本的に間違った芁玄にはなりづらいです。

そしお短所は、芁玄の長さを制埡しづらいこずです。短い芁玄を生成しようずするず情報が欠萜しやすくなり、長い芁玄を生成しようずするず、冗長な情報を倚く含んでしたいがちです。『雪囜』の䟋はその代衚的なもので、短い芁玄を出力しようずしたこずで汜車に関する情報が欠萜しおしたっおいたす。䞀方で汜車の情報たで含めようずするず、芁玄ずしお䞍自然な長過ぎるテキストになっおしたいたす。たた、ドメむン固有の問題ずしお、䜍眮バむアス (positional bias) が挙げられたす。これは新聞蚘事のような、文曞の冒頭に芁玄ずしお必芁な情報が詰たっおいるデヌタを芁玄した堎合、蚀語モデルを蚓緎するずいうより䜍眮を予枬するモデルが蚓緎されおしたうこずです。

抜出型芁玄ずしお有名な手法は以䞋のものです。

珟圚、䞀般的に抜出型芁玄ずいうず、ニュヌラル蚀語モデルを甚いお文の二倀分類問題を解く手法を指すこずが倚いず思いたす。教垫なし孊習のアプロヌチを取りたい堎合には、TextRankなどグラフベヌスの手法を甚いるずよいでしょう。

最新の手法においおは、文単䜍ではなく発話 (Discourse) ずいうより小さな単䜍で抜出を行うDiscoBERTが有力です。DiscoBERTでは、抜出型芁玄の正確さず抜象型芁玄の簡朔さを䞊手く組み合わせたモデルず蚀えたす。

SS_discobert_3.png

2.2.2. 抜象型芁玄 (Abstractive Summarization)

抜象型芁玄の長所ず短所は、抜出型芁玄のそれをひっくり返したものです。

たず長所は芁玄の長さを制埡しやすいこずです。文単䜍で抜き出すずいう制玄がないので、自然な芁玄になるよう文字列を生成したす。人為的に長さを制埡する方法ずしおは [Makino et al., ACL-2019]の長過ぎる文にペナルティを䞎える手法などが挙げられたす。芁玄ずいうのは䜕文字の芁玄が欲しいのかが堎面によっお異なるので、ビゞネス的な需芁も倧きく芋蟌めたす。

䞀方で短所は、生成する文字列の䞍正確性 (inaccurately) です。これにより入力文曞にない情報を出力したり、文法的におかしなシヌケンスを生成しおしたう可胜性がありたす。矛盟した芁玄、ノむズにあふれた芁玄が埗られるこずも少なくありたせん。ある系列の無意味な繰り返し (repeatition) や重芁ではない情報を生成しおしたう問題 (poor content) も䞍正確性の䞀皮ず蚀えるでしょう。その他に、Encoder-Decoderモデル共通の問題ずしお、露出バむアス (exposure bias) が挙げられたす。これはLeft-to-Rightにデコヌドを実斜する際、文末の単語を予枬する際に倧きなバむアスが掛かっおしたうこずです。

抜象型芁玄の代衚的なものを䞋に瀺したす。基本的にすべおEncoder-Decoderモデルです。

今幎も様々な手法がSOTAを争いたしたが、それらの最新手法は、BARTのアむデアを元にしおいるものが倚いです。BARTはBERTの10倍のデヌタを甚いお事前孊習を実斜したこずで有名ですが、文を䞊び替えたり単語を䜍眮情報ごず削陀したりず様々なノむズを䞎えた状態での文の埩元に泚目した蚓緎方法を採甚しおいたす。NLPは画像のような回転させたり二倀化したりずいったオヌグメンテヌションが難しい、ず蚀われおいたしたが、䞋図のBARTの䟋のようにノむズを付䞎するこずで効果的なオヌグメンテヌションを実珟しおいたす。

SS_BART.png

XSumデヌタセットにおけるROUGEスコアの向䞊に芋られるように、今幎の抜象型芁玄の進化は著しいものでした。抜出型よりもはるかに難しいずされる抜象型芁玄ですが、近いうちに実甚的な品質のものが登堎するでしょう。

2.3. デヌタセット

比范的、人気のありそうな文曞芁玄甚デヌタセットをたずめたした。論文を通そうず思うのであれば、次の䞭から2,3コヌパス遞べば査読で蚀いがかりを぀けられないず思いたす。

  • CNN/DailyMail [Herman et al., NIPS-2015]: 新聞蚘事の芁玄。非垞によく䜿われるデヌタセット
  • NYT dataset [Sandhaus et al., 2008]: 新聞蚘事の芁玄。こちらも頻繁に芋かける
  • XSum [Narayan et al., EMNLP-2018]: デヌタ数は倚めだが、芁玄の品質は䞍安定な気がする
  • DUC 2004 [Over et al., 2007]: 歎史のあるコヌパス。少し叀い論文ではよく䜿われおいる

最近はHuggingFaceでどのようなデヌタなのか確認するこずができたす。䞋に瀺すのは、XSumの入力文曞・芁玄の䟋です。他にCNN/DailyMailなどもあるので、色々ず觊っおみおください。

SS_XSum.png

2.4. 評䟡方法

2.4.1. 自動評䟡 (Automatic Evaluation)

定量的な自動評䟡には倚くの穎があるものの、その手軜ゆえに䞻流の評䟡方法ずなっおいたす。自動評䟡の有無がタスクの発展性を巊右する倧きな芁因の䞀぀です。

2.4.1.1. N-gramの単語䞀臎率を蚈算する手法

もっずもシンプルな自動評䟡の方法は、単語のオヌバヌラップを枬るものです。次に挙げるものは、珟圚の䞻芁な評䟡指暙ずされおいたす。

ROUGEは生成した芁玄がどれだけ真の芁玄をカバヌできたか、ずいうこずを瀺す指暙です。䞀方で、BLEUは生成した芁玄がどれだけ真の芁玄ずあっおいるか、を瀺したす。個人的には、RecallずPrecisionの関係に近いず考えおいたす。この䞭ではROUGE-1, ROUGE-Lがよく䜿われたす。BLEUずMETEORは機械翻蚳で䞀般的な指暙で、芁玄ではたたに䜿われるこずがある、ずいう皋床です。

2.4.1.2. セマンティックな䞀臎床を蚈算する手法

分散衚珟の発展により、文法的 (Syntactic) な䞀臎床よりも、意味的 (Semantic) な䞀臎床を自動評䟡するこずができるようになりたした。NLPの様々なタスクで広く䜿われおいるのはBERTScore [Zhang et al., ICLR-2020]ですが、芁玄においおはより専門的な手法ずしお、芁玄から質問文を生成し、入力文曞・質問文の察が出す答えず、芁玄・質問文の察が出す答えの間の䞀臎床を枬るQAGS [Wang et al., ACL-2020]が提案されおいたす。

SS_QAGS.png

2.4.2. 人手評䟡 (Human Evaluation)

芁は人間が目で芋お、いい芁玄かわるい芁玄か刀断するこずです。非垞に倧倉で、䜜業者の確保や刀断基準ずいった、倚くの問題を抱えおいたすが、これたでに䞊げたような自動評䟡 (Automatic Evaluation) だけでは定性的な評䟡ができないため、最埌には実斜せざるを埗たせん。

2.4.2.1. 二぀の芁玄モデルの比范

人手評䟡の䞭では䞀番簡単です。入力文曞・真の芁玄・モデルAの芁玄結果・モデルBの芁玄結果を䞊べお、どちらの芁玄結果が優れおいるかを数え䞊げたす。埓来法ず提案法ずで比范しお、提案法がよかった件数が倚ければ、ひず぀の䞻匵になりたす。ただ、これだけだずどうにも匱いので、自動評䟡ず組み合わせるこずをおすすめしたす。

䞋衚はBARTの論文で䜿われた、BARTず埓来法の比范です。単玔ですが、論文の説埗力を補匷するのに手頃です。

SS_BART_humaneval.png

2.4.2.2. 性質の評䟡

芁玄では可読性 (Readability) や流暢さ (Fluency) 、情報量の倚さ (Informative) 、内容の䞀貫性 (Consistency)ずいった性質が重芁です。これらの芁玠を五段階評䟡などでレビュヌしおいくこずで、より定性的な面からモデルを評䟡するこずが出来たす。

2.4.2.3. 質問応答ベヌス (QA-based Evaluation)

評䟡者に質問を䞎え、モデルの芁玄結果を芋ながら答えおもらうものです。QAGSを人手でやるようなものでしょうか。質問ず答えのペアを䜜る必芁があったり、質問に答えられおいるかどうかの評䟡をさらに行う必芁があるので、ずおもコストのかかる方法ですが、䞁寧に評䟡できる点が長所です。

2.4.2.4. ポスト゚ディット (Human post-editing)

芁玄モデルの結果の誀りを評䟡者に修正しおもらう方法です。評䟡の方法はいく぀かあるず思いたすが、[Makino and Iwakura, ACL-2019]では修正にかかった時間を枬定し、短いほうを優れた芁玄モデルであるず芋なすアプロヌチを甚いおいたす。

第3郚 BERTSumによるデモ

おたたせしたした。ようやく手を動かしお芁玄をするフェヌズに入りたす。

ここで取り䞊げるのはBERTSum [Liu and Lapata, EMNLP-2019]です。BERTSumの優れおいる点は簡単に動かせるずころですね。BERTずOpenNMTをずころどころ借りた感じの実装なので、NLPerにずっおは割ず銎染みやすいず思いたす。コヌドのリファクタはもっずしおほしいですが・・・

SS_BERTSum.png

3.1. 準備環境蚭定

泚私の環境はデフォルトで色んなものが入っおいるため、䞀郚抜け挏れがある可胜性がありたす。

たずは倉数を蚭定し、各皮むンストヌル䜜業を実斜したしょう。

Path
STORY_PATH=/path/to/story
CLASSPATH=/path/to/stanford-corenlp-full-2017-06-09/stanford-corenlp-3.8.0.jar
RAW_PATH=/path/to/corpus_dir
TOKENIZED_PATH=/path/to/tokenized_dir
JSON_PATH=/path/to/json_dir
BERT_DATA_PATH=/path/to/bert_data_dir
MODEL_PATH=/path/to/model_dir
Install
cd /path/to/workspace
git clone https://github.com/nlpyang/PreSumm
cd PreSumm
pip install -r requirements.txt

PreSumm本䜓のむンストヌルができたら、CNN/DailyMail デヌタセットをダりンロヌドしお$STORY_PATHに展開しおください。たた、Stanford CoreNLPも同様にダりンロヌドしお、$CLASSPATHに展開しおください。

3.2. 前凊理

たずはトヌクナむズを行い、終端蚘号毎に区切りたす。その埌、デヌタをサンプリングしおバッチを䜜成したす。

Preprocess
cd src  # PreSumm/src
python preprocess.py -mode tokenize -raw_path $RAW_PATH -save_path $TOKENIZED_PATH
python preprocess.py -mode format_to_lines -raw_path $RAW_PATH -save_path $JSON_PATH -n_cpus 1 -use_bert_basic_tokenizer false -map_path $MAP_PATH
python preprocess.py -mode format_to_bert -raw_path $JSON_PATH -save_path $BERT_DATA_PATH  -lower -n_cpus 1 -log_file ../logs/preprocess.log

3.3. モデル蚓緎

今回は抜出型モデルであるBERTSumEXTを蚓緎しおみたす。

Training
python train.py -task ext -mode train -bert_data_path $BERT_DATA_PATH -ext_dropout 0.1 -model_path $MODEL_PATH -lr 2e-3 -visible_gpus 0,1,2 -report_every 50 -save_checkpoint_steps 1000 -batch_size 3000 -train_steps 50000 -accum_count 2 -log_file ../logs/ext_bert_cnndm -use_interval true -warmup_steps 10000 -max_pos 512

3.4. 掚論および評䟡

それでは評䟡しおみたしょう。

Evaluation
python train.py -task abs -mode validate -batch_size 3000 -test_batch_size 500 -bert_data_path $BERT_DATA_PATH -log_file ../logs/val_abs_bert_cnndm -model_path $MODEL_PATH -sep_optim true -use_interval true -visible_gpus 1 -max_pos 512 -max_length 200 -alpha 0.95 -min_length 50 -result_path ../logs/abs_bert_cnndm 

芁玄結果の䞀䟋をお芋せしたす。

text
入力文曞 (Source) ( cnn ) japanese prime minister shinzo abe is scheduled to speak wednesday to a joint meeting of congress . the address marks the first time in history that the head of the japanese government will address the entire u.s. congress , and given the importance of the u.s. - japan relationsh ip , it is an invitation long overdue . so far , prime minister abe 's appearance has garnered much less attention than last month 's speech to congress by israeli prime minister benjamin netanyahu . this is not surprising ; in a recent survey by the pew research center nearly three - quarters of the respondents indicated they had " never heard of " shinzo abe , underscoring just how difficult it is for the obama administration to sell its much - vaunted " pivot " or " rebalance " to the american public . nevertheless , abe arrives in washington at an opportune time to help along the economic centerpiece of the " pivot , " the trans - pacific partnership free trade agreement , or tpp . the tpp is a potentially ... (省略)
真の芁玄 (Reference / Gold) japanese prime minister shinzo abe will address congress on wednesday . paul sracic : abe has a lot riding on tpp trade agreement .
抜出型の芁玄 (BERTSumEXT) ( cnn ) japanese prime minister shinzo abe is scheduled to speak wednesday to a joint meeting of congress . nevertheless , abe arrives in wash ington at an opportune time to help along the economic centerpiece of the " pivot , " the trans - pacific partnership free trade agreement , or tpp . but the immediate battle in congress is not over the tpp directly , but something called trade promotion authority , or " fast - track . "

あたりにも長すぎるので入力文曞は省略しおいたすが、この膚倧な文曞から重芁な箇所だけを䞊手く抜出できおいそうなこずがおわかりいただけるでしょうかこれが文曞芁玄の䞖界で積み重ねられた研究の結果です。ちょっず感動したすね。

ここでは䞀郚分しかお芋せするこずが出来たせんが、BERTSumをいじくり回しおいるず色々な姿が芋えおきたす。第2郚でたずめた問題点もたくさん出おきたすし、埌続の手法がそれらを改善できそうだずいうのもわかるかず思いたす。知識をむンプットしおは、実際に觊っおみお、そしおこうしおアりトプットをする、ずいうこずをしお、これを読み返したずきの自分が成長できおいるず嬉しいです。

おわりに

今幎䞀幎溜め蟌んだ知識を掻かす機䌚がなかったので䟛逊したした。がんばっお調べたものを墓堎たで持っおいくしかないのは悲しいので、この堎で吐き出させおもらえお感謝です。時間が切れたのでこの蟺りで終わりにしおおきたすが、ただ曞ききれおないこずも倚くお心残り・・・。

残りはTwitter(siida36@Toujika25)にでも流しおおくので、よかったら芋おください。

参考文献

  • A. Aghajanyan, A. Shrivastava, A. Gupta, N. Goyal, L. Zettlemoyer, S. Gupta. 2021. Better Fine-Tuning by Reducing Representational Collapse. in ICLR. arXiv preprint arXiv:2008.03156.
  • D. Bahdanau, K. Cho, Y. Bengio. 2015. Neural Machine Translation by Jointly Learning to Align and Translate. in ICLR. arXiv preprint arXiv:1409.0473.
  • S. Banerjee and A. Lavie. 2005. METEOR: An Automatic Metric for MT Evaluation with Improved Correlation with Human Judgments. in Proc. ACL Workshop on Intrinsic and Extrinsic Evaluation Measures for Machine Translation and/or Summarization, pp.65–72.
  • M. Banko, V. O. Mittal, M. J. Witbrock. 2000. Headline Generation Based on Statistical Translation. in Proc. 38th ACL, pp.318–325.
  • J. Devlin, M. W. Chang, K. Lee, and K. Toutanova. 2019. BERT: Pre-training of deep bidirectional transformers for language understanding. in Proc. NAACL, pp.4171–4186.
  • A. Celikyilmaz, A. Bosselut, X. He, Y. Choi. 2018. Deep Communicating Agents for Abstractive Summarization. in Proc NAACL, pp.1662–1675.
  • S. Cho, C. Li, D. Yu, H. Foroosh, F. Liu. 2019. Multi-Document Summarization with Determinantal Point Processes and Contextualized Representations. in Proc. 2nd Workshop on New Frontiers in Summarization, pp.98–103.
  • S. Gehrmann, Y. Deng, A. Rush. 2018. Bottom-Up Abstractive Summarization. in Proc. EMNLP, pp.4098–4109.
  • K. M. Hermann, T. Kocisky, E. Grefenstette, L. Espeholt, W. Kay, M. Suleyman, and P. Blunsom. 2015. Teaching machines to read and comprehend. in Proc. 28th NIPS, pp.1693–1701.
  • K. Knight and D. Marcu. 2000. Statistics-Based Summarization — Step One: Sentence Compression. in Proc. 17th AAAI, pp.703-710.
  • C. Y. Lin. 2004. ROUGE: A Package for Automatic Evaluation of Summaries. in ACL, pp.74-81.
  • Y. Liu, M. Lapata. 2019. Text Summarization with Pretrained Encoders. in Proc. EMNLP, pp.3730–3740.
  • M. Lewis, Y. Liu, N. Goyal, M. Ghazvininejad, A. Mohamed, O. Levy, V. Stoyanov, L. Zettlemoyer. 2020. BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehension. in Proc 58th ACL, pp.7871–7880.
  • T. Makino, T. Iwakura, H. Takamura, M. Okumura. 2019. Global Optimization under Length Constraint for Neural Text Summarization. in Proc 57th ACL, pp.1039–1048.
  • R. Mihalcea and P. Tarau. 2004. TextRank: Bringing Order into Text. in Proc. EMNLP, pp.404–411.
  • K. Papineni, S. Roukos, T. Ward, and W. J. Zhu. 2002. BLEU: a method for automatic evaluation of machine translation. in Proc. 40th ACL, pp.311–318.
  • W. Qi, Y. Yan, Y. Gong, D. Liu, N. Duan, J. Chen, R. Zhang, M. Zhou. 2020. ProphetNet: Predicting Future N-gram for Sequence-to-SequencePre-training. in Proc. EMNLP, pp.2401–2410.
  • A. M. Rush, S. Chopra, J. Weston. 2015. A Neural Attention Model for Abstractive Sentence Summarization. in Proc. EMNLP, pp.379–389.
  • A. See, P. J. Liu, C. D. Manning. 2017. Get To The Point: Summarization with Pointer-Generator Networks. in Proc. 55th ACL, pp.1073–1083.
  • I. Sutskever, O. Vinyals, and Q. V. Le. 2014. Sequence to sequence learning with neural networks. in Proc. 27th NIPS, pp.3104–3112.
  • J. Suzuki and M. Nagata. 2016. RNN-based Encoder-decoder Approach with Word Frequency Estimation.
  • A. Tombros and M. Sanderson. 1998. Advantages of Query Biased Summaries in Information Retrieval. in SIGIR '02: Proc. 21st annual international ACM SIGIR conference on Research and development in information retrieval, pp.2-10
  • A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. Gomez, L. Kaiser, and I. Polosukhin. 2017. Attention is all you need. in Proc. 30th NIPS, pp.5998–6008.
  • A. Wang, K. Cho, M. Lewis. 2020. Asking and Answering Questions to Evaluate the Factual Consistency of Summaries. in Proc. 58th ACL, pp.5008–5020.
  • D. Xiao, H. Zhang, Y. Li, Y. Sun, H. Tian, H. Wu, H. Wang. 2020. ERNIE-GEN: An Enhanced Multi-Flow Pre-training and Fine-tuning Framework for Natural Language Generation. arXiv preprint arXiv:2001.11314.
  • J. Xu, Z. Gan, Y. Cheng, J. Liu. 2020. Discourse-Aware Neural Extractive Text Summarization. in Proc. 58th ACL, pp.5021–5031.
  • J. Zhang, Y. Zhao, M. Saleh, P. J. Liu. 2020. PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarization. in ICML.
  • T. Zhang, V. Kishore, F. Wu, K. Q. Weinberger, Y. Artzi. 2020. BERTScore: Evaluating Text Generation with BERT. in ICLR. arXiv preprint arXiv:1904.09675.
  • X. Zhang, F. Wei, M. Zhou. 2019. HIBERT: Document Level Pre-training of Hierarchical Bidirectional Transformers for Document Summarization. in Proc. 57th ACL, pp.5059–5069.
  • H. Zhu, L. Dong, F. Wei, B. Qin, T. Liu. 2019. Transforming Wikipedia into Augmented Data for Query-Focused Summarization. arXiv preprint arXiv:1911.03324.
219
171
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
219
171

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?