QTN-MLP: Quantum Tensor Network-enhanced MLP for Medical Image Classification
Geng Zhang, Hong Lai
Recently, there has been growing interest in medical image analysis methods based on multi-layer perceptron (MLP), which have demonstrated remarkable performance in this field. Typically, the MLP has been employed to flatten feature maps along the channel and spatial dimensions and encode them to capture the overall feature information. However, for MLP-based architecture, the whole learning process requires high computational complexity, a large number of parameters, and large-scale training datasets. Meanwhile, quantum tensor network (QTN), derived from quantum many-body systems, have gained popularity in machine learning due to their ability to reduce the number of parameters and explore the correlation of remote elements in an image through a tensor contraction operation. In this paper, we propose a novel framework named QTN-MLP based on QTN and MLP to tackle the weaknesses of MLP. This paper has been extensively validated on the COVID-CT dataset and ISIC 2018 dataset, and the experiments show that our method competes with state-of-the-art methods with lower computational complexity and fewer parameters.
International Conference on Computer Supported Cooperative Work in Design (CSCWD)•May 8, 2024•1 citation