arXiv:2505.20251cs.LGcs.CL2025-05ICML

用马尔可夫链数据训练模型,实现高效序列外推设计

Learning Extrapolative Sequence Transformations from Markov Chains

  • 从MCMC生成的马尔可夫链中提取状态训练自回归模型
  • 在蛋白质设计等任务中外推效果优于或等同于MCMC
  • 样本效率高,适合需要快速生成新序列的场景

深度学习多数应用依赖与训练条件相似的测试环境。然而,生物序列设计等任务需寻找性能超越已有值的新序列,这要求模型能对训练数据进行外推。此类任务可通过马尔可夫链蒙特卡洛(MCMC)实现,即从初始状态出发,采样局部变换以逼近奖励目标属性的分布。但即便使用精心设计的提议函数,MCMC在大型结构化状态空间中仍可能探索效率低下。与其依赖随机搜索,更理想的是构建一个贪婪优化目标属性的模型,以最少步骤实现有效外推。本文提出从MCMC搜索产生的马尔可夫链中学习此类模型:选取链中状态作为训练数据,训练自回归模型,使其能高效生成沿目标属性外推的新序列。该方法在蛋白质序列设计、文本情感控制和文本匿名化三个任务上验证,结果表明,自回归模型在外推性能上不逊于甚至优于MCMC,且具备更好的可扩展性和显著更高的样本效率。

原文摘要 · Abstract (English)

Most successful applications of deep learning involve similar training and test conditions. However, tasks such as biological sequence design involve searching for sequences that improve desirable properties beyond previously known values, which requires novel hypotheses that \emph{extrapolate} beyond training data. In these settings, extrapolation may be achieved by using random search methods such as Markov chain Monte Carlo (MCMC), which, given an initial state, sample local transformations to approximate a target density that rewards states with the desired properties. However, even with a well-designed proposal, MCMC may struggle to explore large structured state spaces efficiently. Rather than relying on stochastic search, it would be desirable to have a model that greedily optimizes the properties of interest, successfully extrapolating in as few steps as possible. We propose to learn such a model from the Markov chains resulting from MCMC search. Specifically, our approach uses selected states from Markov chains as a source of training data for an autoregressive model, which is then able to efficiently generate novel sequences that extrapolate along the sequence-level properties of interest. The proposed approach is validated on three problems: protein sequence design, text sentiment control, and text anonymization. We find that the autoregressive model can extrapolate as well or better than MCMC, but with the additional benefits of scalability and significantly higher sample efficiency.

序列生成外推自回归模型MCMC

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