July 15, 2026
Recently, a line of works can generate impressive 3D objects from a single image, but they are limited by restricted representation resolution, making them unsuitable for 3D scene generation. In this work, we introduce HIVE-3D, a novel method for high-quality 3D scene generation based on hierarchical voxel enhancement framework. Specifically, given a single scene image as input, we first produce a coarse initial scene, then introduce image segmentation and attention-based retrieval to align 2D image components with 3D scene components. Subsequently, we organize these scene relations into a hierarchical component tree, where nodes closer to the leaves denote finer-grained components. Finally, we propose a voxel super-resolution model that generates refined voxels for the target instance while maintaining strong consistency with the coarse voxels. Equipped with this model, we perform coarse-to-fine hierarchical super-resolution on images and voxels for each component, producing a high-resolution and high-quality 3D scene. Extensive experiments demonstrate that our method significantly outperforms previous approaches, achieving state-of-the-art performance.
Rapid generation of a 3D scene from a single image is strongly desired across games, film, and industry. However, current scene generation methods [1]–[5] still struggle with complex layouts and diverse, personalized objects, leaving high‑quality 3D scene synthesis an open challenge.
Retrieval-based methods [1], [6], [7] are a classic solution for 3D scene generation. Such methods aim to build a comprehensive CAD template library and treat generation as retrieving the best-matching objects for an input RGB image and assembling them into a spatial layout. However, due to limited template diversity, these methods fail to align precisely with object shapes in the input image and cannot produce textured scenes. To obtain more faithful 3D scene reconstructions, subsequent research [3], [8], [9] leverages neural networks to learn end-to-end mappings from a single input image to complete 3D scenes. These models jointly infer geometry, layout, and texture, enabling holistic scene generation with improved accuracy and robustness. However, generating the entire scene in a single inference inherently limits spatial resolution and detail fidelity. In addition, the scarcity of large-scale, high-quality 3D scene data for training limits the generalization.
To address this issue, an effective approach is to decompose the 3D scene into multiple single-object generations and then assemble them into a coherent layout, which has given rise to a series of compositional methods [4], [10]–[12]. These works can produce more detailed 3D scenes, and thanks to the generalization of single-object 3D generative models, the range of applicable 3D scenes has been further expanded. However, these methods often perform a single pass of generation and assembly for each component in a scene, without considering that different components require varying levels of detail and refinement. Building on these observations, we propose a novel hierarchical voxel enhancement framework for 3D scene generation, which progressively enhances the geometric resolution of each local component in a coarse-to-fine scheme and produces high-fidelity 3D scenes.
First, given an input scene image, we first employ TRELLIS[13] to produce an initial scene. Constrained by the resolution of the TRELLIS model, this initial result is relatively coarse, yet it retains globally coherent geometry and layout information. Subsequently, we aim to dissect the target scene hierarchically and construct a hierarchical scene tree. However, performing instance-level segmentation directly on the 3D initial scene remains nontrivial. To address this issue, we leverage an image-based vision model to perform hierarchical partitioning on the image, and we introduce a 2D-to-3D matching strategy to lift the hierarchical segmentation into the 3D scene space, yielding component images and their corresponding local scene components. Furthermore, we aim to refine coarse scene components while preserving feature consistency. A naive strategy applies an image super resolution model and feeds the enhanced image into the TRELLIS model for higher resolution representations, but the stochasticity of generative models induces deviations from the coarse input, undermining hierarchical generation. To this end, we propose the voxel super resolution model, which takes the coarse voxel and the fine image as conditions to generate a refined voxel with higher resolution and richer geometric detail. Finally, we perform a top-down hierarchical enhancement of geometric fidelity for components in the scene tree, introduce a shape-based strategy to align each regenerated component with its coarse counterpart, and register it back into the original scene, yielding the final high-resolution 3D scene.
Overall, the main contributions of this work can be summarized as follows.
We propose HIVE-3D, a novel hierarchical pipeline that progressively generates high-resolution 3D scenes from a single RGB image.
We introduce a 2D-to-3D matching strategy that lifts hierarchical 2D image segmentation into 3D scene space, producing a hierarchical 3D scene tree.
We propose a voxel super-resolution model that enables increasing the resolution of target 3D components while maintaining consistent features with the coarse voxels clues.
A distinct paradigm in 3D scene generation leverages retrieval from extensive databases of CAD models [1], [6], [7], [14]–[16]. Instead of synthesizing geometry from scratch, this approach composes scenes by identifying and assembling pre-existing assets. While this guarantees high-fidelity components, it introduces fundamental challenges in expressiveness and composition. The expressiveness is bounded by the diversity of the underlying library, hindering generalization to novel scenes. On the other hand, the compositional challenge stems from the mismatch between CAD templates and real objects, making a globally coherent spatial arrangement a difficult combinatorial problem.
A significant category of 3D scene reconstruction methods relies on end-to-end training with 3D supervision [2], [3], [8], [9], [13], [17]–[22]. These approaches typically employ an encoder-decoder architecture to directly regress a complete 3D scene representation, such as a voxel grid with geometry and instance labels, from a single input image. The primary advantage of this holistic approach is that by jointly predicting the scene layout and its contained objects, the relative object poses are intrinsically correct. However, these methods exhibit low output resolution at the component level and suffer poor generalization to out-of-distribution scenes, as large-scale 3D scene datasets are hard to obtain for training.
A prominent direction in 3D scene generation is the compositional approach [4], [10], [23], [24], which leverages powerful pre-trained models to reconstruct scenes in a modular, multi-stage pipeline. These methods typically first decompose a scene from an input image [4], [5], [12], then employ specialized models for tasks like object completion, per-object 3D generation, and finally optimize the spatial layout to reassemble the scene. This modularity enhances generalization by capitalizing on large-scale generative priors. In contrast to holistic generation approaches [9], [13], [22], compositional generation methods excel at producing high-quality object instances. However, they often struggle to preserve global scene context, making spatial alignment a significant challenge. Our approach bridges this gap by first generating the scene holistically to establish a coherent layout, and then hierarchically refining each instance while preserving this overall structure. This results in a scene that is both high-resolution and globally consistent.
With the rise of large-scale, pre-trained generative models [11], [13], [22], [25]–[28], parameter-efficient fine-tuning (PEFT) has emerged as a crucial paradigm for adapting these foundational models to new tasks without incurring the prohibitive costs of full fine-tuning. A prominent strategy in this domain is to introduce small, trainable modules while keeping the original model weights frozen. Prominent PEFT strategies adapt frozen models by introducing small, trainable modules, such as ControlNet [29], which adds a trainable copy of network blocks to process spatial conditions, and LoRA [30], which injects low-rank matrices into existing layers based on the hypothesis that weight updates have a low intrinsic rank.
Building on similar principles, IP-Adapter [31] focuses on enabling image-prompting capabilities. It introduces a lightweight adapter module that uses a cross-attention mechanism to inject visual features, extracted by a separate image encoder, into a frozen text-to-image diffusion model [32], achieving high-fidelity image-conditioned generation in a decoupled manner. Inspired by IP-Adapter, we implement our voxel super-resolution model by introducing a similar fine-tuning strategy to the flow transformer of TRELLIS. Specifically, we treat the coarse voxels as an additional condition to guide the generation of refined voxels, thereby achieving high-resolution output while maintaining structural consistency.
Our approach builds on a state-of-the-art 3D generative model TRELLIS[13], where the target is represented as structured latent voxels. For a 3D asset \(\mathcal{O}\), TRELLIS encode its geometry and appearance information using a unified structured latent representation \(\boldsymbol{z}\), as: \[\boldsymbol{z}=\{(\boldsymbol{z}_i,\boldsymbol{p}_i)\!\}_{i=1}^L,\:\boldsymbol{z}_i\in\mathbb{R}^C,\boldsymbol{p}_i\in\{0,1,\ldots,N-1\}^3, \label{eq:energy}\tag{1}\] where \(\boldsymbol{p}_i\) is the positional index of an active voxel in the 3D grid, \(\boldsymbol{z}_i\) denotes a local latent attached to the corresponding voxel \(\boldsymbol{p}_i\). L is the total number of active voxel grid, and N is the spatial length of the voxel grid. Collectively, these structured latents fully describe the surface of \(\boldsymbol{O}\), capturing its global structure and fine-grained details in a unified representation.
TRELLIS generates these structured latents using a two-stage pipeline. In the first stage, the model uses a sparse structure flow transformer model \(\mathcal{G}_{\mathrm{S}}\) to generate a low-resolution feature grid \(\boldsymbol{S}\in\mathbb{R}^{D\times D\times D\times{C_\mathrm{S}}}\), followed by a latent feature decoder \(\mathcal{D}_{\mathrm{S}}\) to obtain a dense binary voxel grid \(\boldsymbol{O}\in \begin{Bmatrix} 0,1 \end{Bmatrix}^{N\times N\times N}\). This dense grid \(\boldsymbol{O}\) is then processed into a set of sparse voxel coordinates \(\{\boldsymbol{p}_i\}_{i=1}^L\). In the second stage, TRELLIS employs a subsequent structure latent flow transformer \(\mathcal{G}_{\mathrm{L}}\) to generate a set of structural features, \(\{\boldsymbol{z}_i\}_{i=1}^L\) , corresponding to each of the sparse voxel coordinates from the previous stage. The structured latent \(\boldsymbol{z}\) is then processed through specialized decoders(\(\mathcal{D}_{\mathrm{NeRF}}\),\(\mathcal{D}_{\mathrm{Mesh}}\), or \(\mathcal{D}_{\mathrm{GS}}\)) to generate the final 3D asset \(\mathcal{O}\) in various formats(NeRF, meshes, or 3DGS).
In this paper, we propose HIVE-3D, a novel pipeline aims to generates high-resolution 3D scenes from a single RGB image. As shown in Figure 1, Our approach decomposes the hierarchical structure of target scene and constructs a scene tree \(\mathcal{T}_d\) with depth \(d\), where each node denotes a scene component that includes its corresponding voxel representation and cropped image. We denote these components and images as \(\{\mathcal{V}_k^d\}_{k=1}^K\) and \(\{\mathcal{I}_k^d\}_{k=1}^K\), indicating that there are components at level \(d\). Nodes with larger depth \(d\) focus on smaller local regions and exhibit finer geometry and texture. In particular, to obtain hierarchical scene components and their corresponding images, we propose a 2D-to-3D scene tree construction method as shown in Section 4.1. On the other hand, to preserve cross-level component consistency, we propose a voxel super-resolution model in Section 4.2 that refines the target instance while remaining consistent with the coarse voxels. Equipped with this model, we introduce the coarse-to-fine hierarchical scene generation as described in Section 4.3, which aims to progressively refine the target component and assemble it back into the scene, producing a high-resolution and high-quality 3D scene.
We initialize the scene tree \(\mathcal{T}_d\) by generating the root node from the input RGB image \(\mathcal{I}_{input}\) using TRELLIS [13], which yields initial scenes with coherent spatial structure.
Further, the central focus of our subsequent work is to preserve this plausible spatial arrangement while refining the geometry and texture of each individual component and instance. However, performing instance-level segmentation directly on the voxels of the initial scene is a non-trivial task. Fortunately, inspired by Fuse3D [33], which observed that the flow transformer-based model \(\mathcal{G}_{\mathrm{L}}\) within the TRELLIS framework learns to establish correspondences between image regions and 3D voxels during its training process. This key insight enables us to recast the challenging task of 3D voxel segmentation as a more tractable problem of performing instance segmentation in 2D, where mature and powerful models are readily available.
Specifically, for any scene component image \(\mathcal{I}_{k}^d\), we employ the object detection model Florence-2 [34] to produce axis-aligned bounding boxes for each detected instance. Subsequently, these bounding boxes are used as prompts for the Segment Anything 2 (SAM 2) [35] model to perform instance segmentation on the \(\mathcal{I}^{d}_{k}\). Benefiting from the support of SAM2 for hierarchical segmentation, this process yields a precise mask \(\{\mathcal{M}_k^d\}_{k=1}^K\) and the corresponding cropped image \(\{\mathcal{I}_k^d\}_{k=1}^K\) for each component across levels of scene tree \(\mathcal{T}_d\), where K denotes the total number of detected instances at level \(d\).
On the other hand, in generating initial active voxels for a component scene in \(\mathcal{G}_{\mathrm{L}}\) condition on image \(\mathcal{I}_{k}^d\), image tokens are injected into cross-attention layers as keys and values, while the active voxels serve as queries. This produces a cross-attention map that is normalized along the image token axis using a softmax function. Consequently, the attention score attached to each voxel represents its degree of alignment with the corresponding image tokens, and a larger score indicates stronger correspondence between them. Next, we aggregate the attention scores of image tokens covered by the instance mask region \(\mathcal{M}_k^d\). By applying a threshold to these aggregated scores, we select a set of voxel indices denoted as \(\mathcal{V}_k^d\), which correspond to the 3D instance aligned with the image region \(\mathcal{I}_k^d\). This process is repeated independently for each instance. By applying these indices to the global structure latents, we obtain a set of instance-specific structure latents \(\{(\boldsymbol{z}_i,\boldsymbol{p}_i)|i\in\mathcal{V}_{k}^d\}_{k=1}^{K}\). The structure latents corresponding to each instance can be further independently decoded into a 3D Gaussian Splatting (3DGS) representation of the object instance, denoted as \(\mathcal{O}_k\), using the TRELLIS structure latents decoder \(\mathcal{D}_{\mathrm{GS}}\).
Finally, we construct a hierarchical scene tree for the input image, with nodes corresponding to scene components at different granularities along with the image description. Yet the current tree contains only initial scene voxels, which are relatively coarse. Enhancing their resolution while preserving global consistency remains challenging.
In this section, we introduce a method to enhance the resolution and quality of coarse scene component while preserving consistent features. A straightforward approach is to apply an image super resolution model HYPIR [36] to the image corresponding to the coarse component, then feed the enhanced image back into the TRELLIS model to generate higher resolution representations. However, due to the inherent stochasticity of generative models, the regenerated components often deviate from the coarse input, and this inconsistency impedes hierarchical generation.
To address this issue, we propose the voxel super-resolution model, which introduces the coarse voxel as a new condition within the original generative model \(\mathcal{G}_{\mathrm{S}}\) from TRELLIS to generate a refined voxel with higher resolution and richer geometric detail. However, the pretrained model \(\mathcal{G}_{\mathrm{S}}\) is optimized for image conditioning, which makes it nontrivial to enable it to accept an additional coarse voxel conditioning with fundamentally different features. Inspired by IP-Adapter [31], we aim to train an adapter network that enables it to accept encoded coarse voxels as a condition and inject the projected conditional features into the DiT-based pretrained model \(\mathcal{G}_{\mathrm{S}}\) via attention. As shown in Figure 2, we use the pretrained sparse-structure VAE encoder from the TRELLIS model to encode coarse voxels into a latent feature, aligning the voxel condition with the latent space in which the DiT conducts diffusion inference. Further, we add a trainable linear layer as a projection module to map the new geometric condition, aligning it with the original image condition features. Finally, we augment the original DiT block with a dedicated trainable cross-attention layer to inject the new geometric condition feature. Notably, the geometric condition feature and the original image condition jointly control the DiT block. Details on the attention computation can be found in the supplementary material.
During training, we keep the parameters of the \(\mathcal{G}_{\mathrm{S}}\) frozen and exclusively train the newly added cross-attention layers and the projection module. We adopt the same training objective as the \(\mathcal{G}_{\mathrm{S}}\): \[\mathcal{L}_{CFM}(\theta)=\mathbb{E}_{t,\boldsymbol{x}_0,\boldsymbol{\epsilon}}\|\boldsymbol{v}_\theta(\boldsymbol{x},t)-(\boldsymbol{\epsilon}-\boldsymbol{x}_0)\|_2^2 ,\]
where \(\boldsymbol{x}_0\) represents the data samples, \(\epsilon\) denotes noise samples drawn from a prior distribution (typically a standard normal distribution), and \(t\) is the timestep.
By modeling the 3D scene with a hierarchical representation and equipping it with a voxel super-resolution module, we can then perform top-down coarse-to-fine scene generation. Specifically, the root node of the scene tree represents the entire initial scene. We then hierarchically partition higher-level nodes into several child components to form child nodes. Using the coarse voxels from the parent node as a condition, we generate finer-grained child components. The rationale and necessity behind the recursive generation strategy are further elaborated in Section 9.2. In particular, the segmented child components may be occluded, and we adopt the Amodal3R [37] framework to generate complete amodal reconstructions free from occlusion. The above procedure is applied recursively to each node until the desired level of detail is reached.
Nevertheless, due to the structural priors learned from large-scale 3D datasets, the voxel super-resolution model inherently reconstructs these refined components in a canonical reference frame, resulting in a normalized scale and a standardized orientation. Consequently, these high-resolution components exhibit substantial discrepancies in scale and pose relative to the initial scene. To this end, we must align the scale and pose of the finer-grained components with their counterparts in the initial scene, involving the scaling factor \(\boldsymbol{s}\) and a rigid transformation with rotation \(\mathcal{R}\) and translation \(\boldsymbol{t}\).
For the estimation of the scaling factor \(\boldsymbol{s}\), we devise a shape-based strategy. Specifically, we uniformly sample points from the surface of the Gaussian splatting representation and compute the average distance of these points to the centroid of the object. This average distance serves as a robust proxy for the overall scale of the object, which we then use to align the scales of different objects. More details are provided in the supplementary material. On the other hand, since all components are represented by 3D Gaussian Splatting 3DGS and each Gaussian primitive can be treated as a point augmented with rich attributes, we leverage point cloud registration to estimate the rigid transformation with rotation \(\mathcal{R}\) and translation \(\boldsymbol{t}\). A critical challenge is the severe imbalance in the number of Gaussians between the finer-grained child components and its low-resolution counterpart, which induces substantial mismatches and outliers. To address this, we adopt a registration strategy that emphasizes robustness to outliers rather than strict point-to-point fidelity. Specifically, we employ RANSAC [38], which is resilient to spurious correspondences, instead of methods such as ICP [39] that rely on accurate nearest-neighbor matching. The comparison of the two methods is shown in Figure 6. The details of our RANSAC pipeline are provided in the supplementary material.
Our voxel super-resolution model is built upon the flow transformer \(\mathcal{G}_{\mathrm{S}}\) of TRELLIS. It is designed to be conditioned on two distinct inputs to control the voxel generation process: an image of 518x518 resolution and a coarse voxel grid. For the image condition, similar to TRELLIS [13], we utilize a DINOv2 [40] model to extract visual features. For the coarse voxel condition, we employ the sparse VAE encoder to project it into latent space.
Inspired by IP-Adapter [31], we propose a training strategy for our voxel super-resolution model. We freeze the original parameters of \(\mathcal{G}_{\mathrm{S}}\) and integrate a new cross-attention layer into each block. These new layers match the dimensions of the original ones and are initialized with the weights of their corresponding pre-trained counterparts. During training, we exclusively update the parameters of the new projection module and cross-attention layers. The model is trained with 4 NVIDIA 4090 GPUs.
We trained our voxel super-resolution model on a curated subset of 10,000 3D assets from Objaverse-XL [41]. For each asset, we first rendered 24 multi-view images and then downsampled them to generate corresponding coarse voxels. This process yielded a final training set composed of image-voxel pairs.
We compare our method with the state-of-the-art methods in scene generation from single images, which include feed-forward generation methods TRELLIS [13], MIDI [11], SceneGen [5] and compositional generation method Gen3DSR [4]. Furthermore, a detailed comparison with PartPacker [42] is provided in Section 9.3.
We further evaluated our method using rendered images from the 3D-FRONT [43] dataset and a collection of real-world scene photographs. The qualitative results are presented in Figure 3. While existing methods demonstrate certain strengths, they often suffer from significant limitations. TRELLIS excels at generating coherent spatial layouts, but its reliance on a low-resolution voxel representation prevents the synthesis of high-resolution 3D scenes (e.g., Figure 3, third row, fourth column; and fourth row, fourth column). Gen3DSR [4] performs modestly in rendering geometric and texture details, with its outputs often plagued by object interpenetration and surface distortions as shown in Figure 3 (second row, second column and fourth row, second column). Similarly, scenes from SceneGen [5] frequently exhibit severe interpenetration or object displacement artifacts (e.g., Figure 3, third row, third column; and fifth row, third column). Akin to TRELLIS, MIDI [11] also produces scenes with plausible layouts but fails to achieve the desired quality in the geometry and texture of individual assets as shown in Figure 3 (first row, first column and second row, first column). In contrast, our method not only inherits the superior layout understanding of TRELLIS but also dramatically enhances the geometric detail and texture fidelity of in-scene instances, enabling the generation of high-resolution, high-quality 3D scenes.
We evaluate our method and all baselines on the 3D-FRONT dataset using CD [44], F-Score [45], and IoU [46] to assess the geometric fidelity of the generated scenes. As shown in Table 1, HIVE-3D demonstrates consistently strong performance across all metrics. We also report the average runtime for each method to generate one scene. In addition, we conduct complementary evaluations on both synthetic datasets and real-world images using SSIM, PSNR, LPIPS [47], and CLIP [48] to further assess the visual quality and text-image consistency of the generated results. The detailed results are provided in Appendix Table 6.
To validate the efficacy of each core component in HIVE-3D, we perform comprehensive ablation studies. Our analysis isolates the effects of four key aspects: 1) the depth of the hierarchical generation, 2) the voxel super-resolution module, 3) the scale factor estimation, and 4) the RANSAC-based point cloud registration [38]. We present the quantitative results of our ablation study, evaluated with the SSIM, PSNR, LPIPS [47], CLIP [48]. More ablation studies are provided in the supplementary material.
| Method | CD\(\downarrow\) | F-Score\(\uparrow\) | IoU\(\uparrow\) | Runtime\(\downarrow\) |
|---|---|---|---|---|
| Gen3DSR | 0.429 | 5.096 | 0.033 | 9min |
| MIDI | 0.056 | 19.83 | 0.5329 | 40s |
| SceneGen | 0.116 | 18.95 | 0.0587 | 29s |
| TRELLIS | 0.0038 | 81.29 | 0.8603 | 6.3s |
| Ours | 0.0035 | 84.34 | 0.7449 | 36.5s |
4pt
We evaluated the effect of varying the number of recursive layers by testing our generation process with depths of 1, 2, and 3. As illustrated in Figure 4, the quality and level of detail in the generated scenes progressively improve as the recursion depth increases. To quantitatively validate this observation, Table 2 presents a detailed analysis of performance metrics as a function of the generation depth.
| Method | SSIM\(\uparrow\) | PSNR\(\uparrow\) | LPIPS\(\downarrow\) | CLIP\(\uparrow\) |
|---|---|---|---|---|
| Depth=1 | 0.70 | 10.13 | 0.38 | 0.94 |
| Depth=2 | 0.72 | 11.20 | 0.41 | 0.95 |
| Depth=3 | 0.75 | 11.89 | 0.41 | 0.96 |
4pt
Unlike methods that directly generate voxels conditioned on an image, our approach utilizes coarse voxels to guide the generation process. This allows the resulting super-resolved voxels to retain key structural features from the coarse input, as demonstrated in Figure 5. The quantitative results for our voxel super-resolution model are shown in Table 3. Our model achieves the best scores across all evaluated metrics, including PSNR, SSIM, CLIP and LPIPS. This result demonstrates the superior capability of our approach in reconstructing high-fidelity details from coarse voxel inputs while maintaining structural consistency.
Failing to estimate the scale of locally generated scenes or instances from our instance generation module leads to two significant issues. First, it results in noticeable scale inconsistencies across different parts of the scene. Second, and more critically, it severely impairs the subsequent point cloud registration process, as estimating a reliable rigid transformation matrix between two geometries with large scale discrepancies is often infeasible. Figure 6 visually demonstrates the scale artifacts, while Table 4 quantitatively reports the severe drop in registration performance.
| Method | SSIM\(\uparrow\) | PSNR\(\uparrow\) | LPIPS\(\downarrow\) | CLIP\(\uparrow\) |
|---|---|---|---|---|
| w/o voxel SR model | 0.72 | 10.59 | 0.38 | 0.93 |
| w/ voxel SR model | 0.76 | 11.61 | 0.35 | 0.96 |
4pt
Estimating the transformation between two geometries that exhibit significant differences in point density and slight variations in shape presents a considerable challenge. We found that many precision-oriented registration methods, such as ICP [39], failed to produce satisfactory results under these conditions. Ultimately, we adopted the RANSAC algorithm [38], which is known for its strong robustness against outliers and noise, and successfully achieved the desired alignment. We justify this algorithmic choice with both qualitative and quantitative evidence. Figure 6 visually contrasts a successful alignment by RANSAC with a typical failure case from ICP. Concurrently, Table 4 provides a quantitative comparison, demonstrating that RANSAC significantly outperforms ICP in both success rate and registration accuracy.
| Method | SSIM\(\uparrow\) | PSNR\(\uparrow\) | LPIPS\(\downarrow\) | CLIP\(\uparrow\) |
|---|---|---|---|---|
| ICP | 0.88 | 16.08 | 0.28 | 0.66 |
| w/o scale factor estimation | 0.82 | 16.17 | 0.29 | 0.83 |
| Full Model | 0.89 | 17.00 | 0.20 | 0.97 |
4pt
In this paper, we introduced HIVE-3D, a novel framework for generating high-fidelity 3D scenes via a hierarchical voxel super-resolution approach. Taking an image and an initial coarse scene from TRELLIS as input, our method first constructs a scene tree to represent the hierarchical structure of the scene. This hierarchy then guides a coarse-to-fine synthesis process that populates the scene with detailed instances. Crucially, our approach preserves the coherent spatial layout of the initial scene while significantly enhancing it with high-resolution, high-fidelity components. Extensive experiments have demonstrated that HIVE-3D outperforms existing methods in terms of 3D scene generation quality.
Nevertheless, our method still has several limitations. First, our approach is based on TRELLIS, the resolution of each voxel generation is constrained and it struggles with large-scale scenes. Second, constructing the initial 2D hierarchical semantics relies on detection and segmentation models, which may introduce errors and lead to suboptimal results.
Future work will investigate extending our framework to multi-view and video-based 3D scene generation. In particular, we aim to explore joint cross-attention mechanisms that enable Scene Tree nodes to aggregate visual information from multiple viewpoints conditioned on camera poses. We also plan to incorporate multi-view consistency constraints into the intermediate stages of the diffusion process to further improve spatial coherence and temporal stability. We believe these extensions will enhance occlusion reasoning and facilitate the generation of dynamic 3D scenes in more complex real-world environments.
This work was partially supported by National Key R&D Program of China (No. 2025YFG0101300), NSFC (No. 52572504), Key R&D Program of Zhejiang Province (No. 2025C01064), and NSFC Corporate Joint Key Program (No. U22B2034).
This work presents a method for high-quality 3D scene generation from a single image. The proposed approach could help accelerate 3D content creation for applications such as digital entertainment, virtual reality, and interactive design.
As with other generative AI technologies, our method could potentially be misused to create misleading or synthetic digital content. However, we believe the primary impact of this work is to support creative applications and improve the accessibility of 3D content creation tools.
To perform instance-level segmentation directly on the initial scene voxels, we draw inspiration from Fuse3D[33] and first feed the complete set of tokens extracted from the image into \(\mathcal{G}_{\mathrm{L}}\) of TRELLIS[13] to derive the latent features for initial voxels \(\{\boldsymbol{p}_i\}_{i=1}^L\). We obtain the final attention map by averaging the voxel-to-token attention maps from a specific subset of heads, namely head 0, 4, and 12. This map, softmax-normalized along the image token axis, yields an alignment score for each voxel, quantifying its correspondence with the visual content of the image. We then aggregate the scores of image tokens corresponding to the masked region \(\{\mathcal{M}_k\}_{k=1}^K\). Finally, we select the set of voxels whose aggregated scores exceed a predefined threshold.
We observe that the regenerated object \(\mathcal{O}_k\) often exhibits significant discrepancies in scale and pose when compared to the original object. For scale estimation, a straightforward approach is to align the axial lengths based on the objects’ bounding boxes. However, this method proves to be infeasible. The substantial pose difference between the two objects leads to considerable variations in the shapes of their respective bounding boxes, rendering any alignment based on them fundamentally unreliable. At the same time, we note that even when the pose of the object undergoes drastic changes, its overall geometry remains largely consistent, with refinements primarily in the surface details. Therefore, we base our scale estimation on this geometric invariance. Our method involves uniformly sampling a point cloud from the generated object’s surface, computing the centroid of this point cloud, and then calculating the mean distance from all points to this centroid. This distance serves as a robust proxy for the overall size of the object. Finally, we define the scale factor \(s\) as \(d / d'\), where \(d'\) is the size of the regenerated object \(\mathcal{O}_k\) and \(d\) is the size of the original object. We then rescale the object \(\mathcal{O}_k\) by \(s\) along each axis.
In Sec. 4.3, we employ RANSAC-based global point cloud registration to robustly align the refined instance with its initial instance under geometric noise, partial shape discrepancies, and outliers.
Specifically, we use the registration_ransac_based_on_feature_matching function in Open3D [49], together with
TransformationEstimationPointToPoint without scaling, to estimate a rigid transformation in \(\mathrm{SE}(3)\).
Given the point clouds of the refined instance and the initial instance, we first downsample both point clouds using a voxel size of 0.0156, corresponding to \(1/64\) of the normalized unit space. This downsampling step reduces computational cost while preserving the overall geometric structure of each instance. Surface normals are then estimated using a hybrid search radius of 0.0312, i.e., \(2\times\) the voxel size. We further compute Fast Point Feature Histograms (FPFH) descriptors with a search radius of 0.0780, i.e., \(5\times\) the voxel size. For both normal estimation and FPFH computation, the neighborhood search is limited to a maximum of 200 nearest neighbors.
Based on the extracted FPFH descriptors, tentative feature correspondences are established between the two downsampled point clouds. We enable mutual filtering to retain correspondences that are mutually consistent in feature space, which helps suppress ambiguous or one-sided matches. During registration, the maximum correspondence distance is set to 0.0234, corresponding to \(1.5\times\) the voxel size. This threshold defines the maximum Euclidean distance allowed between a transformed source point and its target correspondence for the correspondence to be considered geometrically valid.
To further reject outlier correspondences, we use two correspondence checkers. The first is CorrespondenceCheckerBasedOnEdgeLength with a similarity threshold of 0.9, which enforces approximate preservation of pairwise
distances between sampled correspondences and rejects matches that would induce severe geometric distortion. The second is CorrespondenceCheckerBasedOnDistance, which uses the same distance threshold of 0.0234 to ensure that
the transformed source points remain sufficiently close to their target correspondences.
The RANSAC procedure is configured with a minimal sample size of 3, a maximum of 100,000 iterations, and a confidence level of 0.999. Here, the minimal sample size of 3 refers to the number of point correspondences sampled to estimate a 6-DoF rigid transformation under the point-to-point alignment objective. It should not be interpreted as the number of points required to define a geometric plane. Given three non-collinear point correspondences, a rigid transformation can be estimated by solving for the rotation and translation, e.g., via the Kabsch algorithm using singular value decomposition (SVD). RANSAC repeatedly samples such minimal correspondence sets, generates candidate rigid transformation hypotheses, and selects the transformation with the largest set of geometrically consistent inlier correspondences.
The output of this process is a rigid transformation matrix \[\mathbf{T} = \begin{bmatrix} \mathbf{R} & \boldsymbol{t} \\ \mathbf{0}^{\top} & 1 \end{bmatrix} \in \mathrm{SE}(3),\] where \(\mathbf{R} \in \mathrm{SO}(3)\) is the estimated rotation matrix and \(\boldsymbol{t} \in \mathbb{R}^{3}\) is the translation vector. The estimated transformation aligns the refined instance to the coordinate frame of the initial instance while suppressing unreliable correspondences caused by geometric noise and outliers. These parameter settings provide stable and reproducible global registration across diverse object instances.
In practice, we employ the AdamW optimizer [50] for optimizing all networks and parameters. We set the initial learning rate to \(1 \times 10^{-4}\) and the weight decay to \(1 \times 10^{-2}\). We trained our model for approximately 3 days on 4 NVIDIA RTX 4090 GPUs, with a total batch size of 24. The total number of training iterations is set to 100,000. We apply Classifier-Free Guidance (CFG) [51] training with a condition dropout rate of \(10\%\). To ensure training stability, we maintain an Exponential Moving Average (EMA) [52] of model parameters with a decay rate of 0.9999.
To construct our dataset, we curated a subset of 10,000 3D models from Objaverse-XL [41], excluding those with excessive mesh complexity. Subsequently, the obtained meshes are voxelized into a grid with a resolution of \(64^3\). We then employ the sparse structure VAE encoder from TRELLIS to encode these voxel grids into a feature volume of size \(16 \times 16 \times 16 \times 8\). Finally, following the design of IP-Adapter [31], we reshape this volume into a sequence of 16 tokens, each with a dimensionality of 2048, serving as the coarse voxel condition for the training network.
| Nodes | Seg. + SR (s) | Generation (s) |
|---|---|---|
| 1 (depth = 1) | 0.00 | 59.3 |
| 2 (depth = 2) | 8.45 | 86.6 |
| 3 (depth = 2) | 13.13 | 92.6 |
| 4 (depth = 2) | 19.35 | 98.1 |
12pt
We evaluate the computational efficiency of our proposed pipeline by analyzing the inference latency across various scene complexities. Specifically, we categorize the runtime performance according to the number of nodes in the scene tree (a single root at depth 1, or 2–4 nodes when expanding to depth 2). All benchmarks were conducted on a single NVIDIA GeForce RTX 4090 GPU (The total end-to-end execution time, including model loading and environment initialization), and the detailed results are summarized in Table 5.
We quantitatively evaluate the geometric quality of the scenes generated by our method and the baseline approaches on a synthetic dataset. The evaluation is conducted using SSIM, PSNR, LPIPS [47], CLIP [48]. As shown in Table 6, HIVE-3D demonstrates consistently strong performance across all metrics.
| Method | SSIM\(\uparrow\) | PSNR\(\uparrow\) | LPIPS\(\downarrow\) | CLIP\(\uparrow\) |
|---|---|---|---|---|
| Gen3DSR | 0.76 | 10.96 | 0.34 | 0.91 |
| MIDI | 0.80 | 12.85 | 0.25 | 0.92 |
| SceneGen | 0.79 | 12.25 | 0.31 | 0.96 |
| TRELLIS | 0.80 | 13.31 | 0.31 | 0.95 |
| Ours | 0.79 | 13.39 | 0.33 | 0.97 |
4pt
We conducted ablation studies on the synthetic dataset to evaluate geometric quality. We employed CD, F-Score, and IoU as evaluation metrics. The results are presented in Table 7. The results demonstrate that every core component contributes significantly to the overall performance of the pipeline.
| Method | CD\(\downarrow\) | F-Score\(\uparrow\) | IoU\(\uparrow\) |
|---|---|---|---|
| w/o voxel SR model | 0.0098 | 60.09 | 0.6188 |
| ICP | 0.147 | 23.92 | 0.2932 |
| w/o scale factor estimation | 0.193 | 35.44 | 0.2352 |
| Full Model | 0.0035 | 84.34 | 0.7449 |
4pt
We evaluated our method on synthetic data. We render the final images from two viewpoints: one aligned with the input perspective and the other from the opposite viewpoint. For comparison, we also include the results rendered by TRELLIS from the corresponding viewpoints. Additional results are presented in Figure 7. Please zoom in to view the fine details.
To further justify our design choices, we provide an in-depth analysis of the multi-level recursive structure. Unlike a conventional two-layer architecture—which typically transitions directly from an initial scene to fine-grained components—our approach employs a recursive refinement strategy to address the inherent resolution constraints of the TRELLIS framework.
Since TRELLIS generates scenes within a fixed \(64^3\) canonical voxel space, directly partitioning the scene into the finest components would lead to an extreme jump in spatial resolution. For instance, a small component originally occupying only \(4^3\) voxels in the initial scene would be abruptly re-represented in a \(64^3\) space. This sharp transition introduces significant geometric ambiguity and compromises the precision of spatial registration, as evidenced by the misaligned car in Figure 8. In contrast, our recursive design facilitates a more progressive transition (e.g., \(4^3 \to 16^3 \to 64^3\)), providing a stable geometric bridge that ensures more reliable alignment and superior fidelity in the final scene assembly.
We conduct qualitative comparisons between our method and PartPacker [42]. It should be noted that PartPacker primarily focuses on geometric synthesis and generates only untextured meshes. As illustrated in Figure 9, PartPacker exhibits several structural deficiencies across various scenarios. Specifically, it fails to reconstruct the correct number of table legs and struggles with plant stems in the first row. Furthermore, it suffers from object omissions and significant shape distortions in the third row, where the generated chair deviates notably from the input reference. In contrast, our HIVE-3D consistently produces scenes with both geometric precision and high-fidelity textures, underscoring the clear advantages of our hierarchical refinement strategy in complex scene synthesis.
We conduct qualitative comparisons between SAM3D [53] and our method in terms of 3D scene generation quality. Since SAM3D performs holistic scene generation, its results are constrained by the overall scene resolution. As shown in the first row of Figure 10, the generated small stool exhibits noticeable geometric inconsistencies with the reference image. In the second row, the generated red toy car is incorrectly fused with the small table, leading to severe structural artifacts. In contrast, HIVE-3D produces scenes with substantially higher local geometric fidelity and more coherent object structures.
Although VIAFormer [54] and ULTRA3D [55] are also related to structured 3D generation, our method differs substantially from these approaches in both formulation and design objectives.
VIAFormer primarily focuses on denoising and refining existing occupancy representations under a fixed and stable spatial resolution. In contrast, our proposed Voxel-SR module performs generative voxel super-resolution, progressively enhancing sparse scene nodes into higher-resolution and higher-fidelity 3D structures within a hierarchical coarse-to-fine generation framework. Therefore, our method is designed not only for refinement, but also for hierarchical spatial upsampling and detail synthesis.
Compared with ULTRA3D, which relies on specialized part-aware sensing modules for semantic decomposition and alignment, our framework adopts a lightweight 2D-to-3D semantic lifting strategy that directly reuses intrinsic cross-attention maps from the diffusion model for semantic grounding. This design avoids introducing additional part sensing networks or expensive supervision, resulting in a more integrated and computationally efficient pipeline.
Overall, these distinctions enable HIVE-3D to achieve high-fidelity scene generation while maintaining strong semantic consistency and efficient hierarchical generation capabilities.
Since our pipeline depends on the scene generated by the TRELLIS, its performance is inherently bounded by the quality of the initial output. In cases where the base model fails to generate certain objects or produces incomplete geometry, downstream components of the pipeline may struggle to function correctly. Specifically, this leads to failures in the registration stage, preventing refined components from being accurately placed in their global coordinates, as illustrated in Figure 11. We aim to address this limitation by enhancing the robustness of the initial scene parsing in future work.