用量子启发方法压缩嵌入向量,参数少32倍仍保持竞争力。
Quantum-inspired Embeddings Projection and Similarity Metrics for Representation Learning
- 将经典嵌入映射到希尔伯特空间,用量子电路降维
- 在TREC数据集上参数减少32倍,性能相当甚至更优
- 适合资源受限场景,尤其小数据训练效果显著
近十年来,表示学习通过将海量数据中的复杂信息嵌入稠密向量空间,成为机器学习核心技术。其在大语言模型和基于对比学习的计算机视觉系统中尤为关键。表示学习系统的核心组件是投影头,用于将原始嵌入映射到不同、常被压缩的空间,同时保留向量间的相似关系。本文提出一种量子启发的投影头及其配套的相似度度量方法。具体地,将经典嵌入映射至希尔伯特空间的量子态,并引入基于量子电路的投影头以降低嵌入维度。为验证有效性,我们扩展了BERT语言模型,集成该投影头实现嵌入压缩。在TREC 2019与TREC 2020深度学习基准的信息检索任务中,对比使用量子启发投影头与传统投影头压缩的嵌入表现,结果表明:本方法在参数量减少32倍的前提下,性能具有竞争力;尤其在从零开始训练时,在小数据集上表现尤为突出。该工作不仅验证了量子启发方法的有效性,也强调了神经网络中高效、特定低纠缠电路模拟作为强大量子启发技术的潜力。
原文摘要 · Abstract (English)
Over the last decade, representation learning, which embeds complex information extracted from large amounts of data into dense vector spaces, has emerged as a key technique in machine learning. Among other applications, it has been a key building block for large language models and advanced computer vision systems based on contrastive learning. A core component of representation learning systems is the projection head, which maps the original embeddings into different, often compressed spaces, while preserving the similarity relationship between vectors. In this paper, we propose a quantum-inspired projection head that includes a corresponding quantum-inspired similarity metric. Specifically, we map classical embeddings onto quantum states in Hilbert space and introduce a quantum circuit-based projection head to reduce embedding dimensionality. To evaluate the effectiveness of this approach, we extended the BERT language model by integrating our projection head for embedding compression. We compared the performance of embeddings, which were compressed using our quantum-inspired projection head, with those compressed using a classical projection head on information retrieval tasks using the TREC 2019 and TREC 2020 Deep Learning benchmarks. The results demonstrate that our quantum-inspired method achieves competitive performance relative to the classical method while utilizing 32 times fewer parameters. Furthermore, when trained from scratch, it notably excels, particularly on smaller datasets. This work not only highlights the effectiveness of the quantum-inspired approach but also emphasizes the utility of efficient, ad hoc low-entanglement circuit simulations within neural networks as a powerful quantum-inspired technique.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。