Deep neural networks have greatly improved medical image classification in a variety of applications and modalities. However, designing these networks manually is often time-consuming and suboptimal. This process can be done automatically by using Neural Architecture Search (NAS), which may offer the potential to identify more efficient and effective models. This paper presents an innovative approach that combines Variational Graph Autoencoders (VGAE) and Evolutionary Algorithms (EA) to optimize the NAS framework, with the goal of efficiently finding deep learning models for medical image classification tasks. Specifically, we leverage the NAS-Bench-101 dataset as the candidate architecture pool and utilize the VGAE model to encode and decode the architecture information of the neural network, thereby simplifying the architecture search process. Additionally, we integrate an evolutionary algorithm to explore the latent space, enhancing the model performance. Our method is validated on several datasets under MedMNIST. Experimental results demonstrate the effectiveness of our framework.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Neural Architecture Search for Medical Image Classification via Latent Space and Evolutionary Optimization

  • Jiawen Dai,
  • Kevin Hung,
  • Kwok Tai Chui,
  • Raymond Ho,
  • Jialiang Gu

摘要

Deep neural networks have greatly improved medical image classification in a variety of applications and modalities. However, designing these networks manually is often time-consuming and suboptimal. This process can be done automatically by using Neural Architecture Search (NAS), which may offer the potential to identify more efficient and effective models. This paper presents an innovative approach that combines Variational Graph Autoencoders (VGAE) and Evolutionary Algorithms (EA) to optimize the NAS framework, with the goal of efficiently finding deep learning models for medical image classification tasks. Specifically, we leverage the NAS-Bench-101 dataset as the candidate architecture pool and utilize the VGAE model to encode and decode the architecture information of the neural network, thereby simplifying the architecture search process. Additionally, we integrate an evolutionary algorithm to explore the latent space, enhancing the model performance. Our method is validated on several datasets under MedMNIST. Experimental results demonstrate the effectiveness of our framework.