FP-MAE: A Self-Supervised Model for Floorplan Generation with Incomplete Inputs

Floor plans are a central representational component of architectural design, operating in close relation to sections, elevations, and three-dimensional reasoning to support the production and understanding of architectural space. In this context, we address the bounded computational task of completing incomplete floor plan representations as a form of early-stage design assistance, rather than treating the floor plan as an isolated architectural object. Within this workflow, being able to automatically complete a floor plan from an unfinished draft is highly valuable because it allows architects to generate preliminary schemes more quickly, streamline early discussions, and reduce the repetitive workload involved in revisions. To meet this need, we present FP-MAE, a self-supervised learning framework designed for floor plan completion. This study proposes three core contributions: (1) We developed FloorplanNet, a dedicated dataset that includes 8000 floorplans consisting of both schematic line drawings and color-coded plans, providing diverse yet consistent examples of residential layouts. (2) On top of this dataset, FP-MAE applies the Masked Autoencoder (MAE) strategy. By deliberately masking sections of a plan and using a lightweight Vision Transformer (ViT) to reconstruct the missing regions, the model learns to capture the global structural patterns of floor plans from limited local information. (3) We evaluated FP-MAE across multiple masking scenarios and compared its performance with state-of-the-art baselines. Beyond controlled experiments, we also tested the model on real sketches produced during the early stages of design projects, which demonstrated its robustness under practical conditions. The results show that FP-MAE can produce complete plans that are both accurate and functionally coherent, even when starting from highly incomplete inputs. FP-MAE is a practical and scalable solution for automated floor plan generation. It can be integrated into design software as a supportive tool to speed up concept development and option exploration, and it also points toward broader opportunities for applying AI in architectural automation. While the current framework operates on two-dimensional plan representations, future extensions may integrate multi-view information such as sections or three-dimensional models to better reflect the relational nature of architectural design representations.