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:
1. Tokenização
tip
O objetivo desta fase inicial é muito simples: Dividir a entrada em tokens (ids) de uma maneira que faça sentido.
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.
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.
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.
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.
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.