QTN-MLP: Quantum Tensor Network-enhanced MLP for Medical Image Classification

Geng Zhang,Hong Lai
DOI: https://doi.org/10.1109/CSCWD61410.2024.10580373
2024-05-08
Abstract: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.
Computer Science,Medicine
What problem does this paper attempt to address?