PyTorch深層学習⑳LSTM:実践編
前回は、LSTM(長・短期記憶)の理論的な側面を解説しました。今回は、LSTMを使って画像分類を行ってみます。
そう、自然言語処理ではなく、LSTMで画像分類を行います。そうすることで特徴量抽出器としてのLSTMの処理の流れが理解しやすくなります。
本シリーズの記事リスト
⑳LSTM:実践編
LSTMも再帰型ニューラルネットワークなので、通常は時系列データや言語モデルなどを思い浮かべる人が多いでしょう。しかし、言語モデルなどではトークンの処理など再帰型を理解する目的以外の余計なことが多くなります。よって、今回もRNNを使った実装を行った時のようにMNISTの画像分類を行います。
RNNと同様に、LSTMもシーケンス(順番に並んだ)のデータから特徴量を抽出する手段となります。よって、画像データを順番に並んだピクセルデータと考えれば、そこから特徴量を取り出すことが可能です。
では、さっそく始めましょう。
この記事が気に入ったらサポートをしてみませんか?