想象一下,你是一家大型高科技工厂的负责人。这家工厂拥有数千个传感器,用于测量温度、压力、速度和电压等一切数据。但存在一个问题:一些最重要的传感器要么极其昂贵,要么极易损坏,要么根本尚未问世。
传统上,为了解决这个问题,工程师们会为每一个缺失的传感器构建一个独立的、定制的机器人。如果你需要预测电池的温度,你就构建机器人 A;如果你需要预测车轮的速度,你就构建机器人 B。
- 问题所在:这就像为菜单上的每一道菜都聘请一位不同的厨师。这种方式既昂贵又缓慢,而且厨师们彼此之间并不交流。如果机器人 A 学到了关于热量的知识,机器人 B 却一无所知。此外,你还需要一位人类专家来明确告诉每个机器人应该监听哪些其他传感器,而要为成千上万个机器人准确完成这项任务,难度极大。
新方案:“超级大脑”厨师
本文的作者提出了一种不同的方法:一位“超级大脑”厨师,能够瞬间烹制菜单上的任何一道菜。
他们构建了一个统一的单一 AI 模型,能够同时预测数百种不同的“虚拟传感器”(即缺失的测量值)。以下是他们如何利用简单的类比来实现这一点的:
1. “空盘子”技巧(选择性预测)
通常,AI 模型就像只能复述刚听到的内容的学生。如果你要求它们预测一个从未见过的信号,它们就会感到困惑。
- 创新点:作者赋予了模型一个特殊的技巧。当你要求它预测特定传感器(例如“电池温度”)时,它会先拿起一个空盘子(零向量)。然后,它会在盘子上贴上特殊标签,注明:“这是用于电池温度的”。
- 工作原理:模型会查看所有可用数据(即食材),并找出哪些数据与填充该特定盘子相关。它学会忽略噪声,只关注那些实际上有助于预测该特定温度的信号。这就像一位厨师,当被要求做汤时,会瞬间忽略用于牛排的香料,只抓取正确的蔬菜。
2. 学会监听对象(无需专家)
在旧方法中,人类专家必须坐下来指定:“机器人 A,请监听传感器 X 和传感器 Y。”
- 创新点:这个新模型能够自行学习应该监听谁。它内置了一种“注意力机制”,就像聚光灯一样。在训练过程中,它会确定哪些输入信号对哪些输出至关重要。
- 结果:它为每个虚拟传感器创建了一个独特的“监听列表”。对于电池温度,它可能会监听冷却液流量;对于发动机扭矩,它会监听踏板压力。这一切都是自动完成的,无需人类专家手动挑选输入数据。
3. “智能过滤器”(效率)
你可能会想:“如果一个模型包办所有事情,那它一定非常庞大且缓慢。”
- 创新点:得益于其“智能过滤器”,该模型极其高效。当你要求它仅预测一个传感器时,它不会查看所有工厂数据。它会利用其已学会的“监听列表”来忽略 99% 的输入,仅处理那些至关重要的少数数据。
- 类比:想象一位图书管理员需要找到一本特定的书。与其走遍整个图书馆,他们确切地知道该去哪个书架、哪一排。他们会跳过其余部分。这使得该模型比运行数百个独立模型快 415 倍,且内存占用少 951 倍。
成果:自动运行的工厂
该团队在真实世界数据上测试了这个“超级大脑”,其中包括来自电动汽车的超大规模数据集(包含超过 180 亿个数据点)以及标准的交通和电力数据集。
- 可扩展性:他们证明,增加更多虚拟传感器(从 1 个增加到 512 个)并不会使模型变得更大或更慢。这就像在菜单上增加更多菜肴,而无需聘请更多厨师;只需让那位厨师更擅长多任务处理即可。
- 质量:预测结果与旧方法(为每个传感器使用独立模型)一样好,甚至更好。
- 可解释性:由于模型自行学习其“监听列表”,我们可以观察它并指出:“啊,为了预测电池温度,它正在监听冷却液流量。”这使得 AI 透明且可信,这对于汽车等安全关键领域至关重要。
总结
本文没有为每个缺失的传感器构建独立、昂贵且孤立的机器人,而是引入了一个灵活、高效的 AI,能够同时预测数百个缺失信号。它学会忽略哪些数据、关注哪些数据,从而节省了巨大的计算资源,同时使系统更易于理解和扩展。它将一个由成千上万个独立机器人组成的混乱工厂,转变为一个单一、高度组织化且超高速的团队。
技术摘要:面向虚拟传感器的可扩展多任务模型
问题陈述
虚拟传感器利用机器学习从可用测量值中预测目标信号,为自动驾驶和医疗保健等关键应用中昂贵的物理传感器提供了一种具有成本效益且可靠的替代方案。然而,现有方法存在四个主要局限性:
- 缺乏可扩展性:当前方法依赖于孤立的、特定于应用模型,这些模型为每个虚拟传感器单独训练和维护,输入信号由领域专家手动选择。这阻碍了向大规模传感器网络的扩展。
- 错失协同效应:孤立模型无法利用不同虚拟传感器任务之间的任务协同效应。
- 基准不一致:缺乏一致的评估标准,很少有工作利用标准基准数据集。
- 基础模型的局限性:虽然时间序列基础模型提供了通用解决方案,但它们计算成本高昂,且仅限于预测输入信号。它们无法从其他测量值中预测新的、未见过的信号,因此与虚拟传感器任务不兼容。
方法论
作者提出了一种基于因果解码器-only Transformer 架构的统一、可扩展多任务模型。核心创新在于三种旨在利用任务协同效应同时保持计算效率的机制:
1. 虚拟传感器的选择性预测
为了预测虚拟传感器(输入集中不存在的信号),模型采用基于原型令牌(prototype tokens)的自回归机制。
- 令牌化:输入时间序列被归一化、分块并嵌入为令牌。
- 原型插入:为了预测特定的虚拟传感器 j,将一个零向量原型令牌插入输入序列中。
- 变量嵌入:一个可学习的变量嵌入向量 vj′ 被添加到原型令牌中。这引导神经网络将相关的目标信息映射到空的原型中,使模型能够基于可用的输入信号预测特定的虚拟传感器。
- 自回归:模型预测接下来的 p 个时间步。在后续步骤中,模型自身的先前预测作为虚拟传感器的新输入,模拟现实世界的部署场景,即物理输入可用,但必须预测虚拟输出。
2. 学习相关输入信号
模型不再依赖专家选择的输入,而是端到端地学习每个虚拟传感器的一组相关输入信号。
- 信号相关性向量:模型为每个虚拟传感器引入了可训练的信号相关性向量 R′={rj′}。每个向量编码了所有可用传感器(输入和其他虚拟传感器)的相关性。
- 注意力偏置:相关性向量用于通过外积计算交叉相关性矩阵,该矩阵在时间上复制并作为静态偏置应用于每个 Transformer 层的注意力分数计算中。这使得模型能够学习全局的、与时间无关的信号重要性。
- 可解释性:通过分析这些向量,模型提供了关于哪些输入信号驱动特定虚拟传感器预测的可解释性。
3. 高效稀疏化与选择
为了解决大规模传感器网络的计算约束,模型采用了结构化稀疏性:
- 阈值处理:在训练后期和推理阶段,信号相关性向量基于阈值 rthres′ 进行稀疏化。不重要的信号被掩码为 −∞,从而有效地将它们从注意力机制中移除。
- 动态传感器选择:模型可以在单次生成步骤中灵活选择要预测哪些虚拟传感器。当预测传感器子集时,模型仅查询这些特定传感器的相关输入信号集,从而最小化处理的令牌数量。
- 训练策略:为了避免所有传感器产生相同的梯度(这将导致相同的输入集),模型在每次迭代中针对随机子集的虚拟传感器(Ntrain)进行训练。此外,训练期间使用教师强制(Teacher Forcing)以减少内存和计算成本,并探索了混合方法(使用随时间反向传播进行微调)以减轻误差累积。
主要贡献
- 大规模基准:作者首次在三个标准时间序列预测基准(交通、电力、太阳能)和一个包含超过 180 亿个时间序列样本的特定于应用的汽车 CAN 总线数据集上,对统一虚拟传感器模型进行了大规模分析。
- 统一多任务架构:所提出的模型同时预测多种虚拟传感器,利用任务协同效应和基础时间序列知识。它能够优雅地扩展到数百个虚拟传感器,且参数量几乎保持恒定。
- 学习的相关性与可解释性:模型通过端到端学习相关输入信号,消除了对专家知识进行输入选择的依赖,从而为虚拟传感器设计增加了可解释性。
- 效率机制:通过选择性预测、学习的相关性掩码和稀疏化,该架构显著减少了计算时间和内存使用。
结果
评估表明,在保持或提高预测质量的同时,实现了显著的效率提升:
- 性能与孤立模型对比:与为单个传感器训练的孤立模型相比,统一方法在相似的推理速度下生成了更优的预测。
- 效率提升:在 CAN 总线数据集上,与未利用稀疏化或选择性预测的标准多任务方法相比,统一模型将计算时间减少了高达 415 倍,将内存需求减少了 951 倍。
- 可扩展性:当从少量扩展到数百个虚拟传感器时,模型保持了稳定的预测质量(MSE 的低标准差),而孤立模型则需要线性增加参数量。
- 可解释性:对 CAN 总线数据集的定性评估证实,学习到的输入信号集与领域专家知识一致(例如,温度传感器依赖于其他温度和流速信号)。
- 迁移学习:模型可以仅使用 10% 的训练数据,将新的虚拟传感器高效地引入到已训练的模型中,其性能与从头训练相当。
意义与主张
本文声称提出了第一个统一的虚拟传感器模型,成功 bridged 了特定于应用的虚拟传感器与时间序列基础模型之间的差距。其意义在于:
- 实际部署:通过大幅降低计算和内存需求,该模型使得虚拟传感器能够在边缘设备和大规模传感器网络中部署,而在这些场景中,当前的基础模型因资源密集型而难以应用。
- 消除专家瓶颈:通过自动学习输入相关性,该方法消除了对领域专家进行特征选择的依赖,使虚拟传感器设计更具可扩展性和鲁棒性。
- 协同效应利用:它证明了虚拟传感器任务可以从共享知识中受益,同时提高预测质量和效率。
作者得出结论,这种多任务模型为虚拟传感器开启了新的应用,并通过减少大规模传感器网络的计算足迹,促进了更环保的时间序列建模。他们承认由于训练大规模模型成本高昂,在超参数搜索方面存在局限性,并建议未来有进一步优化的空间。
每周获取最佳 machine learning 论文。
受到斯坦福、剑桥和法国科学院研究人员的信赖。
请查收邮箱确认订阅。
出了点问题,再试一次?
无垃圾邮件,随时退订。