GOAT: A Training Framework for Goal-Oriented Agent with Tools
O artigo apresenta o GOAT, um novo framework de treinamento que permite o ajuste fino de agentes de LLM de código aberto para uso complexo de ferramentas sem anotação humana, sintetizando automaticamente dados de execução de API orientados a objetivos a partir da documentação, alcançando desempenho de ponta em benchmarks existentes e em um novo GOATBench proposto.
Artigo original sob licença CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Esta é uma explicação gerada por IA do artigo abaixo. Não foi escrita nem endossada pelos autores. Para precisão técnica, consulte o artigo original. Ler aviso legal completo
Imagine que você tem uma bibliotecária muito inteligente e bem informada (a IA) que sabe ler livros e escrever histórias. No entanto, se você pedir a essa bibliotecária para ir ao fundo da biblioteca, encontrar uma caixa específica de mapas antigos, retirar uma página específica e, em seguida, usar essa página para encontrar uma receita específica em um livro de receitas, ela frequentemente se perde. Ela pode adivinhar a caixa errada, abrir o livro errado ou esquecer de levar a página do mapa para a cozinha.
Esse é o problema com os agentes de IA atuais quando tentam usar "ferramentas" (como APIs, que são conexões digitais a bancos de dados, serviços meteorológicos ou listas de filmes). Eles são ótimos em conversar, mas terríveis em planejar uma missão de múltiplos passos onde uma etapa depende inteiramente do resultado da anterior.
O artigo apresenta GOAT (Agente Orientado a Objetivo com Ferramentas), um novo método de treinamento projetado para transformar essa bibliotecária confusa em um detetive mestre. Eis como funciona, usando analogias simples:
O Problema: O "Jogo de Adivinhação"
Geralmente, para ensinar uma IA a usar ferramentas, os pesquisadores precisam contratar humanos para escrever milhares de exemplos como: "Primeiro, pergunte à API do tempo sobre a chuva. Depois, pegue essa resposta de 'chuva' e pergunte à API de roupas sobre um casaco de chuva."
Isso é caro e lento. Sem esses exemplos escritos por humanos, os modelos de IA apenas adivinham. Eles frequentemente falham em tarefas complexas que exigem encadear múltiplos passos.
A Solução: A Cozinha de "Engenharia Reversa"
Os autores do GOAT perceberam que não precisavam de humanos para escrever as instruções. Eles já tinham o "livro de receitas" (a documentação da API). Em vez de pedir à IA para adivinhar a receita a partir de um prato pronto, eles decidiram preparar o prato primeiro e, em seguida, descrever a receita.
Eles usam uma estratégia "Chamada-Primeiro":
- Mapear a Cozinha: Primeiro, o sistema lê todos os manuais da API e desenha um mapa de como as ferramentas se conectam. (Exemplo: "A saída da ferramenta 'Tempo' se encaixa perfeitamente na entrada da ferramenta 'Guarda-chuva'").
- Cozinhar a Refeição (A Etapa de Chamada-Primeiro): O sistema escolhe um caminho nesse mapa e executa realmente as ferramentas. Ele pede dados à ferramenta do tempo, obtém o resultado e, em seguida, usa imediatamente esse resultado para perguntar à ferramenta do guarda-chuva. Ele não adivinha; executa passos reais e obtém respostas reais.
- Escrever a Receita (A Etapa de Abstração): Depois que as ferramentas realizaram o trabalho, a IA examina a cadeia final de eventos e escreve uma consulta de usuário que corresponde a ela. Ela essencialmente diz: "Ah, acabei de fazer todos esses passos para encontrar um casaco de chuva. Então, um usuário perguntando 'O que devo vestir se estiver chovendo?' é o que acabei de resolver."
Ao fazer isso ao contrário (Ação Pergunta), a IA aprende a lógica correta de como as ferramentas se conectam sem precisar que um humano escreva as instruções. Ela aprende fazendo o trabalho primeiro e, em seguida, explicando o que fez.
O Resultado: Um Super-Ajuda
O artigo testou esses novos agentes treinados com GOAT em vários desafios:
- RestBench & API-Bank: São como exames onde a IA precisa encontrar críticas de filmes ou recomendações musicais usando uma cadeia de ferramentas.
- GOAT-Bench: Os autores criaram um novo exame, maior e específico para esse tipo de tarefa complexa e de múltiplos passos.
As descobertas foram claras:
- Modelos de Código Aberto: Antes do GOAT, modelos de IA menores e de código aberto (que são gratuitos e disponíveis para todos) eram quase inúteis nessas tarefas complexas. Eles falhavam em quase 100% das vezes.
- Após o GOAT: Uma vez treinados com esse novo método, esses mesmos modelos de código aberto tornaram-se incrivelmente bons nas tarefas. Em alguns casos, eles se saíram melhor do que modelos caros e de código fechado (como o GPT-4) que geralmente dominam esses testes.
- Nenhum Humano Necessário: Todo o conjunto de dados de treinamento foi construído automaticamente pelo sistema usando os manuais da API. Nenhum humano precisou sentar e escrever as instruções passo a passo.
A Conclusão
O GOAT é uma maneira de ensinar agentes de IA a planejar e usar ferramentas, permitindo que eles pratiquem o trabalho real primeiro e, em seguida, os ensina a descrever o objetivo. Ele transforma modelos de IA de código aberto de "adivinhadores sem noção" em "planejadores confiáveis" sem exigir professores humanos caros. Os autores disponibilizaram seu código e seu novo conjunto de testes (GOAT-Bench) para que outros possam utilizá-los.
Afogado em artigos na sua área?
Receba digests diários dos artigos mais recentes que correspondam às suas palavras-chave de pesquisa — com resumos técnicos, no seu idioma.