jenkai-lin/cast-tree-voxel-diffusion

CAST: Connectivity-Aware Sampling for Topology

Connectivity-guided sampling for 3D voxel diffusion models. Improves trunk connectivity success rate from 6.5% to 46% on Minecraft-style tree voxel generation, without retraining or architecture changes.

Sampling progression: baseline DDPM (left) vs CAST with connectivity-aware guidance (right).

• 📄 Project page: https://drinkai.notion.site/Connectivity-Aware-Sampling-for-Topology-CAST-PoC-WIP-32aaa15702ab8010b4ded7ec8110a79d
• 💻 Code: https://github.com/KyleLin0927/cast-gen3d-tree

Files

• ckpt_0025_033_baseline_e0400.pt — Baseline 3D voxel DDPM (16³ resolution, ~313 MB)
• ckpt_0026_001_scorer_e100.pt — Connectivity scorer used for sampling-time guidance (~1.4 MB)

How to Use

from huggingface_hub import hf_hub_download

diffusion_ckpt = hf_hub_download(
    repo_id="jenkai-lin/cast-tree-voxel-diffusion",
    filename="ckpt_0025_033_baseline_e0400.pt"
)
scorer_ckpt = hf_hub_download(
    repo_id="jenkai-lin/cast-tree-voxel-diffusion",
    filename="ckpt_0026_001_scorer_e100.pt"
)

Full sampling pipeline and reproduction scripts: GitHub repository.

Training Details

Baseline diffusion

• Architecture: 3D U-Net
• Resolution: 16³ voxels
• Diffusion: T=1000 timesteps, cosine noise schedule (β: 0.0001 → 0.02)
• Training data: 1,286 Minecraft tree voxel volumes
• Training: 400 epochs, batch size 64, AdamW (lr 1e-4)

Connectivity scorer

• Architecture: 3D CNN
• Task: 4-class classification (positive / floating / disconnected / fragmented)
• Training data: 5,000 samples (20% real + 80% baseline-generated, with hard-negative mining)
• Loss: BCEWithLogitsLoss
• Training: 100 epochs on top of baseline checkpoint

Examples of Minecraft-style tree voxel volumes used for training.

Limitations

• Trained on 16³ resolution only; performance at higher resolutions not validated
• Tree-specific; generalization to other voxel-based topological objects untested
• Guidance produces a quantifiable trade-off between connectivity and structure — see project page for full ablation grid