Architectural floor plan design demands joint reasoning over geometry, semantics, and spatial hierarchy, which remains a major challenge for current AI systems. Although recent diffusion and language models improve visual fidelity, they still struggle with coherent spatial reasoning and controllable generation. We present HouseMind, a multimodal large language model that unifies floor plan understanding, generation, and editing in one framework. We introduce discrete room-instance tokens to construct a unified vocabulary that bridges layouts and symbolic reasoning. With multimodal alignment and instruction tuning, the model synthesizes coherent, controllable layouts from text instructions. Experiments show how the framework achieves superior geometric validity and controllability while remaining efficient and locally deployable.
Figures & tables
Figure 2: Understanding : given a prompt, an outline, and an existing floor plan, the model outputs a textual description, a bubble diagram, and structured JSON capturing spatial semantics. Generation : given a prompt and an outline, the model produces a complete, coherent floor plan. Editing : given a prompt, an outline, and a reference floor plan, the model outputs an updated plan aligned with the editing intent.
Figure 3: Overall framework of HouseMind. The model is trained through a three-stage multimodal alignment and instruction tuning pipeline: (S1) Embedding Initialization establishes cross-modal compatibility between geometric and linguistic tokens; (S2) Multimodal Pre-training aligns text and spatial representations; and (S3) Instruction Tuning (SFT) enables task-aware spatial reasoning.
Table 2: Generation results. Micro/Macro IoU measure pixel-level overlap; SSIM [ 40 ] and PSNR quantify perceptual similarity; FID [ 6 ] and GED [ 30 ] evaluate distributional realism; Node F1 and Edge Overlap assess graph-level correctness. * denotes methods without released code; results are reproduced.
Method
Δ IoU
Δ MSE ↓
Micro IoU
Macro IoU
GED ↓
Node F1
Edge Ovl.
Time (s)
Before edit
–
–
0.880
0.821
3.06
0.934
0.740
–
FLUX.1-Kontext-dev
0.053
0.0162
0.289
0.185
8.91
0.765
0.222
∼ 240
Qwen-Image-Edit-2509
0.088
0.0074
0.567
0.429
7.96
0.915
0.426
∼ 240
HouseMind-E
0.608
0.0019
0.855
0.823
0.467
0.998
0.934
∼ 3
HouseMind-O
0.598
0.0022
0.844
0.813
0.653
0.997
0.908
∼ 3
Table 3: Editing results. Δ IoU and Δ MSE measure editing precision (spatial and pixel-level change correctness); Micro/Macro IoU assess final layout quality; GED evaluates distributional realism; Node F1 and Edge Overlap assess graph-level consistency.
Figure 4: Qualitative comparison results of understanding, generation, and editing. For understanding tasks (U), HouseMind accurately identifies the number of rooms and their connections. For generation tasks (G), HouseMind preserves both the room layout and the overall outline consistency; the generation prompts are provided in the supplementary materials. For editing tasks (E), HouseMind accurately executes the specified modifications when the instructions are explicit.
Model
Train Loss ↓
Eval Loss ↓
w/o Stage 1 & 2
0.0729
0.0836
w/o Stage 1
0.0659
0.0840
w/o Stage 2
0.0712
0.0831
Full
0.0644
0.0830
Table 4: Loss under different training-stage configurations.
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
Figure A.1 : VQ-VAE tokenization framework. The outline branch encodes the global boundary, while the conditional room branch encodes each room with outline context to capture spatial relations.
Ntokens
Codebook
PSNR
SSIM
4×4
256
20.003
0.911
4×4
512
22.188
0.946
4×4
1024
23.202
0.957
8×8
256
35.898
0.998
8×8
512
36.657
0.998
8×8
1024
37.053
0.998
Appendix
Table A.1 : Outline branch: PSNR (dB) and SSIM under different token grids and codebook sizes.
Ntokens
Codebook
PSNR
SSIM
4×4
256
31.896
0.994
4×4
512
33.106
0.995
4×4
1024
34.506
0.997
8×8
256
44.624
1.000
8×8
512
48.467
1.000
8×8
1024
51.769
1.000
Appendix
Table A.2 : Room branch: PSNR (dB) and SSIM under different token grids and codebook sizes.
Size
Ma. IoU
FID ↓
GED ↓
Node F1
Edge Ovl.
256
0.654
1.89
1.03
0.994
0.880
512
0.657
1.95
1.04
0.995
0.877
1024
0.657
1.97
1.00
0.994
0.881
Appendix
Table A.3 : Codebook size ablation study
Figure C.1 : Post-processing examples.
Metric
w/o Correction
w/ Correction
Micro IoU
0.710
0.710
Macro IoU
0.654
0.654
SSIM
0.887
0.887
PSNR
16.0
16.1
FID ↓
1.87
1.89
GED ↓
1.10
1.03
Appendix
Table C.1 : Impact of Structural Correction in HouseMind-O .
Figure D.1 : Result distribution across tasks. HouseMind-O maintains stable, high-performance distributions in understanding and generation, with the bimodal ΔIoU in editing mainly caused by unclear spatial instructions.
Figure D.2 : Pixel–structure coupling analysis. Scatter plots show the correlation between Macro IoU and Edge Overlap for three generation paradigms.
Figure F.1 : Results of HouseMind on out-of-distribution cases.
In this work, we propose HypergraphFormer, a novel and efficient approach to floor plan generation based on learning hypergraph representations with a large language model (LLM). The model is trained via supervised fine-tuning to generate a hypergraph-based textual representation that encodes spatial relationships and connectivity information within floor plans. We train and evaluate our approach on the RPLAN dataset, and further demonstrate its generalizability on a separate out-of-distribution dataset, which we release in this paper. Our method outperforms state-of-the-art techniques based on rasterized or vectorized representations across a diverse set of metrics. We also show improved data efficiency, particularly under distribution shift. The hypergraph formulation enables the generation of floor plans for arbitrary, irregular, user-specified boundaries by decoupling apartment footprints from their functional and geometric subdivisions. Furthermore, we show that the proposed methodology offers a high degree of editability, making it particularly well suited to design-oriented workflows supported by LLMs.
An AI system for professional floor plan design must precisely control room dimensions and areas while respecting the desired connectivity between rooms and maintaining functional and aesthetic quality. Existing generative approaches focus primarily on respecting the requested connectivity between rooms, but do not support generating floor plans that respect numerical constraints. We introduce a text-based floor plan generation approach that fine-tunes a large language model (LLM) on real plans and then applies reinforcement learning with verifiable rewards (RLVR) to improve adherence to topological and numerical constraints while discouraging invalid or overlapping outputs. Furthermore, we design a set of constraint adherence metrics to systematically measure how generated floor plans align with user-defined constraints. Our model generates floor plans that satisfy user-defined connectivity and numerical constraints and outperforms existing methods on Realism, Compatibility, and Diversity metrics. Across all tasks, our approach achieves at least a 94% relative reduction in Compatibility compared with existing methods. Our results demonstrate that LLMs can effectively handle constraints in this setting, suggesting broader applications for text-based generative modeling.
Luis Lara, Aristides Milios, Zhi Hao Luo +5
Mila – Quebec AI Institute · Université de Montréal · Polytechnique Montréal +1
Two structural insights have been overlooked in automated residential floor plan generation. First, design is inherently progressive. Architects begin with rough strokes and refine them over time, whereas existing methods typically require their conditioning representation to be fully specified before generation, a fundamental mismatch with how design actually works. Second, the 2D floor plan is not an optional intermediate but an irreplaceable spatial contract. Once room boundaries, doors, and windows are fixed, furnishing reduces from open-ended spatial reasoning to bounded constraint satisfaction. Bypassing this contract, as existing 3D systems do by delegating layout to language models, yields overlapping rooms and implausible proportions; directly calling general-purpose language models likewise produces geometrically invalid layouts. Guided by these insights, we present PlanCraft. SketchPlan supplies the missing training signal by replaying the architect's drawing process on 80K real floor plans, producing partial sketches at every completeness level. PlanCraft-Diff progressively sharpens an incomplete sketch into a geometrically precise, vectorizable floor plan through a coarse-to-fine strategy. With the spatial contract established, PlanCraft-Agent then furnishes the scene within well-defined room boundaries. Experiments show that PlanCraft achieves a 61.1% lower FID than the best existing 2D method and surpasses existing 3D systems by 15 points in expert-rated spatial rationality, with a sketch at only 25% completion already outperforming all fully specified baselines.