arXiv:2502.10631cs.LGcs.AI2025-02被引 4

提出可精准控制分子序列生成的ControllableGPT,用于药物优化

ControllableGPT: A Ground-Up Designed Controllable GPT for Molecule Optimization

  • 融合MLM、CLM与seq2seq优势,实现序列增删改可控
  • 在病毒与癌症药物任务上超越现有基线模型
  • 适合需要精确分子结构调控的药物研发人员

大语言模型常用掩码语言模型(MLM)、自回归语言模型(CLM)和序列到序列模型(seq2seq)三种训练方式,但各有局限,难以满足药物优化等需可控双向生成的任务。受生物生长与进化过程启发,我们提出ControllableGPT,首次将MLM、CLM与seq2seq优势整合为统一可控的GPT框架。该框架可对序列特定位置或区间进行精确控制,实现任意长度的扩展、缩减或突变,同时保持指定位置或子序列的完整性。本文从零设计该模型,提出因果掩码seq2seq(CMS)目标函数,构建训练语料库,引入新型预训练方法,并设计独特生成流程。在病毒与癌症药物优化任务上验证了其有效性与可控性,性能优于现有基线。

原文摘要 · Abstract (English)

Large Language Models (LLMs) employ three popular training approaches: Masked Language Models (MLM), Causal Language Models (CLM), and Sequence-to-Sequence Models (seq2seq). However, each approach has its strengths and limitations, and faces challenges in addressing specific tasks that require controllable and bidirectional generation, such as drug optimization. To address this challenge, inspired by the biological processes of growth and evolution, which involve the expansion, shrinking, and mutation of sequences, we introduce ControllableGPT. This initiative represents the first effort to combine the advantages of MLM, CLM, and seq2seq into a single unified, controllable GPT framework. It enables the precise management of specific locations and ranges within a sequence, allowing for expansion, reduction, or mutation over chosen or random lengths, while maintaining the integrity of any specified positions or subsequences. In this work, we designed ControllableGPT for drug optimization from the ground up, which included proposing the Causally Masked Seq2seq (CMS) objective, developing the training corpus, introducing a novel pre-training approach, and devising a unique generation process. We demonstrate the effectiveness and controllability of ControllableGPT by conducting experiments on drug optimization tasks for both viral and cancer benchmarks, surpassing competing baselines.

药物生成可控生成分子优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。