Método de classificação de imagem óssea baseado em ACGAN e aprendizado por transferência

YU Hong ,  

LUO Renze ,  

CHEN Chunmeng ,  

TANG Xiang ,  

LUO Renquan ,  

摘要

Devido à quantidade limitada de amostras e ao desequilíbrio de classes na imagem óssea, a classificação da imagem óssea apresenta grandes dificuldades. Para melhorar a precisão da classificação da imagem óssea, este artigo propõe um método de classificação baseado na geração de dados por meio de uma rede adversarial generativa condicional com classificador auxiliar (ACGAN) e aprendizado por transferência. Primeiramente, para resolver o problema do desequilíbrio de classes na imagem óssea, foi projetado o modelo MU-ACGAN. Esse modelo utiliza o U-Net como estrutura do gerador, combinando conexões residuais densas e um mecanismo de atenção canal-espacial para melhorar a geração de características detalhadas da imagem óssea; o discriminador utiliza blocos convolucionais residuais densos com atenção para extrair características da imagem óssea para classificação; em seguida, foi combinada a abordagem tradicional de aumento de dados para expandir ainda mais a quantidade de dados; finalmente, foi projetada uma rede neural convolucional em múltiplas escalas para extrair características da imagem óssea em diferentes escalas, melhorando o desempenho da classificação. Durante o treinamento do modelo, foi adotada uma abordagem de aprendizado por transferência em duas fases para otimizar os parâmetros iniciais do modelo e resolver o problema de overfitting. Os resultados experimentais mostram que o método proposto alcançou uma precisão de classificação de 85,71%, aliviando efetivamente o problema de baixa precisão de classificação em conjuntos de dados ósseos com pequenas amostras.

关键词

imagem óssea; rede adversarial generativa condicional com classificador auxiliar (ACGAN); aprendizado por transferência; mecanismo de atenção; aumento de dados

阅读全文

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

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