[控场AI]
· 4 分钟阅读· 2,296 字

OrderGPT:生成顺序也能学习,AI建模新突破

OrderGPT:生成顺序也能学习,AI建模新突破

OrderGPT让模型从数据中学习生成顺序,在自回归与掩码扩散之间找到中间路线。

OrderGPT 提出将生成顺序作为可学习参数,而非预先固定的超参数。它构建在任意顺序 GPT 之上,通过从随机揭示前缀的预测中提取重复出现的注意力模式,聚合为互惠依赖图,再用谱教师推导连续生成优先级,并由轻量 OrderMLP 在单次训练中预测顺序。训练早期探索顺序,后期冻结顺序用于自回归生成。在语言上模型恢复接近从左到右的结构,在人体姿态上学习到解剖学组织化的分解,在表格数据上发现语义相关字段邻近的分解并提升下游指标。该工作将生成顺序从人工先验转变为可学习参数,为缺乏自然顺序的结构化离散数据提供了介于自回归与扩散之间的新选择。

离散数据的生成建模长期被两种范式主导:GPT式自回归模型(从左到右逐词生成)和掩码扩散模型(任意顺序生成)。看似对立的两条路线,实则是同一根轴上的两个极端——生成顺序。对于没有天然遍历顺序的结构化离散数据(如人体姿态、表格数据),这两个极端都可能不是最优解。近期一篇名为OrderGPT的论文提出,介于两者之间的“单一适应顺序”是被忽视的中间地带,并给出了一种让模型自己学习生成顺序的方法。

OrderGPT论文概览

从两极到连续谱:生成顺序的再思考

自回归模型如GPT将序列概率分解为从左到右的条件概率链,这种顺序在自然语言中很自然,因为语言本身有线性结构。掩码扩散模型则允许任意顺序生成,本质上是一种任意顺序自回归。但很多离散领域并没有“从左到右”的天然概念,比如人体姿态由多个关节坐标组成,表格数据由多个字段组成。对这些数据强行套用从左到右的顺序,可能引入不合理的依赖假设。

OrderGPT的核心洞察在于:生成顺序本身可以作为一个可学习的建模选择,而不是预先固定的超参数。论文没有在两种极端之间做非此即彼的选择,而是探索了“学习一个任务级的单一顺序”这一中间路线。

任意顺序自回归(Any-Order Autoregressive)是自回归建模的一种推广形式:它不要求模型严格按从左到右的顺序分解联合概率,而是允许以任意排列依次对变量做条件预测。其数学表达为 ( p(x) = \prod_{t} p(x_{\pi_t} \mid x_{\pi_{<t}}) ),其中 (\pi) 是变量的一种排列。掩码扩散模型可以看作任意顺序自回归的特例——每一步随机选择若干位置进行预测,被预测的位置逐步加入“已知”集合。这一视角的重要性在于:自回归与扩散并非本质对立的两种模型,区别仅在于生成顺序的约束程度。OrderGPT 正是利用了这一连续谱的框架,把顺序从外部给定变为内部学习。

OrderGPT如何学习生成顺序?

OrderGPT构建在Any-Order GPT之上,后者支持任意顺序的预测,但需要显式给定顺序。OrderGPT的创新在于无需规范顺序监督,就能从数据中学习到适合任务的生成顺序。

具体机制分为三步:

  1. 从随机前缀中提取依赖信号:模型在训练时从随机采样的揭示前缀(reveal prefix)预测变量。关键观察是,这种预测会在骨干网络中产生重复出现的注意力模式,这些模式提供了模型相关的预测依赖信号。
  2. 聚合为依赖图并推导优先级:将注意力信号聚合为互惠依赖图,再用谱教师(spectral teacher)推导出连续的生成优先级。
  3. 轻量OrderMLP预测顺序:一个集成在GPT骨干中的轻量级OrderMLP从注意力中学习预测这些优先级,将顺序学习纳入架构,在单次训练中完成。

训练过程会逐渐向学到的顺序偏移,随后该顺序被冻结,用于后续的自回归训练和生成。这意味着模型在训练早期探索顺序,后期固定顺序以稳定生成。

谱教师(spectral teacher)指利用图拉普拉斯矩阵的谱分解来推导节点排序的一种方法。给定一个表示变量间依赖强度的互惠依赖图,可以构造其拉普拉斯矩阵 (L = D - A)((D) 为度矩阵,(A) 为邻接矩阵),并取第二小特征值对应的特征向量(Fiedler 向量)。该向量的各分量大小给出了一种连续的节点排序:分量相近的节点在图上连接更紧密,应被安排在相近的生成步骤。相比直接使用图搜索或贪心排序,谱方法对噪声更鲁棒,且天然输出连续优先级,便于后续用回归目标训练 OrderMLP。

揭示前缀(reveal prefix)是训练时对输入变量的一种随机部分观测:随机选取一部分变量作为已知条件(即“已揭示”),其余变量作为预测目标。在任意顺序自回归框架下,每次训练迭代会采样不同的揭示集合,使模型见到多样的条件组合。OrderGPT 的关键观察是,尽管揭示集合是随机的,模型在预测某个目标变量时,注意力会稳定地集中在某些已揭示变量上——这种跨样本重复出现的注意力模式,正是数据中真实依赖结构的信号。通过聚合大量随机前缀下的注意力,模型无需任何顺序监督即可估计出变量间的依赖关系。

跨模态验证:语言、姿态与表格数据

论文在多个模态上验证了OrderGPT的适应性:

  • 语言:模型自适应地恢复了接近从左到右的依赖结构,说明语言的自然顺序确实是最优的之一,方法没有强行扭曲语言建模。
  • 人体姿态:学习到解剖学上组织化的分解方式,即按照人体结构而非任意坐标顺序生成关节。
  • 表格数据:发现了将统计和语义相关字段放在附近的分解方式,同时提升了所有领域的下游建模指标。

这些结果说明,生成顺序的学习不仅没有损害语言建模,还能在缺乏自然顺序的领域带来实质性收益。

意义与展望:顺序学习打开新空间

OrderGPT的价值在于将生成顺序从人工先验转变为可学习参数。对于结构化离散数据,这提供了一种介于自回归和扩散之间的新选择。代码已在GitHub开源(Chenhe-yang/OrderGPT),社区可以进一步探索顺序学习在更多领域的应用。

一个值得思考的方向是:顺序学习是否也能反哺语言模型?虽然语言有自然顺序,但某些任务(如代码生成、数学推理)可能存在更优的生成顺序。OrderGPT的框架为这类探索提供了基础工具。

#OrderGPT#生成顺序#自回归模型#掩码扩散#任意顺序GPT#离散数据生成#注意力机制#结构化数据建模
分享:

相关推荐