CATransformers: Carbon Aware Transformers Through Joint Model-Hardware Optimization
Abstract
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}}}}.
Cite
@article{arxiv.2505.01386,
title = {CATransformers: Carbon Aware Transformers Through Joint Model-Hardware Optimization},
author = {Irene Wang and Newsha Ardalani and Mostafa Elhoushi and Daniel Jiang and Samuel Hsia and Ekin Sumbul and Divya Mahajan and Carole-Jean Wu and Bilge Acun},
journal= {arXiv preprint arXiv:2505.01386},
year = {2025}
}