Ajusta Gemini Pro en Google AI Studio o con la API de Gemini

MAR 19, 2024
Cher Hu Product Manager Gemini API
Saravanan Ganesh Software Engineer Gemini API

La siguiente entrada se publicó originalmente en octubre de 2023. Hoy, la actualizamos para compartir cómo puedes ajustar fácilmente los modelos de Gemini en Google AI Studio o con la API de Gemini.


El año pasado, lanzamos Gemini 1.0 Pro, nuestro modelo multimodal de tamaño mediano, optimizado para escalar en una amplia gama de tareas. Y con 1.5 Pro, este año, demostramos lo que los modelos de lenguaje grandes pueden hacer con una ventana de contexto experimental de 1 millón de tokens. Ahora, para personalizar rápida y fácilmente el modelo (texto) de Gemini 1.0 Pro, que tiene disponibilidad general, según tus necesidades específicas, agregamos la función de ajustes de Gemini a Google AI Studio y la API de Gemini.


¿A qué nos referimos con “ajustes”?

Por lo general, los desarrolladores requieren resultados de mayor calidad para casos de uso personalizados que los que se pueden lograr con indicaciones de pocos disparos. El ajuste mejora esta técnica, ya que permite entrenar aún más el modelo base en muchos más ejemplos específicos de tareas, tantos que no todos pueden adaptarse a la indicación.


Ajuste fino frente a ajuste eficiente de parámetros

Es posible que hayas oído hablar del “ajuste fino” clásico de los modelos. En este aspecto es donde un modelo previamente entrenado se adapta a una tarea en particular si se entrena con un conjunto más pequeño de datos etiquetados específicos de la tarea. Pero con los LLM de hoy en día y su gran cantidad de parámetros, el ajuste fino es complejo: requiere experiencia en aprendizaje automático, muchos datos y mucho procesamiento.

Para realizar el ajuste en Google AI Studio, se utiliza una técnica llamada “ajuste eficiente de parámetros” (PET) a fin de producir modelos personalizados de mayor calidad y menor latencia, en comparación con las indicaciones de pocos disparos, y sin incurrir en los costos adicionales ni en la complejidad del ajuste fino tradicional. Además, PET produce modelos de alta calidad con solo unos pocos cientos de puntos de datos, lo que implica para el desarrollador una reducción de la carga de recopilar datos.


¿Por qué realizar ajustes?

Los ajustes te permiten personalizar los modelos de Gemini con tus propios datos para que funcionen mejor en tareas de nicho y, al mismo tiempo, reducir el tamaño del contexto de las indicaciones y la latencia de la respuesta. Los desarrolladores pueden usar los ajustes para una variedad de casos de uso, entre los que se incluyen los siguientes:

  • Clasificación: ejecuta tareas de lenguaje natural, como clasificar tus datos en categorías predefinidas, sin necesidad de toneladas de trabajo manual o herramientas.
  • Extracción de información: extrae información estructurada de fuentes de datos no estructurados para respaldar las tareas downstream dentro de tu producto.
  • Generación de resultados estructurados: genera datos estructurados, como tablas, de forma rápida y sencilla.
  • Modelos de crítica: usa el ajuste para crear modelos de crítica con el objetivo de evaluar los resultados de otros modelos.


Comienza rápidamente a usar Google AI Studio

1. Crea un modelo ajustado

En Google AI Studio, es fácil ajustar los modelos, ya que se elimina cualquier necesidad de experiencia en ingeniería para crear modelos personalizados. Comienza seleccionando “Nuevo modelo ajustado” en la barra de menú de la izquierda.

moving image showing how to create a tuned model in Google AI Studio by opening 'New Tuned Model' from the menu

2. Selecciona los datos del ajuste

Puedes ajustar tu modelo desde una indicación estructurada existente o importar datos desde Hojas de cálculo de Google o un archivo CSV. Puedes comenzar con tan solo 20 ejemplos y, para obtener el mejor rendimiento, te recomendamos que proporciones un conjunto de datos de al menos 100 ejemplos.

moving image showing how to select data for tuning in Google AI Studio by importing data

3. Ve tu modelo ajustado

Ve el progreso de tu ajuste en la biblioteca. Una vez que el modelo se haya terminado de ajustar, puedes ver los detalles haciendo clic en tu modelo. Comienza a ejecutar tu modelo ajustado mediante una indicación estructurada o de formato libre.

moving image showing how to view your tuned model in Google AI Studio by importing data

4. Ejecuta tu modelo ajustado en cualquier momento

También puedes acceder a tu modelo recién ajustado creando una nueva indicación estructurada o de formato libre y seleccionando tu modelo ajustado en la lista de modelos disponibles.

moving image demonstrating what it looks like to run your tuned model in Google AI Studio after importing data

Ajustes con la API de Gemini

Google AI Studio es la forma más rápida y fácil de comenzar a ajustar los modelos de Gemini. También puedes acceder a la función a través de la API de Gemini pasando los datos de entrenamiento en la solicitud de la API cuando creas un modelo ajustado. Obtén más información sobre cómo empezar aquí.


Nos entusiasman las posibilidades que ofrece el ajuste a los desarrolladores y tenemos muchas ganas de ver qué creas con la función. Si tienes algunas ideas o casos de uso, compártelos con nosotros en X (antes conocida como Twitter) o LinkedIn.