English

Rollback-Free Stable Brick Structures Generation

Machine Learning 2026-05-11 v1

Abstract

While autoregressive models have advanced 3D generation, creating physically stable brick structures remains a challenge due to the strict requirements of gravity and interconnectivity. Existing approaches rely on external physical simulators during inference to perform rejection sampling and brick-by-brick rollbacks, which severely bottlenecks efficiency. To address this, we propose a reinforcement learning paradigm that shifts physical validity enforcement from test-time correction to training-time policy optimization. By utilizing assembly-level rewards, the model optimizes for collision avoidance, global connectivity, structural interlocking, and shape conformity. This paradigm allows the model to internalize physical priors, enabling the first rollback-free generation of stable brick structures. Experimental results demonstrate that our approach achieves state-of-the-art generation quality while accelerating inference speed by orders of magnitude. Our code and dataset are available at https://github.com/miniHuiHui/STABLE. Our models are available at https://huggingface.co/miniHui/STABLE.

Keywords

Cite

@article{arxiv.2605.06947,
  title  = {Rollback-Free Stable Brick Structures Generation},
  author = {Chenhui Xu and Ziyue Bai and Fuxun Yu and Heng Huang and Jinjun Xiong},
  journal= {arXiv preprint arXiv:2605.06947},
  year   = {2026}
}
R2 v1 2026-07-01T12:56:18.510Z