Eliminating Rasterization - Direct Vector
Eliminating Rasterization - Direct Vector
Abstract—The boundary-constrained floor plan generation generative model (e.g., CNN or GAN) to directly generate a
problem aims to generate the topological and geometric proper- raster image that contains the geometric and topological in-
ties of a set of rooms within a given boundary. Recently, learning- formation of the room(s), and eventually extracting the vector
based methods have made significant progress in generating
realistic floor plans. However, these methods involve a workflow information from this raster output. Recently, some approaches
arXiv:2508.13738v1 [[Link]] 19 Aug 2025
of converting vector data into raster images, using image-based have used a hybrid strategy, i.e., using graph neural networks
generative models, and then converting the results back into vec- (GNN) or Transformer, to directly predict vector information
tor data. This process is complex and redundant, often resulting about the layout, such as the coordinates of a bounding box
in information loss. Raster images, unlike vector data, cannot representing a room [8], [11], or a bubble diagram containing
scale without losing detail and precision. To address these issues,
we propose a novel deep learning framework called DiffPlanner the locations of nodes and their adjacencies [28]. However,
for boundary-constrained floor plan generation, which operates they still depend on image-based generative models, e.g., to
entirely in vector space. Our framework is a Transformer-based perform feature extraction on the input raster boundaries and
conditional diffusion model that integrates an alignment mecha- the ground truth raster images [8], [11], [28], or to generate a
nism in training, aligning the optimization trajectory of the model raster layout image to provide additional information for floor
with the iterative design processes of designers. This enables our
model to handle complex vector data, better fit the distribution plan prediction [8], [11].
of the predicted targets, accomplish the challenging task of floor Existing methods could generate some results comparable
plan layout design, and achieve user-controllable generation. to those of professional architects, thanks to the carefully
We conduct quantitative comparisons, qualitative evaluations, designed models. However, their workflow of converting vec-
ablation experiments, and perceptual studies to evaluate our tor information into images, then performing the generation
method. Extensive experiments demonstrate that DiffPlanner
surpasses existing state-of-the-art methods in generating floor process via an image-based generative model, and finally
plans and bubble diagrams in the creative stages, offering more converting the raster results into vector data is unnecessarily
controllability to users and producing higher-quality results that redundant and overly complex. Moreover, when converting
closely match the ground truths. vector data, like floor plans, to raster images, information loss
Index Terms—Floor plan generation, bubble diagram, deep often occurs [5], [13]. Vector data, which is defined by precise
generative modeling. geometry and topology, can scale infinitely without distortion.
In contrast, raster images are made of fixed-resolution pixels,
which can result in lost details and precision, and are scale-
I. I NTRODUCTION
dependent. Common issues include blurred edges, altered
}
(a) Vector floor plan (c) Data tensors of vector-based floor plan generation
Fig. 1. Most previous methods with input and output vector data (a), internally rasterize the input into a series of images representing the layout and then
use image-based networks for the prediction process followed by vectorization of the output (b). In contrast, our DiffPlanner performs the prediction process
directly based on data tensors representing the vector space (c).
of autonomously generating the complete floor plan layouts based on diffusion models, which directly predicts vector
without any information beyond the boundaries. However, floor plans from vector boundaries. Previous methods
the process should be controllable on a high level, allowing depended (internally) on raster space, requiring the
users to input simple conditions such as the number of rooms conversion of vector information into a set of raster
and their categories. Additionally, it should be adjustable, images, followed by prediction processes using image-
permitting iterative user interventions at each step of the based generative CNN or GAN models.
design. 2) We have also designed an alignment mechanism to better
Based on this, we propose DiffPlanner, a novel deep learn- align the optimization trajectory of the model with the
ing framework for boundary-constrained floor plan generation. common iterative manual design processes during the
Our framework leverages recent popular diffusion models training stage.
(DM) [10], which have shown remarkable success in various 3) Our framework can address a coarse-to-fine floor plan
generative tasks, to handle the non-trivial challenge of floor layout design problem, even for boundary-unconstrained
plan layout design. It incorporates a Transformer [29] as the floor plan generation. It produces bubble diagrams that
backbone of the model, enabling the handling of complex represent geometric and topological information of de-
vector layout data. Additionally, it integrates a conditioning sign elements in the early stages, and detailed floor plans
mechanism that allows for controlled floor plan generation in the later stages. Our approach supports various levels
under specific conditions. We have designed an alignment of user interaction, including fully automatic, coarsely
mechanism during the training phase of the model, which en- controllable, and finely controllable options. Compared
ables the model to extract information from the iterative design to previous methods, our approach can generate more
processes of designers, thereby better fitting the distribution of layout information and provide users with richer inter-
predicted targets. actions.
Moreover, our approach also supports generating bubble
diagrams and floor plans without predefined boundaries. Pre- II. R ELATED WORK
vious methods that do not support predefined boundaries have A. Bubble Diagram Generation
been significantly limited in terms of user controllability – they
either fully automate the generation process without allowing Bubble diagrams establish a connection between the intent
user intervention [28] or only support bubble diagram-driven and solution of designer, and they are widely used in data-
floor plan generation [21], [22], [25], neglecting the genera- driven layout planning tasks such as architectural design and
tion of bubble diagrams representing the early design stages. interior scene synthesis [11], [21], [22], [25]. In previous
Compared to previous methods, our new approach produces work, they often serve as an input condition representing
higher-quality results that are closer to the ground truths and user intent, or as an intermediate representation between input
supports various forms of user interaction. conditions and final layout prediction. Almost no work focuses
on data-driven bubble diagram generation. Recently, Sun et
Overall, our contributions are as follows:
al. [28] proposed a data-driven approach (BubbleFormer) for
1) We frame the task of boundary-constrained floor plan generating bubble diagrams to better drive downstream layout
generation as a generative task fully in vector space, planning tasks. BubbleFormer requires rasterizing the input
for the first time, and propose a generation framework boundaries and the target bubble diagrams into images, then
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 3
TABLE I
C OMPARED TO EXISTING METHODS FOR BOUNDARY- CONSTRAINED BUBBLE DIAGRAM AND FLOOR PLAN GENERATION , OUR APPROACH IS BASED ON
STATE - OF - THE - ART GENERATIVE MODELS , SPECIFICALLY DIFFUSION MODELS (DM), AND PERFORMS PREDICTIONS DIRECTLY IN VECTOR SPACE
WITHOUT ANY INTERMEDIATE RASTERIZATION OPERATIONS . O UR METHOD SUPPORTS THE MOST EXTENSIVE RANGE OF USER INTERACTIONS , FROM
FULLY AUTOMATIC GENERATION , WHERE A VECTOR LAYOUT CAN BE OBTAINED ONLY FROM THE INPUT BOUNDARY, TO COARSELY CONTROLLABLE
PREDICTION , WHERE USERS CAN PROVIDE SIMPLE INSTRUCTIONS TO SPECIFY ROOM NODES (Rnode ), ROOM ADJACENCIES (Radja ), AND ROOM
PARTITIONING (Rpart ), TO FINELY CONTROLLABLE DESIGN , WHERE USERS CAN ITERATIVELY ADJUST EACH STEP OF THE DESIGN PROCESS .
using a CNN to extract features and subsequently performing floor plans not as raster images but as geometric vectors of
the prediction process. Finally, it needs to employ the Hungar- the bounding box coordinates [8], [11]. However, these ap-
ian matching algorithm [16] to obtain unique matches between proaches still required starting from a rasterized boundary and
the generated nodes and edges. generating a rasterized layout image as auxiliary information
While BubbleFormer can automatically generate diverse for the final layout prediction.
results for users to browse and select their preferred de- Unlike all previous methods, our approach defines the
signs, the entire generation process does not allow for user boundary-constrained floor plan generation problem as a
interaction. But interactivity is very useful for an AI tool vector-to-vector generation task, directly predicting the vector
focused on layout planning [26]. Unlike BubbleFormer, our floor plan from the vector boundary without any rasterization
method supports various forms of user interaction, as shown operations, as shown in Table I. This enables our model
in Table I. It allows users to simply input instructions, such to generate higher-quality and more reasonable floor plans.
as the number and categories of nodes, and also supports Additionally, our model supports more diverse user-controlled
more detailed interactions, such as iterative design of nodes generation options, enhancing interactivity compared to previ-
and adjacencies. Additionally, our method does not require ous methods. This makes our approach more aligned with the
redundant rasterization operations. All predictions are made user needs and preferences in the design process.
directly in vector space, and no post-processing is needed.
C. Floor Plan Generation without Boundary Constraints
Floor plan generation without boundary constraints is of-
B. Boundary-constrained Floor Plan Generation
ten driven by bubble diagrams. Bubble diagrams effectively
Floor plan generation has been a prominent topic in the capture the design intent of architects during the initial stages
intersection of computer graphics [3], [19], [31] and architec- of design. Therefore, some learning-based approaches have
ture [1], [20], [24] for decades, aiming to achieve automated been proposed to enable architectural floor plan generation
design of architectural floor plans through computational by mapping bubble diagrams to floor plans [21], [22], [25].
design methods. The exterior boundary of a building is a However, these methods typically focus only on the final
crucial condition for floor plan generation [32]. In recent years, design stage, converting already finalized design ideas into the
more and more researchers have started using learning-based final layout, while neglecting the early creative stages, such as
methods to tackle the problem of boundary-constrained floor the bubble diagram generation phase.
plan generation [4], [8], [11], [27], [30], [32]. Architects indeed need an automated tool to help convert
Early researchers directly employed image-based generative bubble diagrams, which convey design intent, into final floor
models, such as Pix2Pix [14], to predict a rasterized floor plans to reduce their workload. However, they also need an
plan image from a rasterized boundary image. However, the AI design tool that collaborates with them in the creative
generated results were often too noisy to be vectorized [4]. stage, providing more inspiration and reducing their workload
Subsequently, some researchers began dividing the floor plan in conceptual design [26]. Unlike previous methods, our
generation process into multiple sub-tasks [27], [30], [32]. proposed approach collaborates with users from the initial
Starting with a rasterized boundary, they used image-based creative stage to the final design stage, first creating bubble
models to first predict intermediate raster representations, diagrams and then generating the final floor plans. Compared
such as location masks, activity maps, or wall maps, and to the state-of-the-art method in bubble diagram-driven floor
then predicted rasterized floor plans from these intermediate plan generation, HouseDiffusion [25], our approach produces
representations, resulting in more reasonable outputs that could higher-quality results that are closer to the ground truths and
be vectorized. Next, some researchers attempt to represent supports richer user interactions.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 4
"nodes": [{
floor plans. We use the same data preprocessing method as "id": int,
"category": int,
in [11] to extract the necessary vector information, such as the "size": int,
"location": [x, y]}, ... ]
room numbers, room categories, room sizes, room locations, "adjacencies": [[id, id], ... ],
"partitioning": [[[x, y], ... ], ]} partitioning: 8x4
room adjacencies, and other relevant details (Figure 2(a)). We
-1 1 Living room Bedroom Kitchen
further process this information into a data tensor format that
-
Bathroom Balcony Storage
our vector-based model can accept (Figure 2(b)) to obtain the
final dataset including 80K pairs of data, and randomly split Fig. 2. Overview and data representation of DiffPlanner and previous
methods. Our DiffPlanner decomposes the floor plan design process into
it into 56K-12K-12K for training-validation-test sets. several stages based on established design principles, allowing for human
Challenge. Achieving our goal is non-trivial. Firstly, it input at various levels of detail. Unlike previous methods that use raster
is challenging for the model to directly extract information images (c), which require the rasterization of vector data, our approach
operates directly on data tensors representing the vector space (b). Starting
from vector data and learn layout planning capabilities. For with input vectors for the room boundary and entrance, our model first predicts
example, humans find it easier to capture initial layout in- room nodes including their number, categories, sizes, and rough locations. It
formation from images compared to structured vector data then determines the adjacency matrix between these nodes to form a bubble
diagram, representing the initial design phase. Finally, our model predicts the
(Figure 2(a) vs. Figure 2(c)). Secondly, we expect this model to room partitioning to complete the floor plan.
handle diverse layout planning tasks, such as generating bubble
diagrams in the early design stages, producing final floor
plans in the later stages, and even tackling layout planning as specifying the number and categories of rooms, iteratively
tasks without boundary constraints. Finally, we want this controlling the room locations and their adjacencies, and
model to deeply interact with users. It should not only be determining the partitioning of rooms.
fully automatic but also allow for coarse user control, where
users can provide simple instructions, and fine control, where
IV. M ETHOD
users can iteratively control the process. Overall, the model
is expected to output complex layouts from simple input In this section, we first provide a brief introduction to the
information, with the entire generation process being highly diffusion models (DM) and then explain how our method
controllable by the user and performed entirely in vector space. progressively predicts room nodes, room adjacencies, and
Methodology. Similar to existing methods [8], [32], we also room partitioning from the input boundary.
appropriately decompose the floor plan design procedure into
several stages based on fundamental design principles and
widely adopted practices [23], allowing for human input at A. Diffusion Models (DM)
various levels of detail. Note that, unlike previous methods, The diffusion model [10] is a generative framework that
our approach performs predictions entirely in vector space, gradually transforms a simple initial distribution into a com-
without the need to rasterize the vector information. As shown plex target distribution through a series of iterative steps. This
in Figure 2(b), starting from a data tensor storing the input model operates by adding noise to the data and then learning to
vector boundary, that includes the entrance, our model first reverse this process, effectively denoising to generate realistic
predicts a set of room nodes including the information about samples. It is particularly effective for capturing intricate
number, categories, sizes, and approximate locations. Next, patterns and dependencies in data, making it suitable for
it predicts the adjacency matrix between these nodes to then tasks that require fine-grained detail and structure. Specifically,
create a bubble diagram, reflecting the initial design stage. diffusion models (DM) transform Gaussian noise xT into a
Subsequently, our model further predicts the room partitioning data sample x0 through a series of T denoising steps, involving
to generate the final floor plan. We allow the user to interact both forward and reverse processes during training. In the
with the model throughout the entire generation process, such forward process q(xt |x0 ), a data sample x0 is progressively
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 5
where ϵθ is a function approximator and is optimized by: Boundary Entrance Number Category Partial input
be represented by elements that have been confirmed by the mask [28], combined with a rasterized node mask, to indicate
designer and elements that remain unconfirmed in the mind room adjacencies (Figure 2(c)). In contrast, we adopt a more
of the designer. For a design target x0 with n elements to be intuitive and efficient approach by using an adjacency matrix
predicted, we can easily calculate the number of confirmed to represent the adjacencies between rooms. In this matrix,
nec = min(n(1 − t/T ) + 1, n) and unconfirmed elements a value of 1 indicates that two nodes are connected, while a
neunc = n−nec in the intermediate result xinter
t at time step t. value of 0 indicates that they are not. We then normalize the
Clearly, the confirmed elements ec should match the prediction matrix to a range of [−1, 1]. This new representation allows
target values, while the unconfirmed elements eunc should also the prediction of room adjacencies to be carried out entirely
possess a certain level of plausibility, as the designer may in vector space, eliminating the need for any rasterization
already have a rough idea of them, even if they have not been operations.
finalized. Thus, we first initialize xinter
t with x0 . Then we AdjacencyDiff. We propose AdjacencyDiff to predict room
randomly select neunc indices and set the elements with these adjacencies under the conditions of boundary, entrance, room
indices to a weighted combination of x0 and random noise nodes, and/or partial input. Its network architecture, training
xnoise ∼ N (0, I), xinter
t eunc = x0eunc k + xnoiseeunc (1 − k), strategy, and implementation details are the same as those of
where k = (1 + nec /n)/2. As the number of confirmed ele- NodeDiff.
ments nec increases, k will approach 1. Because, as the design Alignment. The design of room adjacencies is also an itera-
becomes progressively refined, the unconfirmed elements eunc tive process. Therefore, AdjacencyDiff incorporates an align-
will increasingly resemble the final design x0 . ment strategy in training, with slight differences from NodeD-
Now, at each time step t, we have a rough estimate x̃θt of x0 iff in obtaining intermediate results xinter
t . We first extract a
and an intermediate result xinter
t corresponding to the iterative list containing n room adjacencies to be predicted, where each
design process of the designer. Thus, the total loss is defined room adjacency is a tuple of room IDs. We then calculate the
as L = Lt-1 + λLalign : number of room adjacencies nec = min(n(1−t/T )+1, n) that
should be confirmed at time step t. Next, we randomly select
nec adjacencies as confirmed adjacencies, while the remaining
L = ||ϵθ (xt , t, c) − ϵ||2 + λ||x̃θt (xt , ϵθ ) − xinter
t ||2 . (7)
neunc = n − nec adjacencies are considered unconfirmed.
where Lt-1 is the conditional DDPM [10] loss for time step Similar to NodeDiff, we first initialize xinter
t as x0 . Then, for
t, and Lalign is the alignment loss for time step t. This align- the indices represented by the unconfirmed room adjacencies
ment mechanism effectively guides the sampling trajectory at eunc , we assign a weighted sum of x0 and random noise
each step to match the iterative design process, distilling the xnoise ∼ N (0, I), xinter
t eunc = x0eunc k + xnoiseeunc (1 − k),
optimizer throughout the reverse process in training. where k = (1 + nec /n)/2.
User-controllable generation. Our NodeDiff is not only User-controllable generation. Similar to NodeDiff, our Ad-
fully automated but also easily supports user-controllable gen- jacencyDiff model also supports generating room adjacencies
eration. By simply modifying the conditions input to the model under various conditions, such as specifying the boundary or
in (5), various forms of user control can be achieved. NodeDiff not, and providing partial input. This can be easily achieved
supports a range of user inputs, such as specifying whether or by simply adding or removing conditions in (5).
not to include the boundary and entrance, providing the desired
number of rooms and their categories, and incorporating partial D. Partitioning Generation
input. This flexibility allows users to guide the generation Problem. In this task, our goal is to predict room parti-
process according to their specific needs and preferences. tioning from the input boundary with entrance, as well as the
Implementation details. We implemented our NodeDiff us- room nodes generated by NodeDiff and the room adjacencies
ing PyTorch, based on the public implementation of Guided- predicted by AdjacencyDiff. This will generate a floor plan
Diffusion [6]. The optimizer used is Adam [15] with decoupled that adapts to the final stage of the design process.
weight decay [18] for 300k steps with a batch size of 1024 on Representation. Unlike some previous methods that repre-
an NVIDIA RTX 2080 Ti GPU. The initial learning rate was sent room partitioning as a set of multi-channel binary raster
set to 1e-3 and was reduced by a factor of 10 after every masks [27], [32] or three-channel color images [4], [30], we
100k steps. We set the number of diffusion steps to 1000 follow the approach in [8], [11] by using the coordinates of
and uniformly sampled t during training. Please refer to the the top-left and bottom-right corners of room bounding boxes
supplementary materials for more details. to represent room layouts. This allows our model to perform
the prediction process entirely in vector space.
C. Adjacency Generation PartitioningDiff & Alignment. We propose PartitioningDiff
Problem. In this task, our goal is to predict room adjacencies to predict room partitioning under the conditions of boundary,
from the input boundary with the entrance, as well as the room entrance, nodes, adjacencies, and/or partial input. Its network
nodes generated by NodeDiff. The room adjacencies reflect architecture, training strategy, alignment mechanism, and im-
the relationships between rooms, forming an essential part of plementation details are the same as those of NodeDiff.
the bubble diagram and serving as a crucial condition for the User-controllable generation. By simply adding or remov-
subsequent floor plan design. ing conditions in (5), our PartitioningDiff also supports gen-
Representation. Previous methods represent room adjacen- erating room partitioning under various conditions, such as
cies as specific line segments in raster space to create a connect specifying the boundary or not, and providing partial input.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 7
TABLE IV TABLE V
S TATISTICS COMPARISON . E ACH STATISTIC IS THE RATIO CALCULATED Q UANTITATIVE EVALUATION OF HOW WELL OUR D IFF P LANNER ADHERES
BASED ON THE GROUND TRUTH BUBBLE DIAGRAMS AND FLOOR PLANS . A TO THE INPUT CONDITIONS UNDER DIFFERENT CONTROL MODES FOR
RATIO CLOSE TO 1 SUGGESTS THAT THE GEOMETRY AND TOPOLOGY OF BUBBLE DIAGRAM AND FLOOR PLAN GENERATION TASKS . A LOWER
THE MODEL - GENERATED RESULTS ARE MORE SIMILAR TO THE GROUND MAEavg INDICATES STRICTER ADHERENCE TO THE INPUT CONDITIONS .
TRUTHS .
Bubble Diagram Input Condition Output MAEavg
Bubble Diagram Statistics Generation B Rn,c,s,l w/ Proc Rn Rc Rs,l
n l r Ln La
Generation Ravg Cavg Cavg avg avg DiffPlannerbII ✓ Rn ✗ 0.0 - -
BubbleFormer 0.932 0.376 0.369 0.969 1.074 DiffPlannerbIII ✓ Rn,c ✗ 0.0 0.0 -
DiffPlannerbI 0.996 1.000 1.004 0.998 1.004 DiffPlannerbIV ✓ Rn,c,s,l ✗ 0.0 0.0 0.0
Floor Plan Statistics Floor Plan Input Condition Output MAEavg
Generation n
Ravg l
Cavg r
Cavg Ln La Generation B Rn,c,s,l,a w/ Proc Rn Rc Rs,l Ra
avg avg
DiffPlannerbII ✓ Rn ✓ 0.0 - - -
RPLAN 0.869 0.851 0.997 0.958 1.045
DiffPlannerbIII ✓ Rn,c ✓ 0.0 0.0 - -
ActFloor-GAN 0.904 0.864 0.965 0.935 0.902
DiffPlannerbIV ✓ Rn,c,s,l ✓ 0.0 0.0 0.008 -
WallPlan 0.998 0.968 0.973 0.970 0.911
DiffPlannerbV ✓ Rn,c,s,l,a ✓ 0.0 0.0 0.008 0.025
iPLANI 0.938 0.980 1.049 0.969 1.131
DiffPlannerbVI ✓ Rn,c,s,l,a ✗ 0.0 0.0 0.001 0.007
Graph2PlanI 0.980 0.971 0.988 0.969 1.029
DiffPlannerbI 0.996 0.997 1.001 0.998 0.971
BubbleFormer DiffPlannerI
b b
DiffPlannerII
b
DiffPlannerIII
b
DiffPlannerIV GT DiffPlannerI DiffPlannerII DiffPlannerIII DiffPlannerIV GT
b
GT DiffPlannerI Graph2PlanI iPLANI WallPlan ActFloor-GAN RPLAN ArchiGAN
Fig. 6. Qualitative comparison of floor plan generation from boundary only between our method (DiffPlannerbI ) and previous state-of-the-art methods
(Graph2PlanI , iPLANI , WallPlan, ActFloor-GAN, RPLAN, and ArchiGAN). The unreasonable design is highlighted in the red box. Our method produces
floor plans closest to the ground truths (GT), demonstrating the effectiveness of our vector-based approach in generating accurate and high-quality floor plans.
b
GT DiffPlannerI in vector space, taking vectorized boundaries as input and
progressively generating the final high-quality vector floor
plans. As shown in Figure 6, our method produces floor
plans that are closest to the ground truths (GT), while other
methods exhibit varying degrees of unreasonable results. This
demonstrates the effectiveness of our vector-based approach
in generating accurate and high-quality floor plans.
Figure 7 qualitatively demonstrates that our method
(DiffPlannerbI ) can generate multiple different high-quality
floor plans from a single input boundary. This indicates that
our model successfully captures the diverse design space in the
floor plan generation task, mapping a single input to multiple
possible outputs. Within the diverse outputs of our model,
some results may closely resemble the corresponding ground
truths (GT), as seen in the samples shown in the second and
third rows of Figure 7. This is expected, as the GT itself is one
feasible solution within this diverse design space. For a given
input boundary, we do not expect the outputs of the model to
always match the corresponding specific GT. Instead, we aim
for the model to generate multiple reasonable and high-quality
floor plans, reflecting the inherent diversity of architectural
design solutions.
Our method can also be applied to another prominent task:
bubble diagram-driven floor plan generation. We conducted
a qualitative comparison with the state-of-the-art method,
Fig. 7. Qualitative evaluation of output diversity. Our method (DiffPlannerbI ) HouseDiffusion [25]. Although HouseDiffusion can predict
can generate multiple high-quality floor plans from a single input boundary. floor plans based on the number (Rn ), categories (Rc ), and
adjacencies (Ra ) of rooms provided in a bubble diagram,
it still tends to produce some unreasonable results, such as
that can be vectorized. Graph2Plan and iPLAN take rasterized missing rooms, illogical adjacencies, and noisy boundaries, as
boundaries as input but represent their prediction targets as shown in Figure 8. In contrast, our DiffPlannerIII can generate
vector bounding box coordinates, although they still require reasonable layout results even when only the number (Rn )
the model to predict a rasterized layout image as an auxiliary. and categories (Rc ) of rooms are provided as inputs. When
Unlike all previous methods, our approach operates entirely additional information such as the sizes (Rs ), locations (Rl ),
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 12
NodeDiffI
NodeDiffII
NodeDiffIII
Adjacency
Diff
Partitioning
Diff
Fig. 8. Qualitative comparison of bubble diagram-driven floor plan generation
between our method and HouseDiffusion. HouseDiffusion often produces
unreasonable results like missing rooms and noisy boundaries. In contrast, Fig. 9. Our three components (NodeDiff, AdjacencyDiff, & PartitioningDiff)
our method generates realistic layouts with just the number and categories can predict complete layouts from partial inputs with varying proportions
of rooms (DiffPlannerIII ), and closely matches the ground truths (GT) when (25%, 50%, & 75%), enabling detailed user interaction and fine-grained
additional information is provided (DiffPlannerV ). iterative design. Each component produces high-quality results that closely
match the ground truths (GT) from the partial inputs. Elements with red
borders indicate user-specified partial inputs.
and adjacencies (Ra ) of rooms is added, our DiffPlannerV
produces results that are almost identical to the ground truths
(GT). This demonstrates the robustness and accuracy of our AdjacencyDiff, & PartitioningDiff) to demonstrate its effec-
approach in generating high-quality and realistic floor plans tiveness both qualitatively and quantitatively.
from bubble diagrams. 1) Qualitative comparison: Figure 10 presents visual com-
3) Iterative design with partial input: Real-world floor plan parisons of the intermediate outputs produced at various dif-
layout planning is an iterative design process, making it crucial fusion steps by our three core components (NodeDiffI , Adja-
for a data-driven layout planning tool to support user-driven cencyDiff, & PartitioningDiff) and their variants without the
iterative design. Unlike previous methods that decompose the alignment mechanism (NodeDiffI w/o Align, AdjacencyDiff
generation process into several sub-tasks, allowing users to w/o Align, & PartitioningDiff w/o Align). The input conditions
modify intermediate results for rough control, our goal is used by these components and their variants are listed in
to enable iterative design at each sub-task. This ultimately Table VIII. The comparison results indicate that the alignment
allows users to have fine-grained control over the entire design mechanism allows our model to more quickly and effectively
process, from initial concept to finalization, for every element approximate the ground truth distribution.
in the layout. The key to achieving this goal is equipping the For the room node generation task, NodeDiffI with the
model with the ability to predict the complete layout from alignment mechanism can already generate a preliminary
partial user inputs at each sub-task. result at step 50 that reflects the geometry and topology of
Figure 9 demonstrates the ability of our three components the final output. By step 20, it essentially produces results
(NodeDiff, AdjacencyDiff, & PartitioningDiff) to predict com- consistent with the final output and continues to refine the
plete layouts from partial inputs with varying proportions output in subsequent diffusion steps, finally producing geo-
(25%, 50%, & 75%). The input conditions used by each metrically and topologically coherent room nodes at step 0. In
component are listed in Table VII. Each component performs contrast, NodeDiffI w/o Align only starts to produce results
exceptionally well, predicting high-quality results that closely preliminarily consistent with the geometry and topology of
match the ground truths (GT) from the partial inputs. This the final output at step 20, achieving a result similar to the
indicates that our approach enables detailed user interaction final output at step 0 only by step 10. Comparing the results
with the model throughout the design process, supporting fine- at step 0, NodeDiffI with the alignment mechanism clearly
grained iterative design. generates room nodes with better geometry and topology,
while NodeDiffI w/o Align produces room nodes close to or
C. Ablation Experiments even on the boundary, which is uncommon in the dataset.
We conduct an ablation experiment on the alignment mecha- Once the room nodes are established, the target distribu-
nism for the three core components of our method (NodeDiffI , tion of room adjacencies is easily captured by the model.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 13
Adjacency
Diff
of the ground truth distribution, and enhances the capability
500 100 50 20 10 0 of model to generate layouts that more closely resemble the
Time
ground truths.
Partitioning
Diff
w/o Align
D. Perceptual Studies
Partitioning
Diff
We conduct the perceptual studies with designers to evaluate
whether the bubble diagrams and floor plans generated by our
Fig. 10. Visual comparisons of the intermediate outputs produced at various DiffPlannerbI are comparable to those created by professional
diffusion steps by our three core components (NodeDiffI , AdjacencyDiff, & designers. We invite three professional designers, all experi-
PartitioningDiff) and their variants without the alignment mechanism (Align). enced in graphic design and floor plan layout planning, to
participate in the studies. Two groups of experiments are set
Therefore, both AdjacencyDiff and AdjacencyDiff w/o Align up to evaluate bubble diagrams and floor plans, respectively.
produce results consistent with the final output at step 0 by Each group includes 50 comparison tasks where each pair
step 100. However, compared to AdjacencyDiff w/o Align, of bubble diagrams (or floor plans) consists of one from the
AdjacencyDiff with the alignment mechanism is still faster, ground truths (GT) and one generated by our DiffPlannerbI with
achieving a result very close to that at step 0 by step 150. only boundary input. The participants are not aware of the
For the prediction of room partitioning, PartitioningDiff origin of the bubble diagrams or floor plans (being from GT
with the alignment mechanism has essentially moved all room or DiffPlannerbI ), and are asked to complete the forced-choice
boxes inside the boundary by step 50. Although at this time comparison task by rating each pair as “GT better/DiffPlannerbI
the rooms are not well-aligned and the shapes are somewhat better/equally good”. In total, we obtain 150 answers (50 pairs
irregular, it further refines the results, eventually producing × 3 participants) for each group (bubble diagrams and floor
reasonably distributed room boxes at step 0. In contrast, plans).
PartitioningDiff w/o Align has still not completely moved all We further compute Fleiss' Kappa [7], a statistical measure
room boxes inside the boundary by step 20, and by step 0 it of inter-rater agreement, obtaining 0.2 for floor plans and 0.1
produces some impractical designs, such as the balcony that for bubble diagrams, indicating slight agreement among raters.
extends beyond the boundary and does not fully occupy the This result is expected, as layout evaluation is inherently sub-
corners (depicted in yellow located at the bottom right corner), jective, and our studies allow users to make holistic decisions
as shown in Figure 10. without strict evaluation criteria. Interestingly, agreement is
2) Quantitative comparison: We further conduct quantita- higher for floor plans than for bubble diagrams, which aligns
tive comparisons of the intermediate outputs produced at vari- with expectations. Designers rely on well-established evalua-
ous diffusion steps by our three core components (NodeDiffI , tion criteria for floor plan layouts, such as spatial efficiency,
AdjacencyDiff, & PartitioningDiff) and their variants without functionality, and circulation, whereas bubble diagrams lack
the alignment mechanism (NodeDiffI w/o Align, Adjacency- standardized assessment frameworks.
Diff w/o Align, & PartitioningDiff w/o Align). We use the Despite individual user preferences, our perceptual studies
Fréchet Inception Distance (FID) as the evaluation metric be- are highly successful. Figure 11 shows the proportion of votes
cause it assesses the similarity in overall distribution between for the results generated by our DiffPlannerbI and from the GT
the generated results and real data. in the perceptual studies. The voting results for both groups
As shown in Table VIII, the FID scores for all three indicate that the results generated by our DiffPlannerbI are
components decreased at various steps when the alignment nearly indistinguishable from the GT, demonstrating that our
mechanism was included, indicating an improvement in per- method can produce high-quality, realistic bubble diagrams
formance. The results clearly demonstrate that the alignment and floor plans comparable to those designed by professional
mechanism enables faster and more accurate approximation designers.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 14
42.7 % 42 % 40 % 40.7 %
15.3 % 19.3 %
Fig. 11. The result of perceptual studies. The results indicate that the bubble
diagrams and floor plans generated by our method (DiffPlannerbI ) are nearly
indistinguishable from the ground truths (GT).
E. Discussion
1) New data representation: Previous methods [4], [8],
[11], [27], [28], [30], [32] often rasterize vector boundaries and
entrances into sets of binary raster masks, then utilize image-
based networks to extract features. In contrast, our method
feeds the vector boundary and entrance directly into our model,
as data tensors and not as images, avoiding the redundant step
of rasterizing them to extract their embeddings. Additionally,
Fig. 12. Qualitative comparison between the results generated by our
we also represent the prediction targets as data tensors, as DiffPlannerbI based on input samples from the test dataset (a) and their top
illustrated in Figure 2(b). This new representation for both four retrieved nearest neighbors in the training dataset (b).
input and output allows our method to operate entirely in the
vector space, eliminating the need for rasterization in bubble
diagram and floor plan generation tasks. identical solution to the closest neighbor, our model learns
In our approach, the room categories are mapped to a con- meaningful design patterns from the training data, enabling
tinuous range of [−1, 1] and represented as a one-dimensional it to generate reasonable and diverse results. For example, in
tensor. This choice is primarily made to maintain a unified the second row of Figure 12, our DiffPlannerbI , like its nearest
input format for diffusion models training and to mitigate neighbors, places a balcony on the protruding lower-left edge
the high-dimensional sparsity of one-hot encoding. Addition- of the input boundary. However, for other areas, our method
ally, this normalization stabilizes gradient optimization and explores alternative design choices within the diverse design
facilitates smoother training. However, we acknowledge that space instead of replicating the nearest neighbors. Similarly,
converting categorical variables into continuous values may in the fifth row of Figure 12, our method, like its nearest
introduce implicit numerical relationships that do not naturally neighbors, places bedrooms in the upper-left and lower-left
exist between discrete categories. Alternative approaches, such corners, away from the entrance. However, the placement
as discrete diffusion models [2] and hybrid representations, of other rooms differs significantly from that of its nearest
could be explored in future work to better preserve the discrete neighbors in the training dataset, further demonstrating that
semantics while maintaining the efficiency of the generative our model does not overfit but instead generalizes well to new
process. designs.
A natural concern is that data with a tensor-based represen- We also conduct a quantitative comparison of the differences
tation might be too easy to fit, potentially leading to overfitting. between the result generated by our DiffPlannerbI (Our) and its
To address this, we conduct both qualitative and quantitative first nearest neighbor (NN) in the training dataset on the areas
evaluations to demonstrate that our model does not suffer from of six room categories (R) for each sample (s) in the test
this risk. For each sample in the test dataset, we first use dataset (S), using the following metric similar to CoverageGTavg
DiffPlannerbI to generate a corresponding floor plan (shown in in (9):
Figure 12(a)) only from the input boundary. Then, we apply
the retrieval algorithm from Graph2Plan [11] to find a set CoverageN N
avg = [Es∈S IoUs,r (Our, N N )]r∈R . (10)
of the nearest neighbors from the training dataset based on
the similarity of input boundary (shown in Figure 12(b), with The CoverageN N
avg ranges from 0 to 1, where lower scores
similarity decreasing from left to right). indicate less similarity. As shown in Table IX, our DiffPlannerbI
As illustrated in Figure 12, the results generated by our achieves consistently low CoverageN N
avg scores across all six
DiffPlannerbI differ significantly from their nearest neighbors room categories. This indicates that the generated results by
in the training dataset. This suggests that our method does our method differ significantly from their nearest neighbors in
not suffer from overfitting; rather than simply memorizing the training dataset, further demonstrating that our method is
the input boundary-to-floor plan mapping and outputting an robust and does not suffer from overfitting.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 15
with existing architectural heuristics, its real-world relevance [7] J. L. Fleiss. Measuring nominal scale agreement among many raters.
depends on its integration into design workflows. Conducting Psychological Bulletin, 76(5):378–382, 1971.
[8] F. He, Y. Huang, and H. Wang. iPLAN: Interactive and procedural layout
user studies with professional designers to assess its effective- planning. In Proceedings of IEEE Conference on Computer Vision and
ness in real applications is an important direction for future Pattern Recognition (CVPR), pages 7793–7802, 2022.
research. Also, while the bubble diagram serves as an intuitive [9] M. Heusel, H. Ramsauer, T. Unterthiner, B. Nessler, and S. Hochreiter.
GANs trained by a two time-scale update rule converge to a local nash
and effective intermediate representation, we acknowledge equilibrium. In Advances in Neural Information Processing Systems,
that other representations may be better suited for certain pages 6629–6640, 2017.
application scenarios. Investigating alternative representations [10] J. Ho, A. Jain, and P. Abbeel. Denoising diffusion probabilistic models.
In Advances in Neural Information Processing Systems, pages 6840–
for improved model controllability and generation quality 6851, 2020.
remains an interesting avenue for further exploration. [11] R. Hu, Z. Huang, Y. Tang, O. van Kaick, H. Zhang, and H. Huang.
Graph2Plan: Learning floorplan generation from layout graphs. ACM
Transactions on Graphics, 39(4):118:1–118:14, 2020.
VI. C ONCLUSION [12] S. Hu, W. Wu, R. Su, W. Hou, L. Zheng, and B. Xu. Raster-to-
In this work, we addressed the challenges of boundary- Graph: Floorplan recognition via autoregressive graph prediction with
an attention Transformer. Computer Graphics Forum, 43(2):1–14, 2024.
constrained floor plan generation by proposing DiffPlanner, a [13] X. Huo, C. Zhou, Y. Xu, and M. Li. A methodology for balancing the
novel deep learning framework that operates entirely in vector preservation of area, shape, and topological properties in polygon-to-
space. By leveraging the diffusion models and a Transformer- raster conversion. Cartography and Geographic Information Science,
49(2):115–133, 2022.
based noise predictor, DiffPlanner effectively handles com- [14] P. Isola, J.-Y. Zhu, T. Zhou, and A. A. Efros. Image-to-image trans-
plex vector data and produces high-quality floor plans. Our lation with conditional adversarial networks. In Proceedings of IEEE
approach eliminates the need for redundant rasterization pro- Conference on Computer Vision and Pattern Recognition (CVPR), pages
1125–1134, 2017.
cesses, avoiding common issues such as information loss and [15] D. P. Kingma and J. L. Ba. Adam: A method for stochastic optimization.
distortion. DiffPlanner integrates a conditioning mechanism arXiv preprint arXiv:1412.6980, 2014.
for controlled floor plan generation and an alignment mecha- [16] H. W. Kuhn. The hungarian method for the assignment problem. Naval
Research Logistics Quarterly, 2(1-2):83–97, 1955.
nism during training, aligning the optimization of model with [17] C. Liu, J. Wu, P. Kohli, and Y. Furukawa. Raster-To-Vector: Revisiting
the iterative design processes of designers. This ensures that floorplan transformation. In International Conference on Computer
the model can produce detailed vector layouts from various Vision (ICCV), pages 2214–2222, 2017.
[18] I. Loshchilov and F. Hutter. Decoupled weight decay regularization.
user inputs, and supports various stages of the design process, In International Conference on Learning Representations, pages 1–10,
from early-stage rough designs to detailed floor plans. 2019.
Our extensive experiments, including quantitative com- [19] P. Merrell, E. Schkufza, and V. Koltun. Computer-generated residential
building layouts. ACM Transactions on Graphics, 29(6):181:1–181:12,
parisons, qualitative evaluations, ablation experiments, and 2010.
perceptual studies, demonstrate that DiffPlanner outperforms [20] J. Michalek, R. Choudhary, and P. Papalambros. Architectural layout
existing state-of-the-art methods in generating both bubble di- design optimization. Engineering Optimization, 34(5):461–484, 2002.
[21] N. Nauata, K.-H. Chang, C.-Y. Cheng, G. Mori, and Y. Furukawa.
agrams and floor plans. It offers enhanced user controllability, House-GAN: Relational generative adversarial networks for graph-
supporting fully automatic, coarsely controllable, as well as constrained house layout generation. In Proceedings of European
finely controllable generation modes, thus providing users with Conference on Computer Vision (ECCV), pages 162–177, 2020.
[22] N. Nauata, S. Hosseini, K.-H. Chang, H. Chu, C.-Y. Cheng, and
richer interactions and higher-quality results. Y. Furukawa. House-GAN++: Generative adversarial layout refinement
Overall, our contributions highlight the importance of di- network towards intelligent computational agent for professional archi-
rectly learning from vector data and the potential of diffusion tects. In Proceedings of IEEE Conference on Computer Vision and
Pattern Recognition (CVPR), pages 13632–13641, 2021.
models in architectural design and floor plan generation. [23] R. Rengel. The interior plan: Concepts and exercises. Fairchild
DiffPlanner sets a new standard for boundary-constrained and Books/Bloomsbury Publishing Incorporated, 2016.
boundary-unconstrained floor plan generation tasks, show- [24] E. Rodrigues, A. R. Gaspar, and A. Gomes. An approach to the multi-
level space allocation problem in architecture using a hybrid evolutionary
casing the effectiveness of a vector-to-vector approach in technique. Automation in Construction, 35:482–498, 2013.
generating realistic and precise floor plans. [25] M. A. Shabani, S. Hosseini, and Y. Furukawa. HouseDiffusion: Vector
floorplan generation via a diffusion model with discrete and continuous
R EFERENCES denoising. In Proceedings of IEEE Conference on Computer Vision and
Pattern Recognition (CVPR), pages 5466–5475, 2023.
[1] S. A. Arvin and D. H. House. Modeling architectural design objectives [26] B. Shneiderman. Human-centered artificial intelligence: Reliable, safe
in physically based space planning. Automation in Construction, & trustworthy. International Journal of Human–Computer Interaction,
11(2):213–225, 2002. 36(6):495–504, 2020.
[2] J. Austin, D. D. Johnson, J. Ho, D. Tarlow, and R. van den Berg. [27] J. Sun, W. Wu, L. Liu, W. Min, G. Zhang, and L. Zheng. WallPlan:
Structured denoising diffusion models in discrete state-spaces. In Synthesizing floorplans by learning to generate wall graphs. ACM
Advances in Neural Information Processing Systems, pages 17981– Transactions on Graphics, 41(4):92:1–92:14, 2022.
17993, 2021. [28] J. Sun, L. Zheng, G. Zhang, and W. Wu. BubbleFormer: Bubble diagram
[3] F. Bao, D.-M. Yan, N. J. Mitra, and P. Wonka. Generating and exploring generation via dual Transformer models. Computer Graphics Forum,
good building layouts. ACM Transactions on Graphics, 32(4):122:1– 42(7):1–13, 2023.
122:10, 2013. [29] A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez,
[4] S. Chaillou. ArchiGAN: Artificial intelligence x architecture. Architec- Ł. Kaiser, and I. Polosukhin. Attention is all you need. In Advances in
tural Intelligence, pages 117–127, 2020. Neural Information Processing Systems, pages 6000–6010, 2017.
[5] R. G. Congalton. Exploring and evaluating the consequences of vector- [30] S. Wang, W. Zeng, X. Chen, Y. Ye, Y. Qiao, and C.-W. Fu. ActFloor-
to-raster and raster-to-vector conversion. Photogrammetric Engineering GAN: Activity-guided adversarial networks for human-centric floorplan
and Remote Sensing, 63(4):425–434, 1997. design. IEEE Transactions on Visualization and Computer Graphics,
[6] P. Dhariwal and A. Nichol. Diffusion models beat GANs on image 29(3):1610–1624, 2023.
synthesis. In Advances in Neural Information Processing Systems, pages [31] W. Wu, L. Fan, L. Liu, and P. Wonka. MIQP-based layout design for
8780–8794, 2021. building interiors. Computer Graphics Forum, 37(2):511–521, 2018.
IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS 17
[32] W. Wu, X.-M. Fu, R. Tang, Y. Wang, Y.-H. Qi, and L. Liu. Data-driven
interior plan generation for residential buildings. ACM Transactions on
Graphics, 38(6):234:1–234:12, 2019.