想象一下,你正在教导一位极具天赋、富有创造力的艺术家(一个神经网络)作画。这位艺术家擅长模仿你要求的任何风格或形状。然而,你有一条严格的规定:画作绝不能超出某个特定的、蜿蜒曲折的边界。
在现实世界中,这个“边界”代表安全规则。对于自动驾驶汽车,边界是“保持在道路上”;对于机器人,则是“不要撞墙”。
旧方法的问题
传统上,教导艺术家保持在边界内有两种方法:
- “斥责”法(软约束): 你告诉艺术家:“如果你越线,我就给你打低分。”艺术家会努力保持在界限内,但由于他们只是为了避免低分,因此仍可能意外越线,尤其是在规则变得棘手或画作变得复杂时。这无法保证他们始终安全。
- “编辑”法(后处理): 你让艺术家随意作画,然后对完成的画作进行裁剪,切掉超出边界的部分。问题在于,这往往会破坏画作。画作可能看起来怪异,或者艺术家可能学会绘制一种“安全”版本,但这实际上并非最佳解决方案。
新解决方案:CAffNet
本文作者杨昭及其同事构建了一个名为CAffNet的新框架。可以将 CAffNet 视为一位在物理上无法画出界线的艺术家,而不是一位试图遵守规则的艺术家。
其工作原理如下,使用一个简单的类比:
1. “魔法蹦床”(CAffine 层)
想象艺术家站在一个蹦床上。蹦床的形状完全符合你的安全边界。无论艺术家如何用力跳跃或试图落在何处,蹦床的物理特性都会迫使他们只能落在安全区域内。
- 从技术术语来说,CAffNet 在神经网络中添加了一个特殊的“层”。该层从数学上保证输出(即画作)始终被投影回安全区域,无论网络试图做什么。
2. 处理“纠缠”的规则(约束分解)
有时,安全规则非常复杂。也许你有 100 条不同的线来定义边界,其中一些相互重叠或以令人困惑的方式相互矛盾。如果规则过于混乱(就像试图用缺失的拼图块来解谜),旧方法就会陷入停滞或失败。
- CAffNet 的诀窍: 它将庞大而混乱的拼图分解为微小、可管理的碎片。它一次查看一小组规则,解决它们,然后将答案组合起来。这使得它能够处理任意数量的规则,即使这些规则是冗余或令人困惑的。
3. “智能向导”(可训练零空间)
有时,并非只有一个唯一的“安全落点”,而可能是一整个安全的区域(比如一条长长的走廊)。
- 旧方法可能只是随机选择走廊里的一个点。
- CAffNet 拥有一个“智能向导”(一个可训练的组件),它能学习在安全走廊中哪一个点最适合特定任务。它不仅迫使艺术家保持安全,还能帮助他们找到通往目标的最佳安全路径。
为何这很重要(根据论文)
论文通过三种方式测试了这个新系统:
- 绘制复杂形状: 它学会了绘制复杂的曲线,这些曲线必须保持在特定且不断变化的边界内。它完美地做到了这一点,而其他方法则犯了错误。
- 解决数学谜题: 它充当了一个超高速计算器,在严格遵守规则的同时解决优化问题(寻找最佳解决方案)。它比传统的数学求解器快得多,且从未违反规则。
- 驾驶机器人: 他们教导一个机器人(单轮车模型)在有障碍物的房间中导航。
- 结果: 使用 CAffNet 的机器人成功到达目标,未撞墙。
- 对比: 使用旧“斥责”法或“编辑”法的机器人撞上了障碍物,因为它们无法保证自身安全。
核心结论
CAffNet 是一种构建 AI 的方法,使其无法违反规则,并非因为它害怕惹麻烦,而是因为其结构本身使其不可能这样做。它在保留 AI 学习复杂任务的能力(其“通用近似”能力)的同时,增加了一个 100% 可靠的数学安全网,前提是安全规则本身是合理的。
重要提示: 论文强调,这仅保证 AI 遵守你给出的特定规则。如果你忘记告诉 AI“不要驶下悬崖”,AI 就会愉快地驶下悬崖,因为该规则并未包含在其数学安全网中。该系统保证遵守定义的约束,而非安全计划的完整性。
技术摘要:CAffNet:硬约束仿射神经网络
问题陈述
神经网络(NN)在近似复杂非线性函数方面具有卓越的表达能力,使其在自动驾驶、机器人和航空航天等安全关键领域极具价值。然而,确保学习到的控制器提供形式化的硬约束满足仍然是一个重大挑战。
现有方法存在特定的局限性:
- 软约束:基于惩罚的方法鼓励安全输出,但无法保证满足约束,特别是在训练数据有限或分布外边缘案例中。
- 后处理:在训练后强制执行约束可能导致次优解。
- 与输入无关的硬约束:C-DGMs 或 RAYEN 等方法处理一般约束,但假设可行集与输入无关。
- 与输入相关的硬约束:近期工作(如 POLICE、KKT-hPINN、HardNet-Aff)解决了与输入相关的约束问题,但面临关键限制。具体而言,HardNet-Aff假设约束矩阵 A(x) 具有满行秩。这将约束数量限制为输出维度,并在约束线性相关时失效,可能导致输出投影到可行区域之外。此外,当约束矩阵不是满列秩时,这些方法通常难以处理非唯一的投影。
方法论:CAffNet 框架
作者提出了CAffNet(硬约束仿射神经网络),这是一个闭式框架,将可训练的**约束仿射(CAffine)**层嵌入到标准前馈或 Transformer 架构中。该框架强制执行与输入相关的仿射不等式约束,形式如下:
A(x)y≤b(x)
其中 x 是输入,y 是输出,且对约束数量(m)或 A(x) 的秩没有限制。
核心组件
约束分解:
为了处理任意数量的约束,该方法将约束集分解为较小的子集(子约束)。受多面体几何的启发,作者观察到可行区域由最小面定义,这些面最多需要 nout 个线性无关的约束。该方法考虑 k 个约束的组合(其中 1≤k≤min(m,nout))以形成子矩阵 Aγ 和子向量 bγ。
带零空间的训练投影:
对于每个子约束 γ,框架使用闭式表达式计算投影 Pγ(x):
Pγ(x)=fθ(x)−Aγ†(x)(Aγ(x)fθ(x)−bγ(x))+(I−Aγ†(x)Aγ(x))wϕ(x)
- fθ(x):无约束神经网络的输出。
- Aγ†(x):子约束矩阵的 Moore-Penrose 伪逆。
- wϕ(x):由神经网络参数化的可训练零空间分量。
项 (I−Aγ†Aγ)wϕ(x) 至关重要。当约束线性相关(导致解空间不唯一)时,该项允许网络在零空间内学习一个最优解(例如,在可行边或面上选择一个特定点),而不是仅依赖正交投影。
可行候选选择:
最终输出 P∗(x) 从所有有效投影集合 SP(x)(即满足 A(x)y≤b(x) 的投影)中选出,这些投影最小化与无约束输出 fθ(x) 的距离。如果无约束输出本身已是可行的,则直接使用它。
主要贡献
- 闭式硬约束强制执行:CAffNet 保证任意数量的与输入相关的仿射不等式约束的硬满足,无需约束矩阵 A(x) 具有满行秩。
- 通过分解实现的鲁棒投影:通过引入约束分解和可训练零空间模块,该方法处理线性相关约束和非唯一投影,克服了以往基于投影的方法(如 HardNet-Aff)的局限性。
- 通用近似性质:作者证明了 CAffNet 保留了底层神经网络(包括前馈网络和 Transformer)的通用近似性质。该框架可以近似可行约束集内的任何连续函数。
- 可扩展性(CAffNet-Lite):提出了一种简化变体 CAffNet-Lite,通过将搜索空间限制为最小面(顶点)而非所有组合来降低计算复杂度,将约束数量上的复杂度从指数级降低到多项式级。
实验结果
该论文在三种场景下验证了 CAffNet,将其与带有软约束的标准 NN 和 HardNet-Aff 进行了比较:
学习具有分段约束的非线性函数:
- CAffNet 变体严格满足所有约束,而 NN 和 HardNet 表现出违规。
- CAffNet-TF(基于 Transformer)实现了最低的均方误差(MSE),与软约束 NN 相比降低了73.33%。
- CAffNet 表现出更快的收敛速度和更低的初始损失。
学习优化求解器:
- 在一个涉及不等式和等式约束的问题中,CAffNet 变体严格满足所有约束(0% 违规),而 NN 和 HardNet 在等式约束上严重失败。
- CAffNet-FF 实现了与基准求解器 IPOPT 相当的目标值,但推理时间显著更快,使其适用于实时应用。
- CAffNet-Lite 实现了与完整版本相同的性能,同时将训练和测试时间减少了约 45%。
学习安全关键系统的控制策略:
- 在带有避障的单轮机器人控制任务中,CAffNet-FF 成功导航至目标,未发生碰撞或约束违规。
- NN 和 HardNet 均因约束违规而与障碍物发生碰撞。
- 关键的是,后验强制执行(仅在训练后应用投影)导致机器人卡住,证明了网络参数与投影层之间联合优化的必要性。
意义与主张
该论文声称,CAffNet 解决了在安全关键系统中部署神经网络的一个根本性开放挑战:同时保证近似精度和可证明的约束满足。
- 形式化保证:与基于惩罚的方法不同,CAffNet 提供数学保证,只要可行集非空,输出将满足所有输入的约束。
- 架构通用性:该框架兼容标准架构(前馈神经网络和 Transformer),并且不限制约束空间相对于输出的维度。
- 最优性:通过联合训练网络和投影层,CAffNet 找到最优可行解,而后处理方法可能导致次优性能或不可行性。
作者总结道,虽然该框架在所述假设下提供了强有力的保证,但它本身并不确保约束的完备性或系统模型的准确性。因此,在部署到现实世界的安全关键应用中,仍然需要进行独立的验证和特定领域的安全分析。未来的工作致力于将该框架扩展到非仿射约束,并在物理机器人平台上验证性能。
每周获取最佳 machine learning 论文。
受到斯坦福、剑桥和法国科学院研究人员的信赖。
请查收邮箱确认订阅。
出了点问题,再试一次?
无垃圾邮件,随时退订。