Publicado: Sep 8, 2022
O Amazon SageMaker oferece um conjunto de algoritmos incorporados, modelos pré-treinados e modelos de solução incorporados para ajudar cientistas de dados e profissionais de machine learning a começar a fazer treinamentos e implantar modelos de machine learning com rapidez. Esses algoritmos e modelos podem ser usados para aprendizado com ou sem supervisão. Eles podem processar vários tipos de dados de entrada, como dados tabulares, imagens e textos.
A partir de hoje, o Amazon SageMaker oferece um novo algoritmo incorporado para classificação de imagens, o Image Classification – TensorFlow. Ele é um algoritmo de aprendizado supervisionado que permite o aprendizado por transferência em vários modelos pré-treinados disponíveis no TensorFlow Hub. O algoritmo recebe uma imagem como entrada e calcula a probabilidade de cada um dos rótulos de classe. É possível refinar o ajuste desses modelos pré-treinados usando aprendizado por transferência, mesmo quando não há disponibilidade de um grande número de imagens de treinamento. Ele está disponível nos algoritmos incorporados do SageMaker e no SageMaker JumpStart UI do SageMaker Studio.
O Image Classification – TensorFlow no Amazon SageMaker oferece aprendizado por transferência em vários modelos pré-treinados disponíveis no TensorFlow Hub. Em machine learning, a capacidade de utilizar resultados de treinamento de um modelo para gerar outro modelo é chamada de aprendizado por transferência. Uma camada de classificação é anexada ao modelo pré-treinado no TensorFlow Hub de acordo com o número de rótulos de classe nos dados de treinamento. A camada de classificação consiste em uma camada de descarte e uma camada densa, totalmente conectada, com regularizador de norma-2 e inicializada com pesos aleatórios. O treinamento de modelos tem hiperparâmetros para a taxa de descarte da camada de descarte e o fator de regularização L2 da camada densa. Dessa forma, é possível fazer o refinar o ajuste de toda a rede (incluindo o modelo pré-treinado) ou apenas da camada principal de classificação com os novos dados de treinamento. O algoritmo oferece uma grande variedade de hiperparâmetros de treinamento para o refinar o ajuste com conjuntos de dados personalizados.
Para aprender a usar este algoritmo, consulte Image Classification - TensorFlow na documentação da AWS e o exemplo de notebook Introduction to SageMaker TensorFlow - Image Classification. Para ver uma explicação mais detalhada sobre como usar esses algoritmos, consulte a publicação Transfer learning for TensorFlow image classification models in Amazon SageMaker.
Todos esses algoritmos podem ser usados em todas as regiões em que o Amazon SageMaker está disponível. Para começar a usar esses novos modelos no SageMaker, consulte a documentação.