Robust Predict-Then-Optimize Handles Feature Perturbations

Aakil Caunhye, Xuefei Lu, Belen Martin-Barragan· July 27, 2026 View original

Summary

This paper introduces a robust variant of the smart predict-then-optimize approach, designed to account for prediction shifts caused by noisy or corrupted covariate features. It integrates robust optimization principles into the predictive-prescriptive pipeline, establishing a tractable convex surrogate that hedges against worst-case feature perturbations and consistently outperforms standard methods.

Traditional integrated learning and optimization models, often referred to as "predict-then-optimize," typically assume that side information or features are perfectly accurate. However, in real-world scenarios, empirical data-driven features are frequently noisy, corrupted, or subject to disturbances at the point of decision-making. This can lead to fragile operational policies and suboptimal outcomes. To bridge this critical gap, this research proposes a "smart predict-then-robustly-optimize" framework. This novel approach directly integrates principles of robust optimization into the predictive-prescriptive pipeline. It achieves this through a specialized loss function designed to hedge against worst-case feature perturbations, ensuring that the resulting operational policies are resilient to real-world data imperfections. The paper establishes a computationally tractable convex surrogate for this robust loss, proving its structural validity with an exponentially decaying approximation error probability. Furthermore, it demonstrates that this framework is Fisher consistent with high probability under mild assumptions. Numerical experiments consistently validate that this robust approach yields significant performance improvements over standard predict-then-optimize methods, both in out-of-sample performance and training stability, even when standard methods employ regularized predictions.

Why it matters

Professionals can develop more resilient and effective decision-making systems by accounting for real-world data noise and uncertainty, leading to more robust operational policies and improved business outcomes.

How to implement this in your domain

  1. 1Identify decision-making processes in your organization that rely on predictions from potentially noisy or uncertain data.
  2. 2Review the principles of robust optimization and the proposed "smart predict-then-robustly-optimize" framework.
  3. 3Explore how to model potential feature perturbations or disturbances relevant to your specific domain.
  4. 4Consider piloting this robust optimization approach on a critical operational problem, such as supply chain management or resource allocation.
  5. 5Benchmark the performance and stability of the robust framework against current predict-then-optimize solutions to quantify its benefits.

Who benefits

LogisticsManufacturingFinanceEnergyHealthcare

Key takeaways

  • Traditional predict-then-optimize models are vulnerable to noisy or corrupted input features.
  • The "smart predict-then-robustly-optimize" framework integrates robust optimization to mitigate this risk.
  • It uses a tractable convex surrogate to hedge against worst-case feature perturbations.
  • This approach consistently outperforms standard methods in out-of-sample performance and stability.

Original post by Aakil Caunhye, Xuefei Lu, Belen Martin-Barragan

"arXiv:2607.21773v1 Announce Type: new Abstract: In this paper, we propose and study a robust variant of the smart predict-then-optimize approach that accounts for prediction shifts due to disturbance in the covariate feature space. While traditional integrated-learning-and-optimi…"

View on X

Originally posted by Aakil Caunhye, Xuefei Lu, Belen Martin-Barragan on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses