Método de classificação semi-supervisionado baseado no treinamento conjunto de redes generativas adversariais

XU Zhe ,  

GENG Jie ,  

JIANG Wen ,  

ZHANG Zhuo ,  

ZENG Qing-jie ,  

摘要

Redes neurais profundas exigem grandes quantidades de dados para o aprendizado supervisionado, porém, na prática, muitas vezes é difícil obter muitos dados rotulados. O aprendizado semi-supervisionado pode reduzir a dependência de redes profundas em dados rotulados. Redes generativas adversariais baseadas em aprendizado semi-supervisionado podem melhorar o desempenho da classificação, mas ainda enfrentam problemas de instabilidade no treinamento. Para melhorar ainda mais a precisão da classificação e resolver o problema da instabilidade do treinamento, este artigo propõe um método de classificação semi-supervisionado baseado no treinamento conjunto de redes generativas adversariais, que utiliza o treinamento conjunto de dois discriminadores para eliminar o erro de distribuição de um único discriminador, ao mesmo tempo que seleciona amostras de alta confiança de dados não rotulados para expandir o conjunto de dados rotulados, aumentando a precisão da classificação semi-supervisionada e a capacidade de generalização do modelo de rede. Os resultados experimentais nos conjuntos de dados CIFAR-10 e SVHN mostram que o método proposto obtém melhor precisão de classificação independentemente da quantidade de dados rotulados. Quando o número de rótulos é 2.000, a precisão na base CIFAR-10 atinge 80,36%; quando o número de rótulos é 10, a precisão da classificação melhora cerca de 5% em comparação com os métodos semi-supervisionados existentes. Em certa medida, resolve o problema de overfitting das redes GAN em condições de amostras pequenas.

关键词

Redes generativas adversariais;Aprendizado semi-supervisionado;Classificação de imagens;Aprendizado profundo

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website