AttenMfg: An Attention Network Based Optimization Framework for Sensor-Driven Operations & Maintenance in Manufacturing Systems

Abstract

Operations and maintenance (O&M) scheduling is a critical problem in leased manufacturing systems, with significant implications for operational efficiency, cost optimization, and machine reliability. Solving this problem involves navigating complex trade-offs between machine-level degradation risks, production throughput, and maintenance team logistics across multi-site manufacturing networks. Conventional approaches rely on large-scale Mixed Integer Programming (MIP) models, which, while capable of yielding optimal solutions, suffer from prolonged computational times and scalability limitations. To overcome these challenges, we propose AttenMfg, a novel decision-making framework that leverages multi-head attention (MHA), tailored for complex optimization problems. The proposed framework incorporates several key innovations, including constraint-aware masking procedures and novel reward functions that explicitly embed mathematical programming formulations into the MHA structure. The resulting attention-based model (i) reduces solution times from hours to seconds, (ii) ensures feasibility of the generated schedules under operational and logistical constraints, (iii) achieves solution quality on par with exact MIP formulations, and (iv) demonstrates strong generalizability across diverse problem settings. These results highlight the potential of attention-based learning to revolutionize O&M scheduling in leased manufacturing systems.