TopoDroneX Logo

TopoDroneX: A Multimodal Benchmark Dataset for Topology-optimized 3D Drone Design Exploration

Yeongtae Kim,  Hoonhyung Chung,  Donghyun Ra,  Sooyoung Lee

Industrial AI Lab (IAI Lab)

School of Mechanical Engineering, Chung-Ang University

Design Grid Animation

Overview

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.

Dataset Structure

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.
 

Design Shape

Stress Field Projection

Four direction views with stress distribution overlaid. Stored in img_s/ shards as {sample_id}.{view}.png.

Stress Field Projection

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.

3D Mesh Model

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).

3D Point Cloud

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)

Benchmark Results

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 Modeling

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.

Predictive Model Results

Image-based Generative Modeling

Image-based generative models are evaluated on a reconstruction task using four direction images as input.

Generative Model Image Results

Point-cloud-based Generative Modeling

Point cloud-based generative models are evaluated on an unconditional generation task.

Generative Model Point Cloud Results

Citation

@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}
}

License

This dataset is released under CC BY-NC 4.0 and is meant for academic use only.