EAAI Journal 2025 Journal Article
Control oriented fast optimisation with SHapley additive explanations assisted two-stage training of input convex neural network
- Chuang Wang
- Lijun Zhang
Advancement of technology has enabled operation optimisation control of complex systems, which involves solving an optimisation problem. However, there are still challenges to harness the power of such technologies. This is especially true when it is required to solve a complex optimisation problem within prescribed time and resource limits. This paper introduces a novel fast optimisation framework by utilising input convex neural networks to approximate time-consuming optimal control algorithms. A two-stage training structure, consisting of offline data training and online fine-tuning, is put forward to ensure the effectiveness of the trained model. Moreover, an expansion data collection method and a roaming-style data augmentation strategy are proposed to improve the quality of training samples. A SHapley additive explanations assisted data selection algorithm is then developed to identify the most useful samples for training the model. Thus, reducing the training complexity. Further, a predictive safety filter is employed to guarantee that the control inputs produced by the trained model do not violate any constraints. Experiments on an inverted pendulum problem under four types of disturbances demonstrated that the trained input convex neural network controller exhibits superior control performance in comparison to a model predictive controller. Experimental results indicate that, in benchmark inverted pendulum on a cart control tests encompassing eight distinct scenarios, the trained input convex neural network controller achieves similar control performance comparable to that of the model predictive controller. Compared with the time model predictive controller, the trained input convex neural network controller demonstrates superior performance in scenarios with large initial angles and disturbances, though it performs slightly worse under disturbance-free conditions. Furthermore, the trained input convex neural network controller significantly outperforms both the model predictive controller and time model predictive controller in terms of computational efficiency, reducing computation time by 49. 42% and 94. 71%, respectively.