Assessing Machine Learnability of Image and Graph Representations for Drone Performance Prediction

Abstract

Deep learning (DL) from various representations have succeeded in many fields. However, we know little about the machine learnability of distinct design representations when using DL to predict design performance. This paper proposes a graph representation for designs and compares it to the common image representation. We employ graph neural networks (GNNs) and convolutional neural networks (CNNs) respectively to learn them to predict drone performance. GCNs outperform CNNs by 2.6-8.1% in predictive validity. We argue that graph learning is a powerful and generalizable method for such tasks.