School of Mechanical Engineering, Chung-Ang University
Unmanned aerial vehicle (UAV) design requires the optimal configuration of components and frame geometries to satisfy mission-specific performance requirements. However, conventional drone design processes often rely on repeated design modifications, numerical analysis, and flight tests, leading to substantial computational cost and time consumption.
Although data-driven approaches have recently gained attention as an alternative, the lack of multi-modal design datasets that jointly capture the complex mechanical characteristics and geometric diversity of multicopters has limited the development of AI-based drone design methods.
To address this limitation, this study proposes a large-scale drone design dataset consisting of 1,961 topology-optimized frame geometries generated under diverse component configurations and mechanical performance conditions, namely a TopoDroneX.
Proposed dataset is constructed based on five key design factors, including payloads and propulsion systems, and provides engineering attributes such as thrust, torque, and stress, together with multi-modal data including multi-view 2D projection images, 3D meshes, and point clouds.
We provide benchmark results for representative prediction and generation tasks across different data modalities. By incorporating both geometric information and physics-related design attributes, proposed dataset provides a physically grounded dataset for drone design optimization and broader data-driven engineering design.
| Property | Value |
|---|---|
| Total samples | 1,961 |
| Modalities | 2D images (8 views/sample), 3D mesh (STL + PLY), 3D point cloud, design parameters & specifications |
| Total size | 21.4 GB |
| License | CC BY-NC 4.0 |
2D Design Model
Design Shape
Four direction (top, front, side, isometric) views per sample. Stored in img/ shards as {sample_id}.{view}.png.
Stress Field Projection
Four direction views with stress distribution overlaid. Stored in img_s/ shards as {sample_id}.{view}.png.
3D Design Model
Mesh Model
3D mesh files in STL and PLY formats, stored in separate shards. geo_stl/ contains {sample_id}.stl and geo_ply/ contains {sample_id}.ply.
Point Cloud
Point clouds with per-point Von Mises stress. Stored in pc/ as Parquet files with columns: sample_id, x, y, z, mises_stress (Pa).
Design Parameter & Specification
spec.csv provides 17 columns per design entry, grouped by category below.
| Propulsion System | Material | Battery | Payload | Acceleration | Frame |
|---|---|---|---|---|---|
| Propeller diameter (inch) | Material density (kg/m³) | Battery length (mm) | Payload weight (kg) | Acceleration (G) | Motor to Motor Distance (mm) |
| Propulsion system thrust (kgf) | Young's modulus (GPa) | Battery width (mm) | Frame Height (mm) | ||
| Propulsion system torque (Nm) | Poisson ratio | Battery height (mm) | Frame Volume (m³) | ||
| Tensile yield strength (MPa) | Battery weight (kg) | Frame Mass (kg) |
We provide two benchmark tasks, predictive modeling and generative modeling, for each modality (2D images and 3D point clouds), yielding four benchmark settings in total.
| Benchmark | Data modality | Model | Metric |
|---|---|---|---|
| Predictive modeling | RGB image | VGG-19, ResNet-152, ViT-B16, MLP-Mixer-B | R², MAE, MSE |
| 3D model | Deepsets, PointNet++, PointCNN, DGCNN | ||
| Generative modeling | RGB image | VQGAN, β-VAE, DDIM, DiT | SSIM, MS-SSIM, KID, LPIPS |
| 3D model | Latent-WGAN, Diffusion PC, Pointgrow, Pointflow | COV-CD, EMD MMD-CD, EMD 1-NND-CD, EMD JSD |
Predictive models are trained to regress the design parameters from each input modality. Models are divided into image-based models, which take four direction images, and point-cloud-based models, which take a point cloud as input.
Image-based generative models are evaluated on a reconstruction task using four direction images as input.
Point cloud-based generative models are evaluated on an unconditional generation task.
@article{Kim2026,
title={TopoDroneX: A Multimodal Benchmark Dataset for Topology-optimized 3D Drone Design Exploration},
author={Kim, Yeongtae and Chung, Hoonhyung and Ra, Donghyun and Lee, Sooyoung},
note={Under review},
year={2026}
}
This dataset is released under CC BY-NC 4.0 and is meant for academic use only.