Otimização Colaborativa

_{Mantido por Arm ML Tooling}

Este documento fornece uma visão geral das APIs experimentais para combinar várias técnicas para otimizar modelos de aprendizado de máquina para implantação.

Visão geral

A otimização colaborativa é um processo abrangente que engloba várias técnicas para produzir um modelo que, na implantação, exiba o melhor equilíbrio de características de destino, como velocidade de inferência, tamanho e precisão do modelo.

A ideia das otimizações colaborativas é construir técnicas individuais, aplicando-as uma após a outra para alcançar o efeito de otimização acumulado. Várias combinações das seguintes otimizações são possíveis:

• Poda de peso
• Agrupamento de peso

• Quantização

  • Quantização pós-treinamento
  • Treinamento consciente de quantização (QAT)

O problema que surge ao tentar encadear essas técnicas é que a aplicação de uma normalmente destrói os resultados da técnica anterior, prejudicando o benefício geral de aplicar todas elas simultaneamente; por exemplo, o clustering não preserva a escassez introduzida pela API de remoção. Para resolver este problema, apresentamos as seguintes técnicas experimentais de otimização colaborativa:

• Agrupamento de preservação de esparsidade
• Treinamento consciente de quantização com preservação de esparsidade (PQAT)
• Treinamento com reconhecimento de quantização de preservação de cluster (CQAT)
• Treinamento consciente de quantização com preservação de esparsidade e cluster

Eles fornecem vários caminhos de implantação que podem ser usados ​​para compactar um modelo de aprendizado de máquina e aproveitar a aceleração de hardware no momento da inferência. O diagrama abaixo demonstra vários caminhos de implantação que podem ser explorados na busca do modelo com as características de implantação desejadas, onde os nós folha são modelos prontos para implantação, ou seja, são parcial ou totalmente quantizados e em formato tflite. O preenchimento verde indica as etapas em que o retreinamento/ajuste fino é necessário e uma borda vermelha tracejada destaca as etapas de otimização colaborativa. A técnica utilizada para obter um modelo em um determinado nó é indicada no rótulo correspondente.

O caminho de implantação direto, somente de quantização (pós-treinamento ou QAT) é omitido na figura acima.

A ideia é alcançar o modelo totalmente otimizado no terceiro nível da árvore de implantação acima; no entanto, qualquer um dos outros níveis de otimização pode ser satisfatório e atingir a compensação necessária de latência/precisão de inferência, caso em que nenhuma otimização adicional é necessária. O processo de treinamento recomendado seria percorrer iterativamente os níveis da árvore de implantação aplicáveis ​​ao cenário de implantação de destino e verificar se o modelo atende aos requisitos de latência de inferência e, se não, usar a técnica de otimização colaborativa correspondente para compactar ainda mais o modelo e repetir até que o modelo seja totalmente otimizado (removido, agrupado e quantizado), se necessário.

A figura abaixo mostra os gráficos de densidade do kernel de peso de amostra passando pelo pipeline de otimização colaborativa.

O resultado é um modelo de implantação quantizado com um número reduzido de valores exclusivos, bem como um número significativo de pesos esparsos, dependendo da esparsidade alvo especificada no tempo de treinamento. Além das vantagens significativas de compactação do modelo, o suporte de hardware específico pode aproveitar esses modelos esparsos e em cluster para reduzir significativamente a latência de inferência.

Resultados

Abaixo estão alguns resultados de precisão e compactação que obtivemos ao experimentar os caminhos de otimização colaborativa PQAT e CQAT.

Treinamento consciente de quantização com preservação de esparsidade (PQAT)

Treinamento com reconhecimento de quantização para preservação de cluster (CQAT)

Resultados CQAT e PCQAT para modelos em cluster por canal

Os resultados abaixo são obtidos com a técnica de agrupamento por canal . Eles ilustram que, se as camadas convolucionais do modelo forem agrupadas por canal, a precisão do modelo será maior. Se o seu modelo tiver muitas camadas convolucionais, recomendamos agrupar por canal. A taxa de compressão permanece a mesma, mas a precisão do modelo será maior. O pipeline de otimização do modelo é 'clustered -> cluster preservando QAT -> post training quantization, int8' em nossos experimentos.

Exemplos

Para obter exemplos completos das técnicas de otimização colaborativa descritas aqui, consulte os notebooks de exemplo CQAT , PQAT , clustering de preservação de esparsidade e PCQAT .