RoomDiffusion: 室内设计行业的专用扩散模型
计算机视觉与模式识别
2024-09-06 v1
摘要
文本到图像扩散模型的最新进展显著改变了视觉内容生成,但其在室内设计等专门领域的应用仍有待深入探索。在本文中,我们提出了 RoomDiffusion,这是一个专为室内设计行业精心定制的开创性扩散模型。首先,我们从零开始构建了一整套数据流水线,用于更新和评估数据以进行迭代模型优化。随后,应用了多宽高比训练、多阶段微调和模型融合等技术,以增强生成结果的视觉吸引力和精度。最后,利用潜在一致性蒸馏方法,我们对模型进行蒸馏和加速以实现最佳效率。与针对通用场景优化的现有模型不同,RoomDiffusion 解决了室内设计中的特定挑战,如缺乏时尚感、家具重复率高和风格不准确。通过我们包含 20 多名专业人类评估者的整体人工评估协议,RoomDiffusion 在美学、准确性和效率方面展现出业界领先的性能,超越了所有现有的开源模型,如 stable diffusion 和 SDXL。
引用
@article{arxiv.2409.03198,
title = {RoomDiffusion: A Specialized Diffusion Model in the Interior Design Industry},
author = {Zhaowei Wang and Ying Hao and Hao Wei and Qing Xiao and Lulu Chen and Yulong Li and Yue Yang and Tianyi Li},
journal= {arXiv preprint arXiv:2409.03198},
year = {2024}
}