arXiv:2507.18989cs.LGcs.AI2025-07中稿 · the 2026 31st Asia…

用AI自动设计低功耗乘法器,比传统方法更省电18%。

GENIAL: Generative Design Space Exploration via Network Inversion for Low Power Algorithmic Logic Units

  • 基于Transformer的模型预测电路功耗,反向搜索最优编码
  • 在典型AI任务中使开关活动降低18%,收敛速度更快
  • 适合芯片设计、低功耗系统开发人员快速优化逻辑电路

随着人工智能算力需求增长,优化算术单元对缩小数字系统面积至关重要。传统设计流程依赖人工或启发式方法,难以充分探索庞大设计空间。本文提出GENIAL,一种基于机器学习的自动算术单元生成与优化框架,聚焦乘法器设计。核心是一个分两阶段训练的Transformer代理模型:先自监督预训练,再监督微调,可从抽象设计表示中稳健预测功耗、面积等关键硬件指标。通过反向求解该模型,GENIAL高效搜索出能直接降低特定输入数据分布下功耗的新型操作数编码。大规模实验表明,GENIAL在样本效率和收敛速度上均优于其他方法,支持高复杂度逻辑综合流程闭环部署,提升代理模型精度。显著成果包括:在代表性AI工作负载下,乘法器开关活动降低最多达18%,优于传统补码表示。此外,该方法在有限状态机上也取得显著改进,证明其适用于多种逻辑函数。这些进展推动了面向高质量结果的组合电路自动化生成。

原文摘要 · Abstract (English)

As AI workloads proliferate, optimizing arithmetic units is becoming increasingly important for reducing the footprint of digital systems. Conventional design flows, which often rely on manual or heuristic-based optimization, are limited in their ability to thoroughly explore the vast design space. In this paper, we introduce GENIAL, a machine learning-based framework for the automatic generation and optimization of arithmetic units, with a focus on multipliers. At the core of GENIAL is a Transformer-based surrogate model trained in two stages, involving self-supervised pretraining followed by supervised finetuning, to robustly forecast key hardware metrics such as power and area from abstracted design representations. By inverting the surrogate model, GENIAL efficiently searches for new operand encodings that directly minimize power consumption in arithmetic units for specific input data distributions. Extensive experiments on large datasets demonstrate that GENIAL is consistently more sample efficient than other methods, and converges faster towards optimized designs. This enables deployment of a high-effort logic synthesis optimization flow in the loop, improving the accuracy of the surrogate model. Notably, GENIAL automatically discovers encodings that achieve up to 18% switching activity savings within multipliers on representative AI workloads compared with the conventional two's complement. We also demonstrate the versatility of our approach by achieving significant improvements on Finite State Machines, highlighting GENIAL's applicability for a wide spectrum of logic functions. Together, these advances mark a significant step toward automated Quality-of-Results-optimized combinational circuit generation for digital systems.

低功耗设计AI芯片神经网络电路优化

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