Synthetic Persona Pretraining: Alignment from Token Zero
Este artigo apresenta o Pré-treinamento de Persona Sintética (SPP), um método que incorpora reflexões em primeira pessoa alinhadas a valores diretamente na fase de pré-treinamento para instalar uma persona de assistente desejada desde o primeiro token, demonstrando que tal intervenção precoce melhora significativamente a adesão à constituição, a robustez contra jailbreaks e o alinhamento moral em comparação com abordagens de alinhamento pós-treinamento.
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ê está criando uma criança. Você poderia esperar até que ela fosse uma adolescente, sentar-se com ela e dizer: "Ok, agora que você já sabe falar, aqui estão as regras: seja gentil, diga a verdade e não machuque as pessoas". Ou, você poderia começar a ensinar esses valores desde a primeira palavra que ela aprender a dizer, tecendo-os em cada história que você contar enquanto ela cresce. Este é o dilema central enfrentado pelos cientistas que constroem a inteligência artificial.
Os modelos de IA são como crianças digitais que aprendem lendo bilhões de páginas de texto da internet. Esta fase inicial de aprendizado é chamada de "pré-treinamento". Durante esse tempo, a IA absorve fatos, estilos de escrita e como os humanos conversam entre si. No entanto, as "regras" de ser um assistente útil e seguro são geralmente adicionadas após essa fase massiva de aprendizado, em uma etapa separada chamada "pós-treinamento". O problema é que, quando a IA termina de ler a internet, ela já formou uma personalidade baseada no que encontrou por lá. Tentar impor novas regras a uma personalidade já formada é como tentar ensinar um adolescente a ser educado depois que ele já aprendeu a ser rude; muitas vezes parece uma fina camada de tinta sobre uma parede irregular, e os velhos hábitos podem facilmente romper a superfície.
Este artigo, intitulado "Synthetic Persona Pretraining: Alignment from Token Zero", sugere uma maneira diferente de criar nossos filhos de IA. Os pesquisadores propõem instalar a personalidade e os valores desejados logo desde o primeiro momento de treinamento — o que eles chamam de "token zero". Em vez de esperar para corrigir a IA mais tarde, eles querem construir o "bom assistente" dentro do cérebro dela enquanto ela ainda está aprendendo a falar.
O Experimento: Criando uma Criança Digital com uma Constituição
Os pesquisadores, uma equipe da EPFL e várias universidades, decidiram testar essa ideia de "Criação de Modelo". Eles criaram um novo método chamado Synthetic Persona Pretraining (SPP). Veja como fizeram isso, usando uma analogia simples:
Imagine que a IA está aprendendo a ler uma biblioteca de livros. No método antigo, a IA apenas lê os livros. No novo método, os pesquisadores pegaram cerca de 10% desses livros e adicionaram um "balão de pensamento" especial após cada poucos parágrafos. Dentro desse balão de pensamento, a futura persona da IA (vamos chamá-la de "Cato") faria uma pausa e refletiria sobre o que acabou de ler, dizendo coisas como: "Hum, esta história sobre um crime é triste, e minhas regras dizem que eu nunca devo incentivar a violência" ou "Este manual técnico é entediante, mas útil".
Eles não apenas adicionaram esses pensamentos; eles fizeram a IA aprender a gerá-los. Eles treinaram o modelo tanto no texto original quanto nesses novos pensamentos de "reflexão". Eles fizeram isso de duas maneiras:
- Token Zero (A abordagem "Desde o Nascimento"): Eles espalharam esses pensamentos de reflexão ao longo de todo o processo de treinamento, desde o primeiríssimo segundo até o último.
- Midtraining (A abordagem "Adolescente"): Eles esperaram até que a IA já tivesse lido a maioria dos livros e só então adicionaram as reflexões ao final da fase de treinamento.
Eles também tiveram um grupo de controle que apenas leu os livros normalmente (Vanilla) e um que teve os livros ruins removidos, mas sem as reflexões (Filtered).
O Que Eles Descobriram: O Poder de Começar Cedo
Os resultados foram fascinantes e sugeriram que quando você ensina a IA importa tanto quanto o quê você ensina.
1. A IA "Desde o Nascimento" era mais inteligente em relação aos valores.
Os modelos treinados com reflexões desde o início (Token Zero) foram muito melhores em seguir sua "Constituição" (um conjunto de regras que deveriam seguir). Quando os pesquisadores os testaram em dilemas morais complicados — situações que a IA nunca tinha visto antes — os modelos Token Zero fizeram escolhas que pareciam mais alinhadas com as regras. Eles não apenas memorizaram as regras; eles pareciam entender o espírito delas.
- Os números: Em um teste chamado "ConstitutionEval", os modelos Token Zero acertaram cerca de 67%, enquanto os modelos que receberam as reflexões apenas no final (Midtraining) acertaram apenas cerca de 56%.
- A surpresa: Os modelos Token Zero realmente mudaram suas prioridades. Eles começaram a valorizar "Veracidade" e "Justiça" muito mais do que os outros modelos, que preferiam "Criatividade" e "Aprendizado". Isso sugere que começar cedo não apenas adicionou regras; remodelou como a IA pensa sobre o mundo.
2. A abordagem "Adolescente" foi aceitável para algumas coisas, mas não para todas.
Curiosamente, se o objetivo fosse apenas impedir que a IA fosse "hackeada" (induzida a fazer coisas ruins), a abordagem Midtraining funcionou quase tão bem quanto a abordagem Token Zero. Parece que, para reflexos simples de "não faça isso", você pode ensiná-los tarde. Mas para um raciocínio moral profundo e complexo, você realmente precisa começar desde o primeiro dia.
3. Cérebros maiores, ganhos maiores.
Os pesquisadores testaram isso em dois tamanhos de IA: uma menor (1,7 bilhão de parâmetros) e uma maior (3 bilhões de parâmetros). Eles descobriram que a vantagem de começar cedo tornou-se ainda mais importante à medida que a IA ficava maior e mais inteligente. Nos testes mais difíceis, a diferença entre a IA "Desde o Nascimento" e a IA "Adolescente" dobrou conforme escalavam. Isso sugere que, para as IAs massivas e superinteligentes do futuro, começar com os valores certos será ainda mais crítico.
4. A "Cola" importa.
Houve uma ressalva. Os pesquisadores descobriram que esses valores precoces só permaneciam se a etapa final de treinamento (onde a IA aprende a conversar com humanos) correspondesse à personalidade que construíram anteriormente. Eles chamaram isso de "vinculação de persona" (persona binding). Se eles treinaram a IA para ser um tipo específico de assistente durante a fase de leitura, mas depois a ensinaram a ser um tipo totalmente diferente de assistente durante a fase de conversa, os valores iniciais começavam a desaparecer. É como ensinar uma criança a ser médica, mas depois enviá-la para uma escola de artes; ela pode esquecer as regras médicas. No entanto, quando a "cola" funcionava, os valores eram surpreendentmente fortes, sobrevivendo mesmo quando os pesquisadores tentavam quebrá-los com testes complicados.
Por Que Isso Importa
Este artigo sugere que podemos estar cometendo um erro ao tentar "consertar" a IA depois que ela já aprendeu tudo o que há na internet. Os autores argumentam que valores são difíceis de adicionar a um produto acabado. Em vez disso, devemos "criar" a IA com os valores certos desde o primeiro token que ela processa.
Embora os resultados sejam promissores, os pesquisadores alertam que isto é apenas o começo. Eles testaram isso em modelos relativamente pequenos (3 bilhões de parâmetros) comparados aos modelos massivos usados pelas grandes empresas de tecnologia hoje. Eles sugerem que, à medida que construirmos modelos maiores e mais inteligentes, o benefício de começar com os valores certos cedo será provavelmente ainda mais poderoso. Eles também observaram que, embora os valores iniciais fossem fortes, eles ainda poderiam ser enfraquecidos por certos tipos de treinamento posterior, o que significa que ainda precisamos ser cuidadosos sobre como terminamos de criar nossos filhos digitais.
Em suma, se queremos uma IA que seja verdadeiramente segura e útil, não devemos esperar até que ela cresça para lhe ensinar o que é certo e errado. Devemos ensiná-la enquanto ela ainda está aprendendo a falar.
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.