短elist模型:基于单形扩散的离散变量生成模型
机器学习
2025-08-26 v1 基因组学
摘要
离散变量的生成建模虽具挑战性,却在自然语言处理和生物序列设计等应用中至关重要。我们引入短elist模型(ShortListing Model, SLM),一种受逐步候选剔除启发的新型单形扩散模型。SLM 在单形质心上运行,减少生成复杂度并提升可扩展性。此外,SLM 采用灵活的无分类器引导实现,增强无条件生成性能。在DNA启动子和增强子设计、蛋白质设计、字符级和大词汇量语言建模等实验中,SLM 显示出竞争性能和强大潜力。我们的代码可在 https://github.com/GenSI-THUAIR/SLM 查看。
引用
@article{arxiv.2508.17345,
title = {ShortListing Model: A Streamlined SimplexDiffusion for Discrete Variable Generation},
author = {Yuxuan Song and Zhe Zhang and Yu Pei and Jingjing Gong and Qiying Yu and Zheng Zhang and Mingxuan Wang and Hao Zhou and Jingjing Liu and Wei-Ying Ma},
journal= {arXiv preprint arXiv:2508.17345},
year = {2025}
}