用扩散模型加强化学习,生成更逼真的材质图像。
RealMat: Realistic Materials with Diffusion and Reinforcement Learning
- 基于SDXL微调并引入真实材质光照数据
- 强化学习提升生成材质的视觉真实感
- 适合3D内容创作与材质生成研究者
高质量材质生成对降低3D内容创作门槛至关重要。现有方法多依赖合成数据,虽便于精确标注,但与真实材质存在显著视觉差异。少数工作使用少量真实闪光照片以保证真实感,但数据规模和多样性受限。为此,我们提出RealMat,一种基于扩散模型的材质生成方法,融合文本到图像模型与自然光照下的真实材质图像数据集。首先,用2×2网格排列的合成材质图对预训练的Stable Diffusion XL(SDXL)进行微调,使其继承部分真实感并学习合成材质分布。然而仍存在视觉不真实问题。为此,我们进一步通过强化学习微调模型,设计一个针对自然光照下材质图像的真实感奖励函数,基于大规模真实材质图像数据集构建。实验表明,该方法在生成材质的真实性上优于基线模型及现有相关工作。
原文摘要 · Abstract (English)
Generative models for high-quality materials are particularly desirable to make 3D content authoring more accessible. However, the majority of material generation methods are trained on synthetic data. Synthetic data provides precise supervision for material maps, which is convenient but also tends to create a significant visual gap with real-world materials. Alternatively, recent work used a small dataset of real flash photographs to guarantee realism, however such data is limited in scale and diversity. To address these limitations, we propose RealMat, a diffusion-based material generator that leverages realistic priors, including a text-to-image model and a dataset of realistic material photos under natural lighting. In RealMat, we first finetune a pretrained Stable Diffusion XL (SDXL) with synthetic material maps arranged in $2 \times 2$ grids. This way, our model inherits some realism of SDXL while learning the data distribution of the synthetic material grids. Still, this creates a realism gap, with some generated materials appearing synthetic. We propose to further finetune our model through reinforcement learning (RL), encouraging the generation of realistic materials. We develop a realism reward function for any material image under natural lighting, by collecting a large-scale dataset of realistic material images. We show that this approach increases generated materials' realism compared to our base model and related work.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。