1 Tianjin University 2 KEDACOM 3 Shenzhen University 4 Cardiff University
# Equal contribution * Corresponding author
Large-scale floorplan generation is critical for virtual space planning and architectural simulation. Although existing methods have shown success in generating small-scale floorplans with simple room shapes, they struggle to handle the complex room connections and irregular room shapes that arise in large-scale floorplans. In this paper, we propose CG-Floor, a centroid-guided hierarchical framework that explicitly decouples topology and geometry to address these issues. We first introduce the size-aware semantic centroid heatmap, derived from predicted room centroids, which provides a structured representation to precisely guide the effective generation of a coarse-to-fine floorplan generator while ensuring semantic alignment. Additionally, we train a vector quantized codebook of floorplans with complex room shapes to capture the diversity of room shapes and employ a latent diffusion transformer to generate large-scale floorplans featuring non-Manhattan room shapes. CG-Floor achieves state-of-the-art performance on the large-scale MSD dataset, and supports 3D floorplan conversion and editing, demonstrating the practicality of our approach.
Fig 1. The overview of our framework.
Hongjin Lian, Jian Ma, Hongjie Chen, Jia Li, Ruizhen Hu, Yu-Kun Lai, Kun Li. "CG-Floor: Centroid-Guided Diffusion for Large-Scale Floorplan Generation". Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2026.
@inproceedings{CGFloor,
author = {Hongjin Lian and Jian Ma and Hongjie Chen and Jia Li and Ruizhen Hu and Yu-Kun Lai and Kun Li},
title = {CG-Floor: Centroid-Guided Diffusion for Large-Scale Floorplan Generation},
booktitle = {Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year={2026}
}