Prediction-Specific Design of Learning-Augmented Algorithms
Sizhe Li, Nicolas Christianson, Tongxin Li
TL;DR
The paper addresses designing online algorithms that leverage ML predictions in a problem-structure-aware, prediction-specific way. It introduces prediction-specific consistency and robustness, and defines strong optimality as Pareto-optimal performance for every prediction value. A bi-level optimization-based meta-algorithm is developed to systematically construct strongly-optimal algorithms, and explicit instances are provided for deterministic and randomized ski rental as well as one-max search. The framework is extended to handle prediction errors via ε-consistency and to support practical cases with empirical validation on dynamic power management and volatility-based trading. Overall, the results show significant performance gains over prior worst-case designs, demonstrating practical impact across online decision-making tasks.
Abstract
Algorithms with predictions} has emerged as a powerful framework to combine the robustness of traditional online algorithms with the data-driven performance benefits of machine-learned (ML) predictions. However, most existing approaches in this paradigm are overly conservative, {as they do not leverage problem structure to optimize performance in a prediction-specific manner}. In this paper, we show that such prediction-specific performance criteria can enable significant performance improvements over the coarser notions of consistency and robustness considered in prior work. Specifically, we propose a notion of \emph{strongly-optimal} algorithms with predictions, which obtain Pareto optimality not just in the worst-case tradeoff between robustness and consistency, but also in the prediction-specific tradeoff between these metrics. We develop a general bi-level optimization framework that enables systematically designing strongly-optimal algorithms in a wide variety of problem settings, and we propose explicit strongly-optimal algorithms for several classic online problems: deterministic and randomized ski rental, and one-max search. Our analysis reveals new structural insights into how predictions can be optimally integrated into online algorithms by leveraging a prediction-specific design. To validate the benefits of our proposed framework, we empirically evaluate our algorithms in case studies on problems including dynamic power management and volatility-based index trading. Our results demonstrate that prediction-specific, strongly-optimal algorithms can significantly improve performance across a variety of online decision-making settings.
