← 返回 2026-01-12

GenCtrl——生成模型的形式化可控性工具包 GenCtrl -- A Formal Controllability Toolkit for Generative Models

Emily Cheng, Carmen Amo Alonso, Federico Danieli, Arno Blaas, Luca Zappella, Pau Rodriguez, Xavier Suau 📅 2026-01-09 👍 5 2026-07-13 08:35
PAC学习 可控性 形式化验证 控制理论 生成模型安全

用控制理论形式化分析生成模型的可控性,发现可控性是脆弱且高度依赖任务的

前置知识

可控性 (Controllability)

控制理论中的核心概念,描述系统能否从任意初始状态在有限时间内到达任意目标状态。形式化地,如果对于状态空间X中任意初始状态x0和任意目标状态xf,都存在一个控制输入序列能在有限时间内将系统从x0驱动到xf,则系统是可控的。在生成模型语境下,这意味着从任何初始提示出发,都能通过某种提示策略达到任何期望的输出属性。

这是论文的核心研究对象。理解可控性的形式化定义是理解整篇论文的基础,也是区分本文与传统可控生成工作的关键。

可达集 (Reachable Set)

从给定初始状态x0出发,通过所有可能的控制输入序列u0,...,u_{t-1},在时刻t能达到的所有输出测量值的集合,记为R(x0, U, t)。可达集刻画了模型从特定初始提示出发能达到的能力边界。例如,如果初始提示是写一个故事,可达集就是通过所有可能的后续提示,模型能生成的所有正式性分数的集合。

可达集是可控集的基础构建块,可控集是所有初始状态可达集的交集。理解可达集是理解论文算法的关键。

蒙特卡洛方法 (Monte Carlo Method)

通过随机采样来估计复杂系统性质的计算方法。本文使用蒙特卡洛方法估计生成模型的可达集和可控集:从初始状态和输入分布中采样大量轨迹,然后用经验频率逼近真实集合。由于生成模型的动态是未知的非线性函数,无法解析计算可达集,必须依赖采样估计。

论文的算法1a和1b都是蒙特卡洛算法。理解蒙特卡洛方法的基本原理是理解这些算法如何工作的前提。

PAC学习 (Probably Approximately Correct Learning)

计算学习理论的基础框架,提供学习算法的理论保证。PAC保证的形式是:以至少1-delta的概率(probably),算法输出的假设与真实目标概念的误差小于epsilon(approximately correct)。在本文中,PAC保证确保估计的可达集/可控集以高概率接近真实集合。

论文的两个主要定理(Theorem 1和2)都是PAC形式的保证。理解PAC框架是理解这些定理含义的关键。

离散瓶颈 (Discrete Bottleneck)

生成模型的一个根本特性:由于LLM的词汇表是有限集合,T2IM的提示空间也是离散的,即使测量值本身是连续的(如正式性分数),真正的可达集也可能是可数的(而非连续的)。例如,LLM在tau步内最多生成|Sigma|^tau种不同输出,其中|Sigma|是词汇表大小。这导致现有的连续可达集估计方法会返回过大的(vacuously large)估计。

离散瓶颈是本文需要引入gamma-量化可达性和粗粒度可达性概念的根本原因,也是本文方法与现有控制理论方法的关键区别。

研究动机

当前生成模型的可控性研究存在一个根本性的盲点:所有控制方法(从提示工程到微调到表示工程)都隐含地假设模型是可控的,但这个假设从未被形式化验证。这个假设建立在三个未验证的前提上:可达性(期望输出可以通过给定控制机制达到)、普遍可控性(从任何初始状态都能达到期望输出)、校准性(输出是控制变量的直接函数)。在实际应用中,当用户要求LLM生成特定正式性水平的文本,或者要求T2IM生成特定数量的物体时,我们没有任何工具来判断这些目标是否真的可达,或者从任何初始提示出发是否都能达到。这导致研究者在模型本质上不可控的情况下盲目尝试控制,浪费资源并可能产生虚假的安全感。

本文的目标是本文的目标是建立一个形式化框架来严格验证生成模型的可控性。具体而言:(1)定义对话过程中生成模型的可达集和可控集;(2)设计蒙特卡洛算法来估计这些集合;(3)提供PAC理论保证,给出估计误差的上界;(4)通过实验验证框架的实用性,并揭示当前模型可控性的真实状况。最终目标是推动一个范式转变:从盲目假设可控性转向首先分析可控性的基本限制。

与已有工作不同的是,本文的独特视角在于它抓住了一个被整个领域忽视的基本问题:在尝试控制生成模型之前,首先应该问这个模型是否可控。现有工作都聚焦于如何控制(设计更好的控制方法),而本文聚焦于是否可控制(验证控制的可能性)。这个视角的转换带来了几个关键区别:(1)本文不设计新的控制方法,而是提供评估现有控制机制的工具;(2)本文不追求更高的控制精度,而是给出控制可能性的理论边界;(3)本文不假设模型是可控的,而是将可控性作为需要验证的假设。这种视角转换使得本文成为第一个提供生成模型可控性形式化分析的工作。

核心方法

论文的核心思路可以用一个类比来理解:想象你在一个黑暗的房间里,想要到达房间里的某个位置。传统的可控生成研究专注于设计更好的导航策略(如何走),但本文首先问的是这个房间是否可达(能不能走到)。具体而言,论文将人与生成模型的对话过程形式化为一个控制过程:用户提示是控制输入,模型生成是系统状态,我们关心的属性(如正式性、物体数量)是测量值。然后,论文使用蒙特卡洛方法估计从给定初始状态能达到的测量值集合(可达集),以及从所有初始状态都能达到的集合(可控集)。技术路线是:首先定义对话过程为离时时间控制系统,然后引入gamma-量化来处理离散瓶颈,最后设计采样算法并提供PAC保证。

本文最核心的创新是将控制理论的可达性和可控性概念首次系统地应用到生成模型的可控性分析中,并针对生成模型的特殊性质(黑盒、高维、离散输出)进行了关键适配。与已有工作的本质区别在于:(1)框架的普适性——将生成模型视为不透明的非线性控制系统,不依赖任何架构假设,适用于任何生成模型;(2)理论的严格性——提供PAC形式的统计保证,而非启发式的方法;(3)处理离散瓶颈——引入gamma-量化可达性和粗粒度可达性概念,解决生成模型输出空间离散导致现有方法失效的问题。特别地,论文证明了定理1(可达集的样本复杂度界)和定理2(可控集的样本复杂度界),这两个定理是分布无关的(distribution-free),只假设输出有界性,适用于任何非线性控制系统。

方法步骤详情

论文方法包含以下关键步骤:(1)定义对话过程:将人与生成模型的交互形式化为离散时间控制系统(phi, T, X, U, h, Y),其中状态空间X是字符串/图像空间,控制输入空间U是提示空间,动力学函数phi定义模型如何根据历史生成下一个状态,读出函数h将状态映射到测量值(如正式性分数、物体数量)。(2)定义可达集和可控集:可达集R(x0, U, t)是从初始状态x0出发在t步内能达到的所有测量值的集合;可控集C_t^alpha是被至少1-alpha比例的初始状态可达的测量值集合。(3)处理离散瓶颈:对于连续测量值,引入gamma-量化可达性R^gamma,允许测量值误差gamma;构建测量空间Y的有限覆盖Y_q,覆盖数N。(4)蒙特卡洛可达集估计(算法1a):从初始状态x0采样m条轨迹,收集测量值,对于离散情况直接返回样本集,对于连续情况返回gamma-覆盖。(5)蒙特卡洛可控集估计(算法1b):采样k个初始状态,分别估计可达集,取交集得到可控集估计。(6)提供PAC保证:定理1给出可达集估计的样本复杂度m >= max(N, log(delta/N)/log(1-p));定理2给出可控集估计的样本复杂度k >= log(epsilon*delta_C)/log(1-alpha)。(7)评估可控性:计算覆盖率cvg = |Y intersect C_hat_t|/|Y|和校准指标(Spearman rho, Pearson R, MAE)。

技术新颖性

本文的技术新颖性体现在多个层面:(1)概念层面——首次将控制理论的可达性和可控性概念系统地应用到生成模型分析中,建立了严格的形式化框架;(2)理论层面——推导了两个新的PAC定理(Theorem 1和2),这些定理是分布无关的,只假设输出有界性,适用于任何非线性控制系统,包括生成模型;(3)算法层面——设计了专门处理生成模型特性的蒙特卡洛算法,特别是通过gamma-量化和粗粒度可达性解决离散瓶颈问题;(4)方法论层面——提出了先验证可控性再尝试控制的范式,而非传统的直接设计控制方法。与已有控制理论工作的具体区别包括:现有方法假设系统动态满足Lipschitz连续性等条件,但这些对不透明的LLM和T2IM无法验证;现有方法返回连续可达集估计,但对LLM的离散输出会返回过大的估计;现有方法关注状态可达性,而本文关注输出可达性(更接近实际应用需求)。

对话过程示意图
Figure 1: 对话过程示意图

实验结果

论文的实验结果揭示了生成模型可控性的几个重要发现,这些发现挑战了领域内的普遍假设。在文本正式性控制任务中,0-shot设置下三个模型(SmolLM3-3B, Qwen3-4B, Gemma3-4B)在5轮对话内都无法达到完全可控,但在5-shot设置下,Qwen3-4B和Gemma3-4B在t=5时达到完全可控(cvg=1.0),而SmolLM3-3B仍然不可控。这表明可控性高度依赖于模型和提示策略的组合。校准指标显示,即使模型可控,其对用户请求的忠实度也有限:Qwen3-4B表现最好,中位MAE=0.09;Gemma3-4B的MAE更高。模型规模与可控性正相关:在Qwen系列(0.6B到14B)上,可控性和校准度随规模增加而提升,但校准指标在4B参数后饱和。即使14B模型的MAE也约为0.25,而随机生成的期望MAE=1/3约0.33,改善有限。在T2IM任务中,物体数量控制比预期更困难:FLUX-s表现最好,MAE=3.52,rho,R>0.9,但不同物体类别间方差很大。物体位置控制对所有T2IM都很困难,模型倾向于将物体放在图像中心。图像饱和度控制对所有T2IM都不可控(rho,R<0.1)。关键发现还包括:反馈和示例的有效性因模型而异——Qwen3-4B和Gemma3-4B从5-shot示例中受益更多,而SmolLM3-3B从多轮反馈中受益更多;对话过程中存在明显的超调现象,即使给出有利的反馈,模型也会在目标值附近震荡;没有单一模型或提示策略能保证跨所有任务的可控性。

文本正式性控制的可控性轨迹
Figure 2: 文本正式性控制的可控性轨迹
模型规模与可控性的关系
Figure 3: 模型规模与可控性的关系
T2IM物体生成任务的可控性
Figure 4: T2IM物体生成任务的可控性
查看结构化数据
任务指标本文基线提升
文本正式性控制 (5-shot, t=5) 可控集覆盖率 cvg Qwen3-4B: 1.0, Gemma3-4B: 1.0, SmolLM3-3B: <1.0 N/A 首次形式化测量
文本正式性控制 (5-shot, t=5) 中位MAE Qwen3-4B: 0.09 随机生成: 0.33 73%降低
物体数量控制 (FLUX-s) MAE 3.52 N/A 首次形式化测量
物体数量控制 (FLUX-s) Spearman rho >0.9 N/A 首次形式化测量
模型规模vs可控性 (Qwen, 14B) 形式性cvg 接近1.0 0.6B: 接近0 随规模显著提升
图像饱和度控制 可控性 不可控 (rho,R<0.1) N/A 发现根本限制

局限与改进

论文承认了几个重要局限性:(1)框架的保证只针对给定的控制系统配置(输入分布、读出映射、初始状态分布),这些由使用者定义,保证不能迁移到新的变量选择。这意味着特定任务的结论只对该任务设置有效。(2)由于将模型视为黑盒,框架不能提供解释性工具来诊断可控性失败的模型内部原因。(3)定理1的样本复杂度随覆盖数N增长,当估计内在复杂的高精度可达集时,样本复杂度可能很高。作者建议构建具有可处理基数的量化Y_q(如二元安全/不安全区域,N=2)作为workaround。从我自己的观察来看,还有几个局限:(4)实验任务相对简单(如奇偶数、字符串长度),实际应用中的复杂语义任务可能表现更差;(5)框架假设读出映射h是准确的,但为复杂语义属性(如有用性)设计准确的读出函数本身就是挑战;(6)框架分析的是静态系统,但实际的生成模型会随时间演变(通过微调等),之前的分析可能失效。

独立分析的弱点

从独立分析的角度,我识别出以下几个弱点和改进方向:(1)高维可扩展性问题:当测量空间维度增加时,覆盖数N可能指数增长,导致样本复杂度爆炸。改进方向是利用可达集的内在维度通常低于外在维度的特性,开发基于流形学习的自适应覆盖方法。(2)读出函数依赖:框架假设读出函数h是准确的,但为复杂语义属性设计准确的读出函数本身是开放问题。改进方向是将读出函数的不确定性纳入分析框架,或者开发端到端的可控性估计方法。(3)静态系统假设:框架分析的是固定时间点的可控性,但实际对话是动态的,用户可能根据模型输出调整目标。改进方向是开发自适应时间范围的可控性分析,或者将框架扩展到闭环控制设置。(4)缺乏控制设计:框架只分析可控性,不提供如何达到可控的指导。改进方向是利用可控性分析的结果自动设计控制器,例如在不可控区域引入额外的控制机制。(5)计算效率:蒙特卡洛方法需要大量采样,对于大型模型和复杂任务可能计算成本很高。改进方向是开发方差缩减技术或重要性采样方法。

未来方向

论文作者提出和基于成果可延伸的未来方向包括:(1)安全和合规应用:将框架应用于生成模型安全,包括严格比较不同控制机制、估计对抗输入下的可达集、在训练期间强制可控性、为政策和部署提供依据。(2)连续控制输入:将框架扩展到激活引导等连续控制机制,这将处理无离散瓶颈的情况,可能实现更精细的控制。(3)多轮对话动态:将分析扩展到自适应时间范围的对话,考虑用户根据模型输出调整目标的情况。(4)控制器设计:利用可控性分析的结果自动设计控制器——如果发现系统在某些区域不可控,可以设计专门的控制器来改善。(5)模型解释性整合:将框架与模型解释性工具结合,实现从形式化验证到诊断的完整流程。(6)训练时可控性:将框架应用于训练过程,在训练期间强制可控性,而非仅在推理时分析。(7)多属性联合分析:扩展框架到多属性可控性分析,例如同时控制正式性和情感。

复现评估

论文的复现性良好。作者在GitHub(https://github.com/apple/ml-genctrl)开源了完整的代码库,并提供了Python包。实验基于开源模型(Qwen3系列、Gemma3-4B、SmolLM3-3B、FLUX系列、SDXL、DMD2)和公开数据集(MS-COCO的80个类别)。由于框架是蒙特卡洛方法,计算需求主要取决于样本数量:定理1的样本复杂度m约O(N log N),定理2的样本复杂度k取决于alpha(部分可控性参数)。对于论文中的简单任务(如文本正式性,N约10),分析可以在单GPU上几小时内完成;但对于更复杂的任务(如物体数量,80个类别),可能需要更多计算资源。论文提供了详细的超参数设置(Tab. J.4和J.5)和输入分布(Sec. J.1),便于复现。复现难度主要在于:(1)需要访问多个大型语言模型和T2IM;(2)需要实现或使用论文开源的controllability包;(3)需要理解控制理论的基本概念来正确解释结果。