English BERT encoder
The English BERT encoder block implements the BERT—Bidirectional Encoder Representations from Transformers—network in its base size, as published in BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding.
BERT pushes the state of the art in Natural Language Processing by combining two powerful technologies:
What’s more, the original authors have released pre-trained weights, so that you can use it with minimal training work.
Using the English BERT encoder
Use the English BERT snippet to directly get a complete model for text classification or text regression that uses the English BERT encoder.
The input of the English BERT encoder must come from a Tokenizer block.
The tokenizer must use English uncased as Vocabulary, so that the tokenized numerical values are compatible with the English BERT encoder block.
The English BERT encoder returns the so-called CLS output. This output is a vector that can be passed to other blocks to perform regression or classification.
The English BERT encoder block implements the base version of the BERT network.
It is composed of 12 successive transformer layers, each having 12 attention heads.
The total number of parameters is 110 million.
Every token in the input of the block is first embedded into a learned 768-long embedding vector.
Each embedding vector is then transformed progressively every time it traverses one of the BERT Encoder layers:
Through linear projections, every embedding vector creates a triplet of 64-long vectors, called the key, query, and value vectors
The key, query, and value vectors from all the embeddings pass through a self-attention head, which outputs one 64-long vector for each input triplet.
Every output vector from the self-attention head is a function of the whole input sequence, which is what makes BERT context-aware.
A single embedding vector uses different linear projections to create 12 unique triplets of key, query, and value vectors, which all go through their own self-attention head.
This allows each self-attention head to focus on different aspects of how the tokens interact with each other.
The output from all the self-attention heads are first concatenated together, then they go through another linear projection and a feed-forward layer, which helps to utilize deep non-linearity. Residual connections from previous states are also used to increase robustness.
The result is a sequence of transformed embedding vectors, which are sent through the same layer structure 11 more times.
After the 12th encoding layer, the embedding vectors have been transformed to contain more accurate information about each token.
This block returns all of them or only the first one (corresponding to the
[CLS] token), which is often sufficient for classification tasks.
Jacob Devlin et al.: BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding, 2019.