CVPR 2026

CG-Floor: Centroid-Guided Diffusion for Large-Scale Floorplan Generation

 

Hongjin Lian1#, Jian Ma1#, Hongjie Chen1, Jia Li2, Ruizhen Hu3,*, Yu-Kun Lai4, Kun Li1,*

1 Tianjin University   2 KEDACOM   3 Shenzhen University   4 Cardiff University  

  # Equal contribution   * Corresponding author

 

[Paper] [Supplemental] [Code]

 

Abstract

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.


Method

 

 

Fig 1. The overview of our framework.

 


Demo

 

 


Technical Paper

 

 


Citation

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