Dual Interior-Point Optimization Learning

2024-02-04

arXiv.org

AI Summary

To address the slow convergence of conventional solvers for large-scale constrained optimization and their inability to meet real-time requirements, this paper proposes a novel machine learning surrogate method that learns dual feasible solutions—thereby overcoming the lack of theoretical guarantees inherent in existing primal-only surrogates. Our approach tightly integrates linear programming structure, convex optimization theory, and dual modeling. Key contributions include: (1) the first smooth, self-supervised dual loss function enabling end-to-end differentiable training; and (2) an implicit-layer-free analytical dual completion strategy that rigorously ensures dual feasibility and enables millisecond-scale inference. Evaluated on large-scale linear optimization benchmarks, our method achieves <1% optimality gap while accelerating solution times by several orders of magnitude over commercial solvers, significantly outperforming unstructured surrogate models.

Technology Category

Machine Learning: OptimizationSearch and Optimization: Non-convex OptimizationConstraint Satisfaction and Optimization: Constraint Optimization

Application Category

Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for rankingSecurity and Privacy: Security and privacy of machine learning and AI applicationsGraph Algorithms and Modeling for the Web: Graph neural networks and deep learning approaches for Web-related graphs

Abstract

In many practical applications of constrained optimization, scale and solving time limits make traditional optimization solvers prohibitively slow. Thus, the research question of how to design optimization proxies -- machine learning models that produce high-quality solutions -- has recently received significant attention. Orthogonal to this research thread which focuses on learning primal solutions, this paper studies how to learn dual feasible solutions that complement primal approaches and provide quality guarantees. The paper makes two distinct contributions. First, to train dual linear optimization proxies, the paper proposes a smoothed self-supervised loss function that augments the objective function with a dual penalty term. Second, the paper proposes a novel dual completion strategy that guarantees dual feasibility by solving a convex optimization problem. Moreover, the paper derives closed-form solutions to this completion optimization for several classes of dual penalties, eliminating the need for computationally-heavy implicit layers. Numerical results are presented on large linear optimization problems and demonstrate the effectiveness of the proposed approach. The proposed dual completion outperforms methods for learning optimization proxies which do not exploit the structure of the dual problem. Compared to commercial optimization solvers, the learned dual proxies achieve optimality gaps below $1%$ and several orders of magnitude speedups.

Problem

• Designing dual feasible optimization proxies
• Learning dual solutions with quality guarantees
• Improving speed and optimality in constrained optimization

Innovation

• Smoothed self-supervised loss function
• Novel dual completion strategy
• Closed-form solutions for dual penalties