LLM Training - Preparação de Dados

Reading time: 3 minutes

Estas são minhas anotações do livro muito recomendado https://www.manning.com/books/build-a-large-language-model-from-scratch com algumas informações extras.

Informações Básicas

Você deve começar lendo este post para alguns conceitos básicos que você deve conhecer:

0. Basic LLM Concepts

1. Tokenização

tip

O objetivo desta fase inicial é muito simples: Dividir a entrada em tokens (ids) de uma maneira que faça sentido.

1. Tokenizing

2. Amostragem de Dados

tip

O objetivo desta segunda fase é muito simples: Amostrar os dados de entrada e prepará-los para a fase de treinamento, geralmente separando o conjunto de dados em frases de um comprimento específico e gerando também a resposta esperada.

2. Data Sampling

3. Embeddings de Tokens

tip

O objetivo desta terceira fase é muito simples: Atribuir a cada um dos tokens anteriores no vocabulário um vetor das dimensões desejadas para treinar o modelo. Cada palavra no vocabulário será um ponto em um espaço de X dimensões.
Note que inicialmente a posição de cada palavra no espaço é apenas inicializada "aleatoriamente" e essas posições são parâmetros treináveis (serão melhorados durante o treinamento).

Além disso, durante o embedding de tokens outra camada de embeddings é criada que representa (neste caso) a posição absoluta da palavra na frase de treinamento. Dessa forma, uma palavra em diferentes posições na frase terá uma representação (significado) diferente.

3. Token Embeddings

4. Mecanismos de Atenção

tip

O objetivo desta quarta fase é muito simples: Aplicar alguns mecanismos de atenção. Estes serão muitas camadas repetidas que vão capturar a relação de uma palavra no vocabulário com seus vizinhos na frase atual sendo usada para treinar o LLM.
Muitas camadas são usadas para isso, então muitos parâmetros treináveis vão capturar essa informação.

4. Attention Mechanisms

5. Arquitetura do LLM

tip

O objetivo desta quinta fase é muito simples: Desenvolver a arquitetura do LLM completo. Juntar tudo, aplicar todas as camadas e criar todas as funções para gerar texto ou transformar texto em IDs e vice-versa.

Esta arquitetura será usada tanto para treinar quanto para prever texto após ter sido treinada.

5. LLM Architecture

6. Pré-treinamento e Carregamento de Modelos

tip

O objetivo desta sexta fase é muito simples: Treinar o modelo do zero. Para isso, a arquitetura LLM anterior será usada com alguns loops sobre os conjuntos de dados usando as funções de perda e otimizador definidos para treinar todos os parâmetros do modelo.

6. Pre-training & Loading models

7.0. Melhorias LoRA em Ajuste Fino

tip

O uso de LoRA reduz muito a computação necessária para ajustar finamente modelos já treinados.

7.0. LoRA Improvements in fine-tuning

7.1. Ajuste Fino para Classificação

tip

O objetivo desta seção é mostrar como ajustar finamente um modelo já pré-treinado para que, em vez de gerar novo texto, o LLM selecione e forneça as probabilidades do texto dado ser categorizado em cada uma das categorias dadas (como se um texto é spam ou não).

7.1. Fine-Tuning for Classification

7.2. Ajuste Fino para Seguir Instruções

tip

O objetivo desta seção é mostrar como ajustar finamente um modelo já pré-treinado para seguir instruções em vez de apenas gerar texto, por exemplo, respondendo a tarefas como um chatbot.

7.2. Fine-Tuning to follow instructions