Desenvolvendo um Ensemble de Redes Profundas para identificação de Lentes Gravitacionais: Aplicação em Regime de Poucos dados

Resumo

O treinamento de redes neurais profundas com pequenos conjuntos de dados é um desafio em diversos casos.
Isso pode ocorrer quando o fenômeno de interesse é imprevisível ou extremamente raro.
Neste trabalho, analisamos o desempenho da classificação realizada pelas redes profundas ResNet 50, EfficientNet B2 e do Ensemble entre elas. O conjunto de dados utilizado é um banco de imagens simuladas de um fenômeno físico bastante raro: o lenteamento gravitacional. Foram feitos ajustes nos modelos, com o objetivo de obter o melhor desempenho com o menor conjunto de dados de treinamento. Testamos os modelos com diferentes quantidades de dados, analisando o desempenho de cada um a partir da métrica AUC (área sob a curva ROC).