WattBot 2025: Estimating AI Emissions with RAG

WattBot was a challenge in the 2025 Machine Learning Marathon (MLM25). Teams built retrieval-augmented generation (RAG) systems to extract credible, citation-backed emissions and cost estimates for AI workloads from a corpus of 35+ peer-reviewed papers and 300+ curated Q&A pairs. Systems were expected to return citation-grounded answers or explicitly abstain when evidence was missing – promoting transparency and reproducibility in sustainability reporting.

Challenge design

• Task: Given a natural-language question about AI energy use, water consumption, or carbon emissions, retrieve relevant passages from the provided corpus and generate a citation-backed answer.
• Evaluation: Answers were scored on factual accuracy, proper citation, and appropriate abstention when evidence was insufficient.
• Corpus: 35+ academic papers covering AI sustainability, energy benchmarking, and environmental impact reporting.

Winning approach

The winning solution by KohakuBlueleaf used a RAG pipeline that was later replicated and deployed in both AWS Bedrock and locally with open-source Hugging Face models. See the follow-up talk below for deployment details.

Links

• Kaggle challenge: WattBot 2025
• Winning solution: KohakuBlueleaf/KohakuRAG
• Deployment repo: WattBot in Bedrock and Local

Related resources

• Talk: Deploying RAG in Bedrock vs. Local: WattBot 2025 Case Study: Follow-up ML+X forum where the winning RAG system was deployed in AWS Bedrock and locally with open-source models.
• Talk: AI’s Environmental Footprint: Insights and Actions: The ML+X forum where WattBot was first introduced alongside LLM energy benchmarking work.
• Notebook: Exploring Fact-Based QA with RAG: Romeo and Juliet: Build an end-to-end RAG pipeline from scratch – a great starting point before tackling WattBot.
• Workshop: Intro to AWS SageMaker for Predictive ML/AI: Covers AWS SageMaker and Bedrock for cloud-based ML/AI workflows, including RAG deployment.
• Notebook: Understanding Quantization and Precision: Learn how quantization (e.g., 4-bit) reduces model size and memory requirements – relevant to the local deployment approach used in this project.

Comments