Application of machine learning for ‘what if?’ stress analysis

Analysis of stresses in components of even modestly complex geometries often require the use of finite element analysis (FEA). For testing a large number of design options quickly, FEA can be time consuming and provides more accuracy than required. In this project, a machine learning-based system is developed to provide quick and approximate solutions to stress analysis problems in parametrised compressor disc geometries. Simple mechanics problems were completed preliminarily to test the practicality of machine learning approaches for this application. This included applying instance selection by the [Formula: see text]-medoids algorithm to a 2D FEA problem. A parametrised compressor disc geometry was designed and defined by eight dimensions. Stress fields were produced from two superimposed loading schemes: the rotational body force and the force exerted on the disc by the blades 495,338 samples of training data were collected from 4374 FEA simulations. Four networks were trained to predict stresses caused by each loading scheme in order to produce stress fields. The best network was of the structure 9/20/20/2 and used normalised training data. For the geometry tested, it predicted stresses with a root-mean-square error of 1.51%. The code took 0.7 s to run in total, from start-up to completion of a stress plot. The importance of the inputs in the training data set were scored with a feature selection algorithm to aid further optimisation of the system. The low computation time makes this system suitable for the early stages in a design process.