mlsanityData to 14 Oct 2026

Decision-Focused Learning Enhanced by Automated Feature Engineering for Energy Storage Optimisation

arXiv 2509.05772 ↗First posted 6 Sep 2025cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Decision-making under uncertainty in energy management is complicated by unknown parameters hindering optimal strategies, particularly in Battery Energy Storage System (BESS) operations. Predict-Then-Optimise (PTO) approaches treat forecasting and optimisation as separate processes, allowing prediction errors to cascade into suboptimal decisions as models minimise forecasting errors rather than optimising downstream tasks. The emerging Decision-Focused Learning (DFL) methods overcome this limitation by integrating prediction and optimisation; however, they are relatively new and have been tested primarily on synthetic datasets or small-scale problems, with limited evidence of their practical viability. Real-world BESS applications present additional challenges, including greater variability and data scarcity due to collection constraints and operational limitations. Because of these challenges, this work leverages Automated Feature Engineering (AFE) to extract richer representations and improve the nascent approach of DFL. We propose an AFE-DFL framework suitable for small datasets that forecasts electricity prices and demand while optimising BESS operations to minimise costs. We validate its effectiveness on a novel real-world UK property dataset. The evaluation compares DFL methods against PTO, with and without AFE. The results show that, on average, DFL yields lower operating costs than PTO and adding AFE further improves the performance of DFL methods by 22.9-56.5% compared to the same models without AFE. These findings provide empirical evidence for DFL's practical viability in real-world settings, indicating that domain-specific AFE enhances DFL and reduces reliance on domain expertise for BESS optimisation, yielding economic benefits with broader implications for energy management systems facing similar challenges.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 57 references that are in the map. First posted on arXiv 6 Sep 2025.

Builds on

22 of 57 references.

References · 22

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  2. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  3. 2017Smart "Predict, then Optimize"cited by 339
  4. 2018Melding the Data-Decisions Pipeline: Decision-Focused Learning for Combinatorial Optimizationcited by 227
  5. 2019Differentiation of Blackbox Combinatorial Solverscited by 226
  6. 2021End-to-End Constrained Optimization Learning: A Surveycited by 157
  7. 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 130
  8. 2023Large Language Models for Automated Data Science: Introducing CAAFE for Context-Aware Automated Feature Engineeringcited by 127
  9. 2019Smart Predict-and-Optimize for Hard Combinatorial Optimization Problemscited by 89
  10. 2020Interior Point Solving for LP-based prediction+optimisationcited by 83
  11. 2021Decision-Focused Learning: Through the Lens of Learning to Rankcited by 62
  12. 2020Contrastive Losses and Solution Caching for Predict-and-Optimizecited by 52
10 more
  1. 2022PyEPO: A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programmingcited by 32
  2. 2023Landscape Surrogate: Learning Decision Losses for Mathematical Optimization Under Partial Informationcited by 18
  3. 2024Decision Focused Causal Learning for Direct Counterfactual Marketing Optimizationcited by 9
  4. 2023Benchmarking PtO and PnO Methods in the Predictive Combinatorial Optimization Regimecited by 8
  5. 2023Electricity Price Prediction for Energy Storage System Arbitrage: A Decision-focused Approachcited by 7
  6. 2020Transfer Learning for Electricity Price Forecastingcited by 5
  7. 2020Effective End-to-End Learning Framework for Economic Dispatchcited by 3
  8. 2022Predict+Optimize Problem in Renewable Energy Schedulingcited by 3
  9. 2021Application-Driven Learning: A Closed-Loop Prediction and Optimization Approach Applied to Dynamic Reserves and Demand Forecastingcited by 2
  10. 2022A Contextual Bandit Approach for Value-oriented Prediction Interval Forecastingcited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.