用量子生成对抗网络设计新药,提升分子结构生成质量。
Latent Style-based Quantum Wasserstein GAN for Drug Design
- 在量子电路中每旋转门注入噪声,结合梯度惩罚防模式崩溃。
- 156量子比特真实硬件上生成分子,相比经典模型提升多样性。
- 适合对量子机器学习和药物设计感兴趣的科研人员。
新药研发平均成本高达约25亿美元,耗时且昂贵。近年来,人工智能辅助的从头药物设计快速发展,生成式AI显著降低了成本与时间。然而,经典生成对抗网络(GAN)训练困难,易出现平坦区和模式崩溃问题。量子计算或可缓解此问题,减少参数量并提升泛化能力。本文提出一种基于潜空间风格的量子生成对抗网络(QGAN),在量子电路每个旋转门中引入噪声,并在损失函数中加入梯度惩罚以抑制模式崩溃。该流程首先通过变分自编码器将分子结构映射至潜空间,再输入QGAN生成。基线模型在最多15量子比特的量子模拟器上验证,推理阶段使用156量子比特的IBM Heron量子计算机(五量子比特设置)评估真实量子硬件的影响。在MOSES基准套件下与经典模型对比,结果表明该方法生成分子更具多样性与有效性。
原文摘要 · Abstract (English)
The development of new drugs is a tedious, time-consuming, and expensive process, for which the average costs are estimated to be up to around $2.5 billion. The first step in this long process is the design of the new drug, for which de novo drug design, assisted by artificial intelligence, has blossomed in recent years and revolutionized the field. In particular, generative artificial intelligence has delivered promising results in drug discovery and development, reducing costs and the time to solution. However, classical generative models, such as generative adversarial networks (GANs), are difficult to train due to barren plateaus and prone to mode collapse. Quantum computing may be an avenue to overcome these issues and provide models with fewer parameters, thereby enhancing the generalizability of GANs. We propose a new style-based quantum GAN (QGAN) architecture for drug design that implements noise encoding at every rotational gate of the circuit and a gradient penalty in the loss function to mitigate mode collapse. Our pipeline employs a variational autoencoder to represent the molecular structure in a latent space, which is then used as input to our QGAN. Our baseline model runs on up to 15 qubits to validate our architecture on quantum simulators, and a 156-qubit IBM Heron quantum computer in the five-qubit setup is used for inference to investigate the effects of using real quantum hardware on the analysis. We benchmark our results against classical models as provided by the MOSES benchmark suite.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。