AITE:英伟达用大模型自主设计无线通信算法,时延降低至1/3.6

Autonomous Discovery of Wireless Communications Algorithms

论文原文 ↗ 论文发布 解读发布 解读:AI前沿分享

AITE:英伟达用大模型自主设计无线通信算法,时延降低至1/3.6 论文图示

传统无线通信物理层算法的设计,长期依赖通信专家推导繁复的数学矩阵并辅以大量仿真微调。面对高移动性或非线性场景,手写算法不仅需要极高的人力成本,而且往往难以在硬件推理延迟与解调准确率之间取得兼顾。

ArXiv URL:https://arxiv.org/abs/2607.17762v1

近期,来自英伟达(NVIDIA)的研究团队提出了名为 AITE(The AI Telco Engineer)的自动化框架。该框架基于大语言模型(LLM)驱动的进化搜索范式,摆脱了单纯依赖人类直觉或神经网络黑盒的传统开发模式。在正交时频空(OTFS)均衡器设计任务中,AITE 自主生成的算法超越了现有最强基准方案,并将计算延迟降低至原本的 $1/3.6$;在更具挑战性的无导频(Pilotless)OFDM 接收机任务中,AITE 首次自主进化出了可解释、显式表达的数学解调算法,其性能完全追平了业界顶级的端到端神经接收机(Neural Receiver)。

这项研究展示了具备工具调用与代码反思能力的 Agentic AI 已经跨过了无线通信物理层算法研究的关键门槛,通信工程的研发范式正在被重构。

从遗传编程到 LLM 进化:AITE 的双层智能架构

利用程序搜索优化算法并非全新概念。早期的遗传编程(Genetic Programming, GP)通过随机变异和交叉重组代码片段来寻找解,但因缺乏对语法与通信逻辑的理解,搜索空间极易发散,无法胜任高复杂度系统的设计。AITE 采用 LLM 充当具有领域常识的“变异算子”与“交叉算子”,把算法发现转化为具有结构感知的编程搜索任务。

针对通信算法兼顾性能与算力的严苛要求,AITE 的目标并非收敛于单个最优点,而是在“性能度量”与“复杂度度量(如硬件时延)”构成的二维平面上推进帕累托前沿(Pareto Front),为工程师提供完整的权衡点云。整个系统由协调器(Orchestrator)与工作池(Worker Pool)构成的双层架构驱动:

  1. 中心化灵感演进(Idea Generation):协调器 LLM 在每轮生成 $N$ 个独立的抽象算法构想,严格按比例分配给全新探索(Exploration)与存量深挖(Exploitation)。为了打破有限上下文窗口的制约,协调器以上一代的帕累托前沿算法为锚点,同时使用基于归一化性能差距的 Softmax 温度采样,从非前沿算法中抽取代表性样本,防止搜索过程陷入局部最优或思路趋同。

  2. 多 Worker 并行验证与反思:每个 Worker 部署在隔离的容器化沙盒内,由独立的 ReAct 代理驱动。多个 Worker 可以分配到相同的构想,利用解码随机性生成多样化的工程实现。Worker 在沙盒中维护 draft.py 与记录最优表现的 solution.py,自主调用仿真与调试工具。

  3. 算法后处理与偏离过滤:针对大模型写代码时容易偏离初始指令(特别是小尺寸开源模型)的痛点,协调器会对 Worker 产出的代码进行语义逆向分析,给出 yes/partial/no 的三元标签。只有严格遵从构想逻辑的代码,才有资格作为后续演进的历史经验,从机制上杜绝了无效变异对进化树的污染。

  4. 元提示词反思进化(Prompt Refinement):协调器在每代结束后汇总各 Worker 的执行日志与工具报错,通过反思总结出具有指导性的防错建议,动态更新下一代 Worker 的 Prompt 模板,形成跨代的自我修正循环。

软硬件协同:Sionna 集成与贝叶斯超参调优

在物理层通信评估中,蒙特卡洛链路仿真通常是效率瓶颈。AITE 将评估工具设计为用户定义的不可变(Immutable)插件,既杜绝了模型自我篡改测试逻辑造成的“刷榜作弊”,又通过超时切断机制保障集群运行通畅。

为了让编写 Python 代码的 Worker 具备标准的物理层开发能力,框架为 Worker 提供了面向开源链路级仿真库 Sionna(基于 PyTorch)的文档检索工具,支持 API 语义搜索与类库定义调取。

同时,为了防止优秀的算法架构因超参数搭配不当而被过早淘汰,Worker 在输出代码中可以声明超参搜索空间。框架通过抽象语法树(AST)解析这些接口,在代码生成后无缝触发基于 Optuna 的多目标贝叶斯优化,对延迟与误块率联合调优,使每个算法构想都能在帕累托前沿上映射出多组最优运行配置。

实战突破:OTFS 均衡器性能时延双超越

AITE 面对的第一个严苛考题,是为新一代高移动性通信核心技术——正交时频空(OTFS)系统设计低复杂度均衡器。在时延-多普勒(Delay-Doppler, DD)域中,信道响应表现出严重的非正交耦合。虽然线性最小均方误差(LMMSE)可以求解,但在 $M=N=64$ 的网格上直接对 $(MN) \times (MN)$ 的矩阵求逆,运算量在实际硬件中难以接受。

通信界过去的主流方案多依赖行稀疏近似展开的消息传递算法,例如期望传播(Expectation Propagation, EP)检测器。AITE 并没有拘泥于微调既有公式,而是借助符号推导,精准发掘了矩形脉冲波形下信道沿多普勒维度的块循环(Block-circulant)特性。

AITE 自主设计的算法利用多普勒离散傅里叶变换(Doppler DFT),成功将规模庞大的耦合矩阵解耦为 $N$ 个独立的 $M \times M$ 小型子块进行并行均衡,并进一步针对 torch.compile 图优化重构了前向计算逻辑。仿真测试显示,在 5G LDPC 编码体系下,AITE 发现的均衡算法在归一化验证误差(NVE)优于业内最佳基准的同时,将硬件计算延迟直接降低为传统最佳方案的 $1/3.6$。

挑战理论盲区:攻克无导频 OFDM 显式接收机

相较于 OTFS 均衡器已有成熟的数学积淀,第二项任务——在完全不发送参考导频(Pilotless)且采用自定义非常规星座图的 OFDM 场景下构建接收机,则属于通信领域的理论“无人区”。

由于缺乏导频辅助,信道估计与符号判决呈现高度交织状态,过去此类极端场景完全由端到端神经网络(Neural Receiver)所垄断。但神经网络模型常因难以解释、算力消耗高、极端条件不可测而受限于实际部署。AITE 介入后,没有采用梯度下降拟合黑盒,而是从第一性原理出发,自主探索出了一套结构可读、参数可调的显式数学算法流程。

这是学术界首次在无导频自定义星座 OFDM 体系下,利用纯显式算法跑通全链路,并在解码误码率等核心指标上完全追平了 SOTA 级神经接收机。这意味着,进化式大模型驱动的算法探索,不仅能承担代码加速和调参的工程职责,更开始具备了填补通信理论结构性空白的潜力。

从英伟达 AITE 的落地表现可以看出,LLM 驱动的进化搜索让算法设计从传统的“人工公式推导”转向了“目标约束引导下的自主发现”。通过将大模型的逻辑推理能力与严格的物理层仿真器闭环结合,通信系统的升级路径已被全面拓宽,这种兼顾算力开销与数学可解释性的新范式,正在为 6G 物理层创新带来全新的想象空间。