Skip to main navigation Skip to search Skip to main content

QSAN: A Near-Term Achievable Quantum Self-Attention Network

  • Jinjing Shi (Co-first Author)
  • , Ren-Xin Zhao (Co-first Author)
  • , Wenxuan Wang
  • , Shichao Zhang*
  • , Xuelong Li
  • *Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

4 Downloads (CityUHK Scholars)

Abstract

Self-attention mechanism (SAM) is good at capturing the intrinsic connection between features to dramatically boost the performance of machine learning models. Nevertheless, the capability of SAM is not equipped with many current quantum machine learning (QML) models, thus confining their expansion on massive high-dimensional quantum data. To address the above problems, a quantum SAM (QSAM) consisting of a quantum logic similarity (QLS)-based quantum bit self-attention score matrix (QBSASM) is introduced to augment the data representation of SAM exponentially. According to QSAM, the framework and quantum circuits of a one-step achievable quantum self-attention network (QSAN) are designed to consider measurement times compression fully. Moreover, a prototype of quantum coordinates is presented during the design process to describe the mathematical relationship between the output bits and the control bits to facilitate the programming. Ultimately, MNIST binary classification experiments on the PennyLane platform and comparisons with cutting-edge QML models demonstrate QSAN converges about 1.7 × and 2.3 × faster than hardware-efficient ansatz and quantum approximate optimization algorithm (QAOA) ansatz, respectively, with similar parameter configurations and 100% prediction accuracy, which indicates that it has a better learning capability. In the CIFAR-10 classification experiments, QSAN achieves high prediction accuracy at a small scale relative to classical machine learning models. Predictably, QSAN elevates the efficiency of QML models and lays the foundation for future quantum computers to perform machine learning on massive amounts of data while promoting the advancement of quantum computer vision and other fields. © 2024 The Authors.
Original languageEnglish
Pages (from-to)13995-14008
Number of pages14
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume36
Issue number8
Online published16 Dec 2024
DOIs
Publication statusPublished - Aug 2025

Funding

This work was supported in part by the National Natural Science Foundation of China under Grant 62272483, in part by the Natural Science Foundation for Distinguished Young Scholars of Hunan Province under Grant 2023JJ10078, in part by the Special Foundation for Distinguished Young Scientists of Changsha under Grant kq1905058, and in part by the Research Fund of Guangxi Key Lab of Multi-Source Information Mining & Security under Grant MIMS24-05.

Research Keywords

  • Image classification
  • machine learning
  • quantum circuit
  • quantum machine learning (QML)
  • quantum neural network
  • quantum self-attention mechanism (QSAM)
  • self-attention mechanism (SAM)

Publisher's Copyright Statement

  • This full text is made available under CC-BY 4.0. https://creativecommons.org/licenses/by/4.0/

Fingerprint

Dive into the research topics of 'QSAN: A Near-Term Achievable Quantum Self-Attention Network'. Together they form a unique fingerprint.

Cite this