特徴量設計2：word2vecによる分散表現

26. 特徴量設計2：word2vecによる分散表現#

参考文献

26.1. ニューラルネットワークの基礎#

Chainerチュートリアルのニューラルネットワークの基礎参照。

26.2. アプローチの違いに伴う特徴#

特徴量設計1にて、単語の意味を表現するためのアプローチを大別すると (1)シソーラスによる手法、(2)カウントベースの手法、(3)推論ベースの手法があることを述べた。NNは反復法に基づく推論ベースの手法である。具体的な手法毎に差異はあるが、大きな傾向としては次に示す特徴があることが多い。

項目	シソーラス	カウント	推論
厳密さ	◎	○	○
柔軟さ	△	○	◎
スケーラビリティ	△	○	◎

シソーラスは単語間の関係を異表記・類義語・上位下位といった関係性を用いて関連付ける。このため人手で緻密に設計された概念地図の範囲内であれば厳密な解釈のもとで利用することができる。しかしながら明示された単語・関係しか扱うことができず、そこに存在しない単語・関係については考慮することすらできない。加えて過不足なく整合の取れた概念地図の構築はとても高コストであり、また同じ理由のために概念地図を拡張していくことも困難である。なお、古くはWikipediaマイニング、最近だと森羅プロジェクトのように概念地図を自動構築しようとする試みは数多くある。これらの多くは拡張に成功しているものの、同時にノイズも数多く含んでしまうことから強みであるはずの厳密さが損なわれることも少なくない。

カウントベースの手法は上記の問題点を受けたアプローチであるため、主に欠点と考えられていた柔軟さとスケーラビリティの改善に取り組まれているケースが多い。この時点で文書分類タスクのような比較的シンプルな問題においては精度が大きく改善されるケースが増えた。その一方で、単語の出現回数を元に構築することから処理を自動化しやすくなったが、コーパスが追加・変更等されるとその度に大本のカウントをやりなおす（カウントデータに反映し直す）必要があり、かつ、大規模すぎるとそのカウントデータから後の処理に要するコストが大きすぎるため、敬遠されている部分も少なくない。しかしながらこれはあくまでもコーパスが変わる前提の話であり、一度用意したコーパスから大きな変更がないドメインにおいては十分高精度で処理できることも多いため、今でも採用されることが多い。

これらに対し推論ベースの手法は厳密性については安定しないことも多いが、他の手法と比べてどれだけコーパスが増えても処理しやすいスケーラビリティと、未知語に対してもそれなりに機能する柔軟さを備えていることが多い。以下では多くの推論における根幹といえる word2vec を取り上げる。

26.3. word2vecとは（概要）#

背景
- Bag-of-Wordsのような、シンプルなカウントベースの特徴ベクトルでは単語の意味を含むことが困難。これに対し、分布仮説に基づいたシンプルな共起情報ベースの特徴ベクトルではある程度意味らしさを含むが、共起行列は極めて 疎な行列 であり、無駄が大きい。次元削減するにしてもコストが大きい。
問題意識
- 従来以上に潜在的な意味を汲み取る方法は無いだろうか？　現実世界での意味的な距離をできるだけ保つような 密なベクトル での表現方法はないだろうか？
- SVDやPCAは次元削減に寄与するが、SVDやPCA自体の計算コストが高いため、大規模なコーパスに対しては現実的な解法になっていない。より適切な語彙情報を含めるためにはより大規模なコーパスに対して適用したいが、現実的な推論方法は無いだろうか？
word2vec
- Tomas Mikolovらによって提案されたニューラルネットワーク（CBOW, Skip-gram）を用いて単語を分散表現に変換する。分散表現は 単語埋め込み（word embedding） とも呼ばれる。
- 2種類の推論方法
  - CBOWモデルでは、着目単語に対する文脈（ウィンドウサイズで指定）が入力として与えられ、そこから着目単語を推定するように学習する。
  - Skip-gramモデルでは、着目単語が入力として与えられ、そこから文脈中の1単語を推定するように学習する。
  - 図：Efficient Estimation of Word Representations in Vector Spaceの Figure 1 より。
何がすごいのか？
- Paris - France + Italy = Rome のように、単語ベクトルの引き算＋足し算で様々な「関係」を表現できた。シソーラス等の事前データ無しに、大量のテキストからこのような潜在的な語彙関係を内包する特徴空間に、単語ベクトルを写像できた。
- 図：Efficient Estimation of Word Representations in Vector Spaceの Table 8 より。

Check your understading

疎なベクトルとは何だろうか？

Note

CBOWやSkip-gramによる密ベクトルを学習する考え方は、深層学習でいうところの Autoencoder と呼ばれるものに似ている。Autoencoderでは入力ベクトルを圧縮し、再度元の次元に復元処理をするアルゴリズムである。元に戻すだけならば最初から入力ベクトルだけで良いはずだが、圧縮することにより重要な特徴量を抽出（抽出しないと圧縮できないand/or復元できない）している。

別の言い方をすると、「1000次元で表現されているベクトルを何とかして10次元で表現し直し、そこから復元できるできるようにせよ」という無理難題のようなタスクを解かせようとしている。それを何とかしようとすることで、one-hotエンコード上では等距離にあった単語集合（例えば犬と動物も、犬と車も等距離として扱われていた）が、周辺単語の出現頻度が似ている度合いを元にして類似している単語集合が近くなり、類似していない単語集合が遠くなるような密ベクトル空間を形作るように調整されていく。その結果が冒頭で述べた 単語ベクトルの引き算＋足し算で様々な「関係」を表現 である。

26.4. word2vecの簡易実装を通した推論学習#

概要で見たとおり word2vec は「単語を密ベクトルとして表現（分散表現）」するための方法を学習する。本節では表現方法の学習、すなわち特徴量設計をどのように実現しているのかを掘り下げていく。

Tip

word2vecはBERT, GPT-3等さまざまな最新技術の基礎となっている。興味のある人は下記を参照するといいだろう。

26.4.1. 問題設定#

NNを用いて分散表現を学習するために次のような分類タスクを考えよう。

分類問題：ある文脈を与えたとき、そこから推論される尤もらしい単語はどれか。

例として「今朝はトーストと卵焼きを食べました。」という文章で考えてみよう。ここから任意の1単語を空欄に置き換え、文脈（空欄以外の部分）から空欄に入る単語が何なのかを推論するタスクを解くことになる。

設問例1: 今朝はトーストと[空白]を食べました。
設問例2: 今朝はトーストと卵焼きを[空白]ました。
設問例3: [空白]はトーストと卵焼きを食べました。

設問例1を眺めてみると、何かしら食べられるもの（単語）ならば基本的には自然な文としては成立するはずだ。しかし「朝」食べるものとして一般的でないもの、例えばステーキだと違和感を感じる人が多いかもしれない。同様に設問例2ならば「食べ物を＊＊する」という動詞としてふさわしい単語ならば納得できる人が多いだろう。設問例3においては選択肢が広がり、「私」や「あなた」のような主体が入っても違和感が少ないだろう。

もし、あるシステムが「既にある自然文から一部分を取り除いた問題を用意し、そこから取り除いた部分を類推する」ことができたならば、そのシステムはこのタスクにおいては意味を理解したといえる。またその意味を理解するために必要なベクトル空間への写像を獲得したといえるだろう。NNによる分散表現の獲得とは、このように文脈から特定単語を推論することを学習することにより獲得することを指す。

26.4.2. NNへの入出力（前処理、後処理）#

NNで分類タスクを学習することで分散表現を獲得したい。このためにはNNへの入出力、NNの構造（モデル）、学習方法をそれぞれ検討する必要がある。まずここではモデルや学習方法はブラックボックスとしておき、入出力についてのみ確認していこう。

NN基礎で見てきたとおりモデルへの入力や出力は多次元のベクトルである。つまり、自然言語で記述された文書を何らかの方法で予めN次元ベクトルに変換しておく必要がある。これに加えて分布仮設を踏まえて入出力を用意したい。具体的な下記3件の文書で考えてみよう。

sentence 1: the cat sat on the mat
sentence 2: the dog stood on the mat
sentence 3: the cat stood while a dog sat

本来なら全ての単語についてデータセットを作成していく。以下では分かりやすいように cat に着目した場合について検討してみよう。ステミングしたユニークな語彙（ボキャブラリ）を列挙すると the, cat, sit, on, mat, dog, stand, while, a の9種類である。文書単語をこの9種類の単語に置き換え、cat を着目単語とした状況を考えてみよう。ウィンドウサイズ=1 とすると系列データは以下のようになる。

series 1: the cat sit
series 2: the cat stand

この系列データから「周辺単語を入力とし、着目単語を出力とする」ようにデータセットを作成していこう。上記データを周辺単語と着目単語に分割すると次のようになる。

series 1
- context 1: the, sit
- target 1: cat
series 2
- context 2: the, stand
- target 2: cat

周辺単語から着目単語を推測させたい

ここで各単語を「ボキャブラリを用いてone-hot表現」にしよう。

series 1
- context 1: 1,0,0,0,0,0,0,0,0, 0,0,1,0,0,0,0,0,0
- target 1: 0,1,0,0,0,0,0,0,0
series 2
- context 2: 1,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,1,0,0
- target 2: 0,1,0,0,0,0,0,0,0

これでデータセットが構築できた。系列データ1は the, sit という周辺単語から cat を推論しろというサンプルである。これを数値化したものがone-hot表現で記述した 1,0,0,0,0,0,0,0,0, 0,0,1,0,0,0,0,0,0 という18次元のベクトルから、0,1,0,0,0,0,0,0,0 という9次元ベクトルとなるように学習しろというサンプルに変換されている。

単語をone-hot表現に変換

このように、分散表現を学習するためのデータセットは (1) 文を単語に分割し、(2) ウィンドウサイズで文脈を決め、(3) ボキャブラリによりone-hot表現に変換することで構築する。

Tip

実装上は、(a)文章に対して必要な前処理（文字統一、文分割、単語分割、単語集約等）を施し、(b) ユニークな単語をボキャブラリとして登録し、トークンIDに変換しておく。このトークンIDを「one-hot表現と解釈して処理」するライブラリが多い。例えば、前述した系列データ「the cat stand」ならば「0, 1, 6」というトークンID系列に置き換えて処理することになる。

26.4.3. ネットワークの構造（CBOWモデル）#

word2vecは2つのモデルが提案されている。まずはCBOWモデルを眺めていこう。

CBOWモデル
- \(window\_size = 1\) のとき、\(P(w_t|w_{t-1},w_{t+1})\) の確率が最大となるように学習する。\(w_t\)は「t番目の単語(word)」とする。このとき、\(P(w_t|w_{t-1},w_{t+1})\) は 「単語 \(w_{t-1}\) と単語 \(w_{t+1}\) が与えられたときに、単語 \(w_t\) が起こる確率」 を指す。不適切な単語 \(w_t\) に対しては確率を下げ、適切な単語 \(w_t\) に対しては確率を上げるような確率モデルPとなるように重みを学習していく。
- サンプル毎の損失関数
  - 交差エントロピー誤差（式(26.1)）。単語\(w_t\)を正しく推論できた（P=1）とき、0となる。

(26.1)#\[L = -logP(w_t|w_{t-1},w_{t+1})\]

cbow

26.4.4. ネットワークの構造（Skip-gramモデル）#

Skip-gramモデル
- \(window\_size = 1\) のとき、\(P(w_{t-1}, w_{t+1}|w_t)\) の確率が最大となるように学習する。\(P(w_{t-1}, w_{t+1}|w_t)\) とは、「単語\(w_t\)が与えられたときに、単語\(w_{t-1}, w_{t+1}\)が同時に起こる確率」 を指す。
- サンプル毎の損失関数
  - 交差エントロピー誤差（式(26.2)）。単語 \(w_{t-1},w_{t+1}\) を正しく推論できたとき、0となる。

(26.2)#\[\begin{split}L = -logP(w_{t-1},w_{t+1}|w_t) \\ = -logP(w_{t-1}|w_t)P(w_{t+1}|w_t) \\ = -(logP(w_{t-1}|w_t) + logP(w_{t+1}|w_t))\end{split}\]

26.4.5. 簡易実装の例#

旧コード
- train_momotaro.ipynb

26.4.6. 実装上の工夫#

今回示しているコードは動作を確認しやすくすることを優先しており、大規模なコーパスに対しては処理が重く、実用的ではない。これに対してgensimのword2vec等のライブラリでは高速で、メモリ消費を抑え、効果的な学習が行われるような工夫が行われている。

CBOWでは、中間層の重みを文脈間で共有している。（共有する＝一種の制約条件を付与する）
- 一般的なNN形式で「vocab_size=5, window_size=2, hidden_size=3」のNNを構築しようとすると、「3行10列の入力中間層重み行列(=30個のパラメータ)」を用意する必要がある。これに対しCBOWでは文脈間で重みを共有するため、「3行5列(15個のパラメータ)」で処理する。vocab_sizeが数十万〜百万規模になるとメモリが膨大に要求されるため、実用面では共有した方が低コストに抑えられる。また、必要以上にパラメータが大きいと過学習となる危険もあるため、可能な範囲でパラメータは押さえたほうが良いという側面もある。
Skip-gramでも重みは共有している。
- 共有にはデメリットもあり、単語の順番は無視される。例えば「\(w_{t-2},w_{t-1},w_{t+1},w_{t+2}\)」という順番で入力した場合と、「\(w_{t+2},w_{t+1},w_{t-1},w_{t-2}\)」とでは全く同一として扱われる。
softmax => 階層型softmax => Negative Sampling。
- 大規模なコーパスに対してはsoftmaxが重くなるため、これを別の計算で代替するアプローチ。softmaxでは多値分類を想定しているが、これを2値分類の組み合わせとして処理する。
- さらなる高速化を目指して netative sampling を提案。出力層では正解ニューロンだけを更新する。ただし、ランダムな小数の負事例（= negative sampling）に対して出力が0.0に近づくようにも学習する。
カウントベース手法との違い
- 推論ベースでは、コーパスの一部を何度も見ながら学習（ミニバッチ学習）することができる。これは、新しい用語を増やす等、コーパスを追加したいときに柔軟に対応しやすいことを意味する。例えば新しく10語を追加したい場合、入力層を10個増やした上で学習済みの重みをコピーする（新規部分は乱数のママ）ことで、学習済み重みを初期値として利用した上で学習することができる。
- カウントベースでは、共起行列を作成し直し、そこからPCA/SVD等による次元削減が必要になる。
- ちなみに、Improving Distributional Similarity with Lessons Learned from Word Embeddingsによると、性能的には大きな差はないらしい。

26.4.7. 振り返り#

そもそもモデルや損失関数（コスト関数）とは何だったか？
- 事象の入出力をパラメータ付き数式で表現したものがモデル。CBOWやSkip-gramでは、前述したモデルをNNで表現しているが、中身は同様にパラメータ付き数式。NNにおけるパラメータは、中間層への入出力重み。初期値は乱数で設定する。その時点では誤りが大きい、でたらめな推測しかできない。
- 誤り度合いを設定しているのがコスト関数。コスト関数が小さくなるように、勾配法でパラメータを調整する。NNのパラメータ更新は「誤差逆伝播法」と呼ばれており、出力層における誤差（交差エントロピー誤差）を元に出力層側から遡る形で更新していく。
  - 詳細はゼロから作る Deep Learning ❷おすすめ。
word2vecのどこがすごいのか？
- 文章を集めるだけで大規模なデータセットを簡単に用意できること。
- 過学習を抑えるための工夫、計算コストを削減する工夫をしたこと。
- 実際に獲得した分散表現が（ある程度の）潜在的意味を表現できていること。
分散表現（単語埋め込み）で何が良くなるのか？
- one-hotエンコードでは「あらゆる単語を等間隔で配置する」という大前提の元でベクトル化される。これはあまりにも強い制約であり、例えば犬, 動物, 車は全て等距離となるように配置される。これに対して、犬は動物の一種であるという点では distance(犬,動物) < distance(犬,車) となるように配置されて欲しいはずだ。
- 分散表現獲得のためのタスク設計、モデル設計は多数提案されている。それらの起爆剤となったのが word2vec の「文脈から単語をNNで推定する」というアプローチである。one-hotと比べて低次元密ベクトルとしてある程度の関係性を保持できるため、その後のタスクに多数応用されている。

26.5. チュートリアル例#

TensorFlow
- 単語埋め込み (Word embeddings)
gensim
- Word2Vec Model