arXiv:2505.01386cs.LGcs.AR2025-05NeurIPS被引 7

首个兼顾模型与硬件的低碳优化框架,显著降低Transformer碳排放。

CATransformers: Carbon Aware Transformers Through Joint Model-Hardware Optimization

  • 联合优化模型架构与硬件,同时考虑运行与制造碳排放。
  • 在保持精度和延迟前提下,碳排放最高减少30%。
  • 适合关注AI可持续性与绿色计算的研究者与工程师。

机器学习应用迅速普及,从对话式AI到科学发现,但其生命周期碳足迹持续增长,包括训练与推理的运行碳排放,以及硬件制造的隐含碳排放。本文提出 ourframework——首个针对Transformer模型与硬件加速器的碳感知协同优化框架。通过将运行碳与隐含碳纳入早期设计空间探索,该框架实现了以可持续性为导向的模型架构与硬件协同设计,揭示了与仅关注延迟或能耗方法截然不同的权衡关系。在多种Transformer模型上评估表明, ourframework 能持续降低总碳排放,最多达30%,同时维持准确率与延迟性能。进一步通过多模态模型案例展示了其可扩展性。结果强调了需采用整体优化方法,在不牺牲模型能力与执行效率的前提下优先考虑碳效率。源代码已开源: https://github.com/facebookresearch/CATransformers。

原文摘要 · Abstract (English)

Machine learning solutions are rapidly adopted to enable a variety of key use cases, from conversational AI assistants to scientific discovery. This growing adoption is expected to increase the associated lifecycle carbon footprint, including both \emph{operational carbon} from training and inference and \emph{embodied carbon} from AI hardware manufacturing. We introduce \ourframework -- the first carbon-aware co-optimization framework for Transformer-based models and hardware accelerators. By integrating both operational and embodied carbon into early-stage design space exploration, \ourframework enables sustainability-driven model architecture and hardware accelerator co-design that reveals fundamentally different trade-offs than latency- or energy-centric approaches. Evaluated across a range of Transformer models, \ourframework consistently demonstrates the potential to reduce total carbon emissions -- by up to 30\% -- while maintaining accuracy and latency. We further highlight its extensibility through a focused case study on multi-modal models. Our results emphasize the need for holistic optimization methods that prioritize carbon efficiency without compromising model capability and execution time performance. The source code of \ourframework is available at {\small{\href{https://github.com/facebookresearch/CATransformers}{\texttt{https://github.com/facebookresearch/CATransformers}}}}.

低碳计算协同优化Transformer可持续性

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