Modelado de lenguaje
El modelado de lenguaje predice palabras en un enunciado. Hay dos formas de modelado de lenguaje.
El modelado de lenguaje causal predice el siguiente token en una secuencia de tokens, y el modelo solo puede considerar los tokens a la izquierda.
El modelado de lenguaje por enmascaramiento predice un token enmascarado en una secuencia, y el modelo puede considerar los tokens bidireccionalmente.
Esta guía te mostrará cómo realizar fine-tuning DistilGPT2 para modelos de lenguaje causales y DistilRoBERTa para modelos de lenguaje por enmascaramiento en el r/askscience subdataset ELI5.
Mira la página de tarea para generación de texto y la página de tarea para modelos de lenguajes por enmascaramiento para obtener más información sobre los modelos, datasets, y métricas asociadas.
Carga el dataset ELI5
Carga solo los primeros 5000 registros desde la biblioteca 🤗 Datasets, dado que es bastante grande:
>>> from datasets import load_dataset
>>> eli5 = load_dataset("eli5", split="train_asks[:5000]")
Divide este dataset en subdatasets para el entrenamiento y el test:
eli5 = eli5.train_test_split(test_size=0.2)
Luego observa un ejemplo:
>>> eli5["train"][0]
{'answers': {'a_id': ['c3d1aib', 'c3d4lya'],
'score': [6, 3],
'text': ["The velocity needed to remain in orbit is equal to the square root of Newton's constant times the mass of earth divided by the distance from the center of the earth. I don't know the altitude of that specific mission, but they're usually around 300 km. That means he's going 7-8 km/s.\n\nIn space there are no other forces acting on either the shuttle or the guy, so they stay in the same position relative to each other. If he were to become unable to return to the ship, he would presumably run out of oxygen, or slowly fall into the atmosphere and burn up.",
"Hope you don't mind me asking another question, but why aren't there any stars visible in this photo?"]},
'answers_urls': {'url': []},
'document': '',
'q_id': 'nyxfp',
'selftext': '_URL_0_\n\nThis was on the front page earlier and I have a few questions about it. Is it possible to calculate how fast the astronaut would be orbiting the earth? Also how does he stay close to the shuttle so that he can return safely, i.e is he orbiting at the same speed and can therefore stay next to it? And finally if his propulsion system failed, would he eventually re-enter the atmosphere and presumably die?',
'selftext_urls': {'url': ['http://apod.nasa.gov/apod/image/1201/freeflyer_nasa_3000.jpg']},
'subreddit': 'askscience',
'title': 'Few questions about this space walk photograph.',
'title_urls': {'url': []}}
Observa que text
es un subcampo anidado dentro del diccionario answers
. Cuando preproceses el dataset, deberás extraer el subcampo text
en una columna aparte.
Preprocesamiento
Para modelados de lenguaje causales carga el tokenizador DistilGPT2 para procesar el subcampo text
:
>>> from transformers import AutoTokenizer
>>> tokenizer = AutoTokenizer.from_pretrained("distilbert/distilgpt2")
Para modelados de lenguaje por enmascaramiento carga el tokenizador DistilRoBERTa, en lugar de DistilGPT2:
>>> from transformers import AutoTokenizer
>>> tokenizer = AutoTokenizer.from_pretrained("distilbert/distilroberta-base")
Extrae el subcampo text
desde su estructura anidado con el método flatten
:
>>> eli5 = eli5.flatten()
>>> eli5["train"][0]
{'answers.a_id': ['c3d1aib', 'c3d4lya'],
'answers.score': [6, 3],
'answers.text': ["The velocity needed to remain in orbit is equal to the square root of Newton's constant times the mass of earth divided by the distance from the center of the earth. I don't know the altitude of that specific mission, but they're usually around 300 km. That means he's going 7-8 km/s.\n\nIn space there are no other forces acting on either the shuttle or the guy, so they stay in the same position relative to each other. If he were to become unable to return to the ship, he would presumably run out of oxygen, or slowly fall into the atmosphere and burn up.",
"Hope you don't mind me asking another question, but why aren't there any stars visible in this photo?"],
'answers_urls.url': [],
'document': '',
'q_id': 'nyxfp',
'selftext': '_URL_0_\n\nThis was on the front page earlier and I have a few questions about it. Is it possible to calculate how fast the astronaut would be orbiting the earth? Also how does he stay close to the shuttle so that he can return safely, i.e is he orbiting at the same speed and can therefore stay next to it? And finally if his propulsion system failed, would he eventually re-enter the atmosphere and presumably die?',
'selftext_urls.url': ['http://apod.nasa.gov/apod/image/1201/freeflyer_nasa_3000.jpg'],
'subreddit': 'askscience',
'title': 'Few questions about this space walk photograph.',
'title_urls.url': []}
Cada subcampo es ahora una columna separada, como lo indica el prefijo answers
. Observa que answers.text
es una lista. En lugar de tokenizar cada enunciado por separado, convierte la lista en un string para tokenizarlos conjuntamente.
Así es como puedes crear una función de preprocesamiento para convertir la lista en una cadena y truncar las secuencias para que no superen la longitud máxima de input de DistilGPT2:
>>> def preprocess_function(examples):
... return tokenizer([" ".join(x) for x in examples["answers.text"]], truncation=True)
Usa de 🤗 Datasets la función map
para aplicar la función de preprocesamiento sobre el dataset en su totalidad. Puedes acelerar la función map
configurando el argumento batched=True
para procesar múltiples elementos del dataset a la vez y aumentar la cantidad de procesos con num_proc
. Elimina las columnas que no necesitas:
>>> tokenized_eli5 = eli5.map(
... preprocess_function,
... batched=True,
... num_proc=4,
... remove_columns=eli5["train"].column_names,
... )
Ahora necesitas una segunda función de preprocesamiento para capturar el texto truncado de cualquier ejemplo demasiado largo para evitar cualquier pérdida de información. Esta función de preprocesamiento debería:
- Concatenar todo el texto.
- Dividir el texto concatenado en trozos más pequeños definidos por un
block_size
.
>>> block_size = 128
>>> def group_texts(examples):
... concatenated_examples = {k: sum(examples[k], []) for k in examples.keys()}
... total_length = len(concatenated_examples[list(examples.keys())[0]])
... total_length = (total_length // block_size) * block_size
... result = {
... k: [t[i : i + block_size] for i in range(0, total_length, block_size)]
... for k, t in concatenated_examples.items()
... }
... result["labels"] = result["input_ids"].copy()
... return result
Aplica la función group_texts
sobre todo el dataset:
>>> lm_dataset = tokenized_eli5.map(group_texts, batched=True, num_proc=4)
Para modelados de lenguaje causales, usa DataCollatorForLanguageModeling
para crear un lote de ejemplos. Esto también rellenará dinámicamente tu texto a la dimensión del elemento más largo del lote para que de esta manera tengan largo uniforme. Si bien es posible rellenar tu texto en la función tokenizer
mediante el argumento padding=True
, el rellenado dinámico es más eficiente.
Puedes usar el token de final de secuencia como el token de relleno y asignar mlm=False
. Esto usará los inputs como etiquetas movidas un elemento hacia la derecha:
>>> from transformers import DataCollatorForLanguageModeling
>>> tokenizer.pad_token = tokenizer.eos_token
>>> data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False)
Para modelados de lenguaje por enmascaramiento usa el mismo DataCollatorForLanguageModeling
excepto que deberás especificar mlm_probability
para enmascarar tokens aleatoriamente cada vez que iteras sobre los datos.
>>> from transformers import DataCollatorForLanguageModeling
>>> tokenizer.pad_token = tokenizer.eos_token
>>> data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm_probability=0.15)
Puedes usar el token de final de secuencia como el token de relleno y asignar mlm=False
. Esto usará los inputs como etiquetas movidas un elemento hacia la derecha:
>>> from transformers import DataCollatorForLanguageModeling
>>> data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False, return_tensors="tf")
Para modelados de lenguajes por enmascaramiento usa el mismo DataCollatorForLanguageModeling
excepto que deberás especificar mlm_probability
para enmascarar tokens aleatoriamente cada vez que iteras sobre los datos.
>>> from transformers import DataCollatorForLanguageModeling
>>> data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False, return_tensors="tf")
Modelado de lenguaje causal
El modelado de lenguaje causal es frecuentemente utilizado para generación de texto. Esta sección te muestra cómo realizar fine-tuning a DistilGPT2 para generar nuevo texto.
Entrenamiento
Carga DistilGPT2 con AutoModelForCausalLM
:
>>> from transformers import AutoModelForCausalLM, TrainingArguments, Trainer
>>> model = AutoModelForCausalLM.from_pretrained("distilbert/distilgpt2")
Si no estás familiarizado con el proceso de realizar fine-tuning sobre un modelo con Trainer
, considera el tutorial básico aquí!
A este punto, solo faltan tres pasos:
- Definir tus hiperparámetros de entrenamiento en
TrainingArguments
. - Pasarle los argumentos de entrenamiento a
Trainer
junto con el modelo, dataset, y el data collator. - Realiza la llamada
train()
para realizar el fine-tuning sobre tu modelo.
>>> training_args = TrainingArguments(
... output_dir="./results",
... eval_strategy="epoch",
... learning_rate=2e-5,
... weight_decay=0.01,
... )
>>> trainer = Trainer(
... model=model,
... args=training_args,
... train_dataset=lm_dataset["train"],
... eval_dataset=lm_dataset["test"],
... data_collator=data_collator,
... )
>>> trainer.train()
Para realizar el fine-tuning de un modelo en TensorFlow, comienza por convertir tus datasets al formato tf.data.Dataset
con to_tf_dataset
. Especifica los inputs y etiquetas en columns
, ya sea para mezclar el dataset, tamaño de lote, y el data collator:
>>> tf_train_set = lm_dataset["train"].to_tf_dataset(
... columns=["attention_mask", "input_ids", "labels"],
... dummy_labels=True,
... shuffle=True,
... batch_size=16,
... collate_fn=data_collator,
... )
>>> tf_test_set = lm_dataset["test"].to_tf_dataset(
... columns=["attention_mask", "input_ids", "labels"],
... dummy_labels=True,
... shuffle=False,
... batch_size=16,
... collate_fn=data_collator,
... )
Si no estás familiarizado con realizar fine-tuning de tus modelos con Keras, considera el tutorial básico aquí!
Crea la función optimizadora, la tasa de aprendizaje, y algunos hiperparámetros de entrenamiento:
>>> from transformers import create_optimizer, AdamWeightDecay
>>> optimizer = AdamWeightDecay(learning_rate=2e-5, weight_decay_rate=0.01)
Carga DistilGPT2 con TFAutoModelForCausalLM
:
>>> from transformers import TFAutoModelForCausalLM
>>> model = TFAutoModelForCausalLM.from_pretrained("distilbert/distilgpt2")
Configura el modelo para entrenamiento con compile
:
>>> import tensorflow as tf
>>> model.compile(optimizer=optimizer)
Llama a fit
para realizar el fine-tuning del modelo:
>>> model.fit(x=tf_train_set, validation_data=tf_test_set, epochs=3)
Modelado de lenguaje por enmascaramiento
El modelado de lenguaje por enmascaramiento es también conocido como una tarea de rellenar la máscara, pues predice un token enmascarado dada una secuencia. Los modelos de lenguaje por enmascaramiento requieren una buena comprensión del contexto de una secuencia entera, en lugar de solo el contexto a la izquierda. Esta sección te enseña como realizar el fine-tuning de DistilRoBERTa para predecir una palabra enmascarada.
Entrenamiento
Carga DistilRoBERTa con AutoModelForMaskedlM
:
>>> from transformers import AutoModelForMaskedLM
>>> model = AutoModelForMaskedLM.from_pretrained("distilbert/distilroberta-base")
Si no estás familiarizado con el proceso de realizar fine-tuning sobre un modelo con Trainer
, considera el tutorial básico aquí!
A este punto, solo faltan tres pasos:
- Definir tus hiperparámetros de entrenamiento en
TrainingArguments
. - Pasarle los argumentos de entrenamiento a
Trainer
junto con el modelo, dataset, y el data collator. - Realiza la llamada
train()
para realizar el fine-tuning de tu modelo.
>>> training_args = TrainingArguments(
... output_dir="./results",
... eval_strategy="epoch",
... learning_rate=2e-5,
... num_train_epochs=3,
... weight_decay=0.01,
... )
>>> trainer = Trainer(
... model=model,
... args=training_args,
... train_dataset=lm_dataset["train"],
... eval_dataset=lm_dataset["test"],
... data_collator=data_collator,
... )
>>> trainer.train()
Para realizar el fine-tuning de un modelo en TensorFlow, comienza por convertir tus datasets al formato tf.data.Dataset
con to_tf_dataset
. Especifica los inputs y etiquetas en columns
, ya sea para mezclar el dataset, tamaño de lote, y el data collator:
>>> tf_train_set = lm_dataset["train"].to_tf_dataset(
... columns=["attention_mask", "input_ids", "labels"],
... dummy_labels=True,
... shuffle=True,
... batch_size=16,
... collate_fn=data_collator,
... )
>>> tf_test_set = lm_dataset["test"].to_tf_dataset(
... columns=["attention_mask", "input_ids", "labels"],
... dummy_labels=True,
... shuffle=False,
... batch_size=16,
... collate_fn=data_collator,
... )
Si no estás familiarizado con realizar fine-tuning de tus modelos con Keras, considera el tutorial básico aquí!
Crea la función optimizadora, la tasa de aprendizaje, y algunos hiperparámetros de entrenamiento:
>>> from transformers import create_optimizer, AdamWeightDecay
>>> optimizer = AdamWeightDecay(learning_rate=2e-5, weight_decay_rate=0.01)
Carga DistilRoBERTa con TFAutoModelForMaskedLM
:
>>> from transformers import TFAutoModelForMaskedLM
>>> model = TFAutoModelForCausalLM.from_pretrained("distilbert/distilroberta-base")
Configura el modelo para entrenamiento con compile
:
>>> import tensorflow as tf
>>> model.compile(optimizer=optimizer)
Llama a fit
para realizar el fine-tuning del modelo:
>>> model.fit(x=tf_train_set, validation_data=tf_test_set, epochs=3)
Para un ejemplo más profundo sobre cómo realizar el fine-tuning sobre un modelo de lenguaje causal, considera PyTorch notebook o TensorFlow notebook.