3D-CovDiffusion3D-Aware Diffusion Policy for Coverage Path Planning

Accepted at IROS 2026

Chenyuan ChenHaoran DingRan DingTianyu LiuZewen HeAnqing DuanYoshihiko Nakamura

Mohamed bin Zayed University of Artificial Intelligence · Abu Dhabi, UAE

Abstract

Diffusion models have shown strong potential for robot skill learning, yet their role in coverage path planning remains underexplored. In industrial surface processing (painting, polishing, spray coating), high coverage requires globally ordered, temporally coherent trajectories rather than stitching unordered local segments. We reformulate coverage path planning as conditional sequence generation and adopt a geometry-conditioned diffusion framework that synthesizes continuous trajectories directly from raw 3D point clouds. Our method produces temporally ordered trajectory chunks and avoids post-hoc heuristic ordering or stitching in prior learning-based methods via simple sequential concatenation, improving sequence-level consistency. A single shared policy generalizes across different geometries without category-specific architectures. Extensive benchmarks show substantial gains over prior learning-based baselines: 98.2% lower point-wise Chamfer Distance (lower is better), 97.0% lower jerk (smoother trajectories), and +67.5 percentage points overlapping surface coverage on average.

Sequence-level planning

Temporal ordering and surface coverage are modeled jointly as trajectory-level properties.

Geometry-aware generation

Raw point clouds and recent motion condition every iterative denoising stage.

Cross-geometry policy

One policy generates coherent paths across windows, cuboids, shelves, and a low-data container setting.

Video

3D-aware trajectory generation

Final 3D-CovDiffusion architecture with a geometry encoder, history context, state guidance, iterative diffusion, and sequential trajectory chunks
Method overview. Geometry and history features form the global condition for iterative denoising; state-guided trajectory chunks are then concatenated into a complete ordered trajectory.

Encode geometry

A point-cloud encoder maps the raw surface into a compact global observation feature.

Denoise a sequence

A conditional diffusion model jointly refines an ordered chunk of 6-DoF poses.

Roll forward

Recent execution history anchors the next chunk for direct sequential concatenation.

Comparison of our Method with Other Baselines

Synchronized trajectory-generation and surface-coverage comparisons against Point-Wise and PaintNet baselines across four matched examples per category.

Controls

Windows

Trajectory Generation

Surface Coverage

Cuboids

Trajectory Generation

Surface Coverage

Shelves

Trajectory Generation

Surface Coverage

Containers

Low-data setting

Trajectory Generation

Surface Coverage

Open resources

Dataset release

3D-CovDiffusion Train-Ready Dataset

2,470 training episodes in processed form, with 5,120-point observations and ordered 24D actions packaged for direct use by the training pipeline.

Open dataset ↗
Checkpoint release

Pretrained category policies

Validated EMA safetensors for Windows, Cuboids, Shelves, and Containers, with inference configs, provenance, and SHA-256 manifests.

Open models ↗

Citation

If this project is useful in your research, please cite the paper.

@misc{chen2026_3dcovdiffusion,
  title  = {{3D-CovDiffusion}: 3D-Aware Diffusion Policy
            for Coverage Path Planning},
  author = {Chen, Chenyuan and Ding, Haoran and Ding, Ran
            and Liu, Tianyu and He, Zewen and Duan, Anqing
            and Nakamura, Yoshihiko},
  year   = {2026},
  eprint = {2510.03011},
  archivePrefix = {arXiv},
  primaryClass  = {cs.RO},
  url    = {https://arxiv.org/abs/2510.03011},
  note   = {Accepted at IROS 2026}
}