O que é destilação de modelos de IA e como modelos pequenos aprendem com modelos grandes?

Por 쉬었음.com

A destilação de modelos de IA é uma técnica de treinamento em que um modelo aluno menor e mais leve aprende parte do comportamento preditivo de um modelo professor maior, mais capaz, porém mais pesado e custoso. Seu nome formal geralmente é destilação de conhecimento (KD, do inglês knowledge distillation). A ideia central não é simplesmente reduzir ou copiar os parâmetros do modelo grande. Em vez disso, o aluno é treinado novamente para se assemelhar à distribuição de respostas do professor para as mesmas entradas e, em alguns casos, também às suas representações internas.arxiv.org

O motivo para essa técnica é relativamente claro. Modelos grandes podem ter bom desempenho em problemas complexos, mas, em produção, enfrentam restrições como latência de resposta, memória, energia, custos de servidor e capacidade do dispositivo. A destilação é uma escolha de usar a capacidade computacional do professor grande durante o treinamento e implantar um aluno menor durante a operação. Portanto, a melhor forma de entender a destilação não é como uma mágica que compacta inteligência em uma forma menor, mas como uma estratégia para distribuir os custos de treinamento e os custos operacionais entre modelos diferentes.arxiv.org

Por que criar um modelo aluno em vez de usar diretamente o modelo grande?

Os requisitos do desenvolvimento e da operação de modelos são diferentes. Durante o treinamento, pode ser possível investir longos períodos e recursos computacionais substanciais, enquanto serviços em produção frequentemente exigem tempos de resposta curtos e custos previsíveis. Em ambientes como dispositivos móveis, equipamentos embarcados, APIs que lidam com grandes volumes de solicitações ou implantações que precisam executar inferência no dispositivo por motivos de privacidade, o tamanho e a latência do modelo tornam-se requisitos do produto.arxiv.orgarxiv.org

Considere um serviço de classificação de documentos. Para uma tarefa de revisão interna em que é aceitável analisar um documento por vários segundos, pode haver espaço para usar um modelo grande. Mas, se os usuários precisarem ver imediatamente um resultado de classificação sempre que pressionarem um botão, ou se milhões de documentos precisarem ser processados continuamente, a memória e o tempo necessários para cada inferência se acumulam. Se um modelo aluno mantiver um desempenho na tarefa próximo ao do professor, ele poderá ser uma opção que atende aos requisitos operacionais, mesmo com alguma perda de qualidade.

Ainda assim, não é seguro concluir que um modelo menor é sempre mais barato. A quantidade de parâmetros é um indicador importante do uso de armazenamento e memória, mas a velocidade real também depende do comprimento da entrada, do tamanho do lote, das bibliotecas de computação, do hardware e dos custos de transferência de memória. Por isso, números de aceleração relatados em um artigo não devem ser aplicados sem alterações a outros hardwares e cargas de trabalho.arxiv.orgarxiv.org

O que modelos professor e aluno trocam na destilação de conhecimento?

No aprendizado supervisionado comum, a resposta correta é fornecida como um único rótulo, como “esta imagem é um gato” ou “esta frase é positiva”. Isso pode ser chamado de alvo rígido. Um modelo professor, porém, não produz apenas uma resposta; ele calcula diferentes pontuações ou probabilidades para cada resposta possível. Para uma imagem, por exemplo, ele pode prever gato 0,70, raposa 0,20 e cachorro 0,08.

A destilação usa essa distribuição como material adicional de treinamento para o aluno. Além do fato de que gato é a resposta correta, ela transmite o sinal de que o professor considerou a entrada um tanto semelhante a uma raposa e menos semelhante a um cachorro. Hinton e coautores explicaram que esses alvos suaves podem transmitir informações úteis que não cabem em um único rótulo rígido.arxiv.org

Aqui, “conhecimento” não significa uma lista de regras escritas em linguagem humana. Mais precisamente, significa a função aprendida e os padrões comportamentais do professor que mapeiam determinadas entradas para saídas. O aluno pode não reter diretamente os parâmetros do professor, e pode ter um número diferente de camadas ou outra arquitetura. O importante é que, para uma determinada entrada, a saída ou a representação do aluno se aproxime da do professor conforme o objetivo escolhido.arxiv.orgarxiv.org

Como funcionam os alvos suaves e a temperatura?

A camada final de um modelo de classificação normalmente produz pontuações chamadas logits e, em seguida, as converte em uma distribuição de probabilidades por meio da função softmax. Na destilação, pode-se usar uma softmax que divide os logits do professor e do aluno por uma temperatura (T) para tornar a distribuição mais suave.

[ p_i^{(T)}=\frac{\exp(z_i/T)}{\sum_j\exp(z_j/T)} ]

Aqui, (z_i) é o logit da classe (i). Quando (T=1), esta é a softmax comum. Aumentar a temperatura achata uma distribuição concentrada apenas na classe de maior probabilidade. Isso pode disponibilizar diferenças relativas entre classes de classificação inferior como sinais de aprendizado. Por isso, é incompleto descrever a temperatura apenas como “desfocar a resposta do professor”. A temperatura não é um interruptor que aumenta automaticamente a quantidade de informação; é um controle que altera a forma da distribuição da qual o aluno aprende.arxiv.orgproceedings.mlr.press

Na prática, o treinamento geralmente combina dois objetivos. Uma perda incentiva o aluno a prever os rótulos verdadeiros, e a outra, a perda de destilação, incentiva-o a se aproximar das saídas suaves do professor. Isso pode ser escrito de forma simples da seguinte maneira.

[ L=\alpha L_{\text{ground truth}}+(1-\alpha)L_{\text{distillation}} ]

(\alpha), a temperatura (T) e o tipo de perda de destilação são valores que precisam ser ajustados. Uma temperatura mais alta ou um peso maior para o sinal do professor nem sempre é melhor. Pesquisas também sugerem que o efeito da seleção de temperatura pode diferir conforme as características da distribuição do professor, como quando o professor foi treinado com suavização de rótulos.proceedings.mlr.pressproceedings.mlr.press

Basta corresponder às saídas?

A forma mais básica de destilação corresponde apenas às saídas finais. Ela pode ser chamada de destilação baseada em respostas. Em modelos de classificação, probabilidades ou logits são os principais alvos; em modelos de linguagem, a distribuição sobre o próximo token é o principal alvo. É relativamente simples de implementar e pode ser considerada quando a arquitetura interna do professor é desconhecida, desde que suas saídas sejam acessíveis.arxiv.orgproceedings.mlr.press

No entanto, algumas abordagens defendem que as respostas finais, sozinhas, não conseguem transmitir adequadamente o processo computacional do professor. Em arquiteturas nas quais representações multicamadas e atenção são importantes, como Transformers, o aluno pode ser treinado para imitar sinais intermediários, incluindo embeddings, estados ocultos e matrizes de atenção do professor. Isso é chamado de destilação baseada em recursos ou destilação de camadas intermediárias.arxiv.org

O TinyBERT propôs usar sinais da camada de embedding, das camadas Transformer e da camada de predição, realizando a destilação tanto durante o pré-treinamento geral quanto no ajuste fino específico da tarefa. Em experimentos de ablação desse artigo, remover a destilação das camadas Transformer levou a uma queda substancial de desempenho. Esta não é uma regra geral de que a destilação de camadas intermediárias seja sempre essencial; é, antes, um exemplo que mostra que sinais além das saídas finais podem ser úteis dependendo da arquitetura e da tarefa.arxiv.org

Outra distinção é entre destilação offline e destilação online. A destilação offline fixa um professor já treinado e treina um aluno. A destilação online usa múltiplos modelos que ensinam uns aos outros durante o treinamento, ou sinais de professor que surgem simultaneamente. A possibilidade de manter um professor separado, o custo de treinamento que se pode assumir e a necessidade de treinar modelos juntos são critérios para escolher entre elas.arxiv.org

Como a destilação difere da poda e da quantização?

Embora todas sejam discutidas sob o objetivo comum de compressão de modelos, elas reduzem coisas diferentes. A destilação trata dos sinais de aprendizado e do comportamento do modelo aluno. A poda, em contraste, reduz a estrutura removendo conexões, canais, cabeças ou camadas de baixa importância. A quantização reduz os custos de armazenamento e computação ao diminuir a largura de bits ou mudar o formato numérico usado para representar pesos e ativações.arxiv.orgarxiv.orgarxiv.org

MétodoO que ele altera principalmentePonto forte típicoCondição a verificar
Destilação de conhecimentoObjetivos de aprendizado e arquitetura do alunoUm modelo pequeno aprende o comportamento do professorQualidade do professor, dados, lacuna de capacidade
PodaConexões, canais, cabeças, camadasRemove redundância de uma estrutura existenteSe o hardware de destino realmente processa estruturas esparsas mais rápido
QuantizaçãoPrecisão numéricaReduz uso de memória e custos de operações inteirasPerda de precisão, suporte de hardware e runtime
Uso combinadoTreinamento e representaçãoPotencial para maior compressãoDegradação de qualidade em cada etapa e complexidade operacional

Por exemplo, converter pesos de ponto flutuante de 32 bits em inteiros de 8 bits é quantização. Treinar um Transformer aluno de seis camadas com a distribuição de saídas de um professor grande como alvo é destilação. Elas também podem ser aplicadas sequencialmente. Porém, o fato de um artigo ter alcançado uma alta taxa de compressão não significa que a latência de um serviço real diminuirá na mesma proporção. Na quantização em particular, o efeito depende de quão bem a CPU, GPU ou NPU de destino oferece suporte às operações inteiras relevantes.arxiv.orgarxiv.org

O que mostram os exemplos da família BERT?

A destilação de modelos de linguagem é um exemplo útil para entender o conceito. O BERT-base é um Transformer pré-treinado amplamente usado em tarefas de compreensão de linguagem, mas pode ser oneroso de implantar em ambientes com recursos restritos. O estudo do DistilBERT apresentou um método de destilação no pré-treinamento que inicializa um modelo menor a partir de algumas camadas do BERT e combina perda de modelagem de linguagem, perda de destilação e perda de distância cosseno.arxiv.org

O estudo relatou a retenção de cerca de 97% do desempenho do BERT no conjunto de desenvolvimento GLUE, ao mesmo tempo que reduziu a quantidade de parâmetros em cerca de 40%. Ele também mostrou tempo de inferência menor nas condições de medição de CPU definidas no artigo. Esses números, contudo, foram obtidos com benchmarks em inglês, uma configuração específica de BERT, determinado hardware e condições de entrada especificadas. Eles não devem ser citados como números universais que garantem os mesmos resultados para português, documentos longos, sistemas com recuperação aumentada, conversas generativas ou outros aceleradores.arxiv.org

O TinyBERT relatou que um modelo menor de quatro camadas manteve forte desempenho em relação ao BERT-base, ao mesmo tempo que reduziu substancialmente o tamanho e o tempo de inferência nas condições do artigo. Também apresentou experimentos de ablação mostrando que a destilação geral, a destilação específica da tarefa, o aumento de dados e os sinais de camadas intermediárias influenciaram o resultado. A lição prática desse exemplo é simples: não basta conectar um professor e um aluno; o momento da destilação, os dados, a arquitetura do aluno e o desenho da perda determinam conjuntamente o resultado.arxiv.org

Quando a destilação é especialmente eficaz?

A destilação vale especialmente a pena não quando a qualidade do modelo grande deve ser preservada perfeitamente, mas quando coexistem um patamar claro de desempenho e fortes restrições operacionais. Alguns exemplos incluem:

  • Inferência no dispositivo: ambientes em que a conectividade de rede é pouco confiável ou em que as entradas são difíceis de enviar a servidores externos devido a requisitos de privacidade.
  • Inferência repetida em alto volume: tarefas de classificação, ranking e detecção que processam muitas entradas curtas, tornando importantes o custo e a latência acumulados.
  • Tarefas específicas de domínio: casos em que o desempenho em um tipo definido de documento ou sistema de rótulos importa mais do que todas as capacidades amplas de um modelo de propósito geral.
  • Substituição de um ensemble de modelos: casos em que a média das previsões de vários modelos beneficia a qualidade, mas é onerosa demais para operar, de modo que um único aluno é usado para aproximar seu comportamento.arxiv.org

Por outro lado, quando o aluno não pode deixar de detectar casos raros ou precisa reproduzir fielmente o longo processo de raciocínio, o uso de ferramentas ou políticas de segurança complexas de um professor, os critérios de avaliação precisam ser mais amplos. Mesmo que a precisão média seja alta, o desempenho pode piorar para grupos específicos, classes raras ou entradas de risco. Pesquisas constataram que a relação entre desempenho na pior classe e precisão média pode variar conforme o desenho do objetivo de destilação.proceedings.mlr.press

Quais são as limitações da destilação?

Primeiro, o aluno aprende não o professor em si, mas o comportamento observável do professor. Se o professor fizer previsões incorretas ou refletir viés dos dados, o aluno também poderá aprender esses sinais. Usar a perda de rótulos rígidos junto com a destilação não resolve automaticamente problemas relacionados a viés, factualidade, segurança ou proteção. A qualidade ainda deve ser verificada após a destilação usando um conjunto de avaliação independente do professor original.arxiv.orgproceedings.mlr.press

Segundo, se o aluno for pequeno demais, ele não conseguirá imitar suficientemente o professor. Isso é chamado de lacuna de capacidade entre professor e aluno. Pesquisas relacionadas relatam que o desempenho pode piorar quando um aluno pequeno tenta seguir sinais de um professor excessivamente grande e que um professor mais preciso nem sempre é um professor melhor para destilação. A seleção de um professor não é apenas uma competição de classificação; também é uma questão de compatibilidade com a arquitetura do aluno.arxiv.orgarxiv.org

Terceiro, a destilação pode deslocar os custos operacionais para custos iniciais de treinamento. Primeiro, o professor precisa ser treinado ou obtido; as saídas do professor para os dados de destilação precisam ser calculadas e armazenadas; e talvez seja necessário explorar temperatura, pesos de perda e mapeamentos de camadas. Portanto, para uma análise executada apenas uma vez, usar o professor diretamente pode ser mais simples; para um serviço repetitivo e de longa duração, há maior oportunidade de recuperar o custo de desenvolver um aluno.arxiv.orgarxiv.org

Quarto, a destilação não garante uma “compressão explicável”. Mesmo que um aluno produza previsões semelhantes às do professor, se ele toma decisões pelos mesmos mecanismos internos e se preserva todas as informações do professor são questões separadas. Pesquisas recentes também apontam que ainda não está claro se um aluno pequeno recebe todas as informações de um professor por meio da destilação.proceedings.mlr.pressproceedings.mlr.press

Quais são os equívocos comuns?

O equívoco de que “a destilação não causa perda de desempenho”

A pesquisa sobre destilação mostra que modelos pequenos podem manter alto desempenho, mas essa é uma observação sob condições de avaliação especificadas. Se a distribuição das entradas mudar, entradas longas se tornarem mais frequentes ou casos raros ganharem importância, a diferença entre professor e aluno poderá aumentar. Portanto, em vez de confiar em um único número, como “uma porcentagem do desempenho do professor”, avalie conjuntamente a precisão, a revocação, a latência, a memória e as métricas de segurança exigidas pelo serviço.arxiv.orgarxiv.org

O equívoco de que “ela copia todo o conhecimento do modelo professor”

Dentro de parâmetros e arquitetura limitados, o aluno aproxima as saídas do professor ou algumas de suas representações. É difícil considerar que todas as informações internas do professor sejam transferidas sem mudanças. Os resultados da destilação dependem dos dados, da função objetivo, da capacidade do aluno e do método de otimização.proceedings.mlr.pressproceedings.mlr.press

O equívoco de que “menos parâmetros necessariamente significam inferência mais rápida”

Reduzir parâmetros geralmente ajuda a reduzir os requisitos de memória, mas a latência real depende do grafo computacional, do comprimento da sequência, do tamanho do lote e dos gargalos de hardware. Em particular, a carga de computação de atenção dos Transformers pode variar substancialmente com o comprimento da entrada; por isso, é necessário fazer medições de ponta a ponta no dispositivo de destino.arxiv.orgarxiv.org

O equívoco de que “as saídas do professor tornam os dados originais desnecessários”

Há pesquisas sobre destilação sem dados ou zero-shot que tentam destilar usando um professor sem os dados de treinamento originais. No entanto, essa é uma configuração especial que exige geração separada de dados sintéticos ou um método para aproximar a distribuição de entradas. A destilação comum usa dados de entrada representativos junto com sinais do professor, e abordagens que também usam rótulos verdadeiros são amplamente utilizadas.proceedings.mlr.pressproceedings.mlr.press

O que verificar antes de implantar a destilação na prática?

É mais seguro partir dos objetivos de implantação, em vez de decidir com base no nome de um modelo. Primeiro, quantifique não “quão pequeno deve ser o modelo”, mas “quais requisitos de qualidade e operação precisam ser atendidos”. A seguir está um checklist mínimo.

  1. Separe os critérios da tarefa. Além da precisão média, defina revocação, os custos de falsos positivos e falsos negativos, o desempenho do pior grupo, o tempo de resposta e os limites de memória.
  2. Valide o professor. Verifique não apenas as pontuações de benchmark do professor, mas também dados reais do domínio, tipos de erro, viés e requisitos de segurança. Um professor não validado não tem garantia de fornecer bons sinais de aprendizado.
  3. Defina primeiro o orçamento do aluno. Especifique claramente a quantidade de parâmetros, a memória máxima, a latência permitida, o comprimento da entrada e o hardware de implantação. Se o aluno for pequeno demais, a lacuna de capacidade pode aumentar.arxiv.org
  4. Escolha o objetivo de destilação. Para classificações simples, a destilação de saídas pode ser um ponto de partida; quando a qualidade das representações importa, como na compressão de Transformers, é possível testar a destilação de camadas intermediárias e de atenção.arxiv.org
  5. Compare professor e aluno com os mesmos testes. Inclua não apenas a precisão geral, mas também comprimentos de entrada semelhantes aos do tráfego real, casos raros e grupos com altas taxas de erro.
  6. Meça no hardware de destino. Diferencie tamanho do arquivo do modelo, memória, carregamento inicial, latência incluindo tokenização e pré-processamento, throughput e consumo de energia.arxiv.orgarxiv.org
  7. Valide separadamente a compressão posterior. Se quantização ou poda forem aplicadas após a destilação, registre separadamente a degradação de qualidade e os ganhos de velocidade após cada etapa.

Conclusão: a destilação cria o equilíbrio operacional adequado, não apenas um “modelo pequeno”

A destilação de modelos de IA transfere distribuições preditivas e representações aprendidas por um professor grande para um aluno, criando um modelo que pode ser usado sob recursos restritos. O ponto principal é adicionar os alvos suaves do professor a um aluno que aprende apenas com rótulos rígidos e, quando necessário, também usar sinais de camadas intermediárias.arxiv.orgarxiv.org

No entanto, a destilação não é uma técnica de compressão de arquivos que preserva o desempenho gratuitamente. Erros e viés do professor, capacidade do aluno, representatividade dos dados de destilação, desenho da temperatura e da perda, e as características do hardware real moldam o resultado. Portanto, o próximo passo mais sensato não é buscar uma taxa de compressão de um artigo específico, mas definir a qualidade mínima e a latência e o custo máximos exigidos pela sua própria tarefa e, então, comparar o professor e os alunos candidatos sob as mesmas condições.

Perguntas frequentes

A destilação de modelos de IA é apenas a compressão de um arquivo de modelo?

Não. A destilação de conhecimento é um método de treinar do zero um modelo aluno menor usando a distribuição de saídas ou representações intermediárias de um modelo professor maior como sinais de aprendizado. Seu mecanismo central é diferente da quantização, que reduz o tamanho do arquivo, e da poda, que remove conexões desnecessárias.

Um modelo professor maior sempre produz um modelo aluno melhor?

Não necessariamente. Se a diferença de capacidade entre o professor e o aluno for grande demais, um problema de lacuna de capacidade poderá impedir que o aluno aprenda suficientemente os sinais do professor. Um professor maior e mais preciso não garante um aluno melhor do que um professor menor.

Um modelo destilado garante a mesma precisão do modelo original?

Não. Estudos como DistilBERT e TinyBERT mostram que modelos menores podem manter alto desempenho sob condições específicas de conjunto de dados e hardware, mas a diferença de desempenho no mundo real depende da tarefa, dos dados, do tamanho das entradas, da arquitetura do aluno e dos hiperparâmetros.

A destilação de conhecimento e a quantização podem ser usadas juntas?

Sim. A destilação trata de como o modelo aluno é treinado, enquanto a quantização trata da representação numérica dos pesos e das operações; portanto, elas abordam problemas diferentes. Porém, quando combinadas, a precisão, a latência e o uso de memória devem ser validados separadamente no hardware de destino.

A destilação pode ser aplicada à IA generativa?

O princípio pode ser aplicado. No entanto, modelos generativos envolvem distribuições de tokens, processos longos de geração, regras de segurança, uso de ferramentas e outros comportamentos, em vez de uma única probabilidade de classificação. Corresponder apenas às saídas finais não demonstra que as capacidades e o comportamento de segurança do modelo original foram reproduzidos de forma idêntica.