arXiv:2606.04850cs.LGcs.AI2026-06

提出可量化不确定性的端到端神经网络芯片协同设计框架。

Uncertainty-Aware End-to-End Co-Design of Neural Network Processors: From Training and Mapping to Fabrication

论文配图:Uncertainty-Aware End-to-End Co-Design of Neural Network Processors: From Training and Mapping to Fabrication
图 1 · 摘自论文原文
  • 用功能-资源接口解耦设计模块,支持独立优化
  • 引入置信度作为可调资源,实现不确定性建模
  • 适用于需要可靠性和成本平衡的芯片研发场景

神经网络处理器的设计是端到端协同设计问题:网络结构与训练预算决定推理负载;硬件映射影响芯片面积、延迟和功耗;这些特性又决定制造良率与成本。当前方法分阶段进行,且紧耦合于特定算法,修改一个环节需重做全链路。本文基于单调协同设计理论,构建包含网络训练、芯片映射、晶圆级制造与计算资源分配的四模块统一框架。各模块仅通过功能-资源接口交互,可独立优化而无需改变整体结构。核心贡献在于对不确定性的处理:不将随机结果简化为点估计,而是引入置信度(成功概率的倒数)作为可优化的显式资源,与成本、时间、功耗并列。三个案例验证:第一,跨异构场景恢复帕累托最优解;第二,置信度可作为连续调节设计参数而非事后诊断;第三,单个模块改进自动传播至全局帕累托前沿,无需修改协同设计图。

原文摘要 · Abstract (English)

Designing a neural network processor is an end-to-end co-design problem: network architecture and training budget determine the inference workload; hardware mapping decisions determine chip area, latency, and energy; and these characteristics govern fabrication yield and manufacturing cost. In practice, these decisions are made in separate stages, and existing co-design methodologies are tightly coupled to specific algorithms, making it difficult to improve one component without reworking the entire pipeline. This paper presents a unified framework, grounded in monotone co-design theory, that composes four interoperable design blocks spanning network training, chip mapping, wafer-level fabrication, and compute resource allocation. Each block exposes only a functionality-resource interface to the rest of the system, so any block can be refined without structural changes elsewhere. A central contribution is the treatment of uncertainty: rather than collapsing stochastic outcomes into point estimates, the framework introduces Confidence, the inverse of success probability, as an explicit and optimizable resource alongside cost, time, and power. Three case studies validate the approach. The first recovers Pareto-optimal implementations across heterogeneous application scenarios. The second confirms that Confidence functions as a continuously tunable design knob rather than a post-hoc diagnostic. The third demonstrates that improving a single block's implementation set automatically propagates to the global Pareto front, without modifying the co-design diagram.

协同设计不确定性芯片优化置信度

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。