昨晚他几乎没睡,整夜都在思考协同推理的具体实现。早上七点,左城到公司的时候,实验室的灯已经亮了。
“一鸣,你这是通宵了?“左城推门进去,看到沈一鸣趴在桌上睡著了。
沈一鸣揉揉眼睛坐起来:“城哥,我想出来了。“
他从桌上拿起一叠手绘的图纸:“你看,这是我设计的分布式边缘ai架构。“
左城接过图纸,仔细看著上面的线条和標註。
“每颗卫星不再运行一个完整的ai模型,而是运行一个精简版的决策代理。“沈一鸣指著图解释,“代理只负责局部决策,通过星间链路与其他代理通信。“
方泽这时候也走了进来,手里端著三杯咖啡。
“我来补充一下硬体层面的思路。“他把咖啡放下,“如果用协同推理的架构,单颗卫星的算力需求可以大幅降低。我们可以考虑用低功耗的嵌入式ai晶片。“
“功耗能压到多少?“左城问。
方泽调出一些测试数据:“我昨晚查了一下,目前行业里最低功耗的嵌入式ai晶片可以做到2瓦左右。如果用三四颗这样的晶片协同,功耗大概在6到8瓦。“
“还是超標。“沈一鸣皱眉,“目標必须压到5瓦以下。
“与其用多颗低功耗晶片,不如把一个高性能晶片的模型压缩到极致。“左城突然开口。
沈一鸣眼睛一亮:“模型压缩?“
“对。“左城在白板上画了一个示意图,“如果我们能把ai模型的
方泽快速计算著:“寒武纪的u270功耗30瓦,如果压缩到三十分之一,理论上功耗可以降到1瓦左右。“
“问题是,我们现在的压缩技术能做到吗?“沈一鸣问。
左城打开系统面板,查看当前的叶片列表。
【科技树叶片】
- 神经网络基础
- 深度学习框架
- 强化学习策略
- 自然语言理解
- 计算机视觉
- 模型压缩优化
- 联邦学习
- 因果推理
- 生成式模型
- 梯度稀疏化压缩
模型压缩优化。
这枚叶片应该有用。
但他不能直接告诉沈一鸣这些。
“一鸣,你觉得我们的压缩技术有提升空间吗?“左城问。
沈一鸣沉吟片刻:“有。但需要新的算法突破。
“什么样的突破?“
“有一种方向叫知识蒸馏。“沈一鸣在白板上写下这个词,“把大模型的知识蒸馏到小模型里。但知识蒸馏的难点在於,我们没有一个足够强大的教师模型来指导学生模型。“
“为什么?“
“因为天穹卫星的ai调度是一个全新的场景。“沈一鸣解释道,“传统的预训练模型在这个场景上表现不好,我们需要从头训练一个专用的大模型。“
“这需要多少时间?“
沈一鸣估算了一下:“至少两到三个月。“
两到三个月。
而他们只有两周。
“还有一个方向。“方泽突然说。 “边缘计算。“方泽说,“与其在卫星上跑完整的ai模型,不如把一部分计算任务搬到地面站或者中继卫星上。“
“你的意思是?“沈一鸣问。
“云边协同。“方泽在白板上画了一个简单的架构图,“卫星只负责简单的感知和决策,复杂的推理放到地面站或者中继卫星上完成。“
“但这样会增加通信延迟。“沈一鸣提出问题,“对於实时性要求高的场景,延迟可能是致命的。“
“所以要用预测性调度。“左城突然开口。
两人一愣。
“我们的目標不是让卫星实时思考,而是让它提前预判。“左城在白板上画了一条时间线,“比如,星间链路的状態变化是可以预测的。如果卫星能提前几秒钟预判到链路拥塞,就可以提前调整路由,避免问题发生。“
“预测性调度“沈一鸣喃喃自语,“这需要对卫星网络的动態特性有深入理解。“
“我们不是有联邦学习吗?“左城说,“用联邦学习的方式,让所有卫星共享网络状態信息,然后用ai预测链路变化。“
“城哥,你是说,用联邦学习训练一个预测模型?“
“对。“左城点头,“这个模型不需要在卫星上运行,只需要在地面站训练好后,下发预测参数给各卫星。卫星根据参数做简单计算就行了。“
沈一鸣眼睛一亮。
“