Nabla-R2D3 : Effective and Efficient 3D Diffusion Alignment with 2D Rewards

Under review

Anonymous Authors

Nabla-R2D3 is a highly effective and sample-efficient reinforcement learning alignment framework for 3D-native diffusion models using pure 2D rewards.

😊 Abstract ✨

Generating high-quality and photorealistic 3D assets remains a longstanding challenge in 3D vision and computer graphics. Although state-of-the-art generative models, such as diffusion models, have made significant progress in 3D generation, they often fall short of human-designed content due to limited ability to follow instructions, align with human preferences, or produce realistic textures, geometries, and physical attributes. In this paper, we introduce Nabla-R2D3, a highly effective and sample-efficient reinforcement learning alignment framework for 3D-native diffusion models using 2D rewards. Built upon the recently proposed Nabla-GFlowNet method for reward finetuning, our Nabla-R2D3 enables effective adaptation of 3D diffusion models through pure 2D reward feedback. Extensive experiments show that, unlike naive finetuning baselines which either fail to converge or suffer from overfitting, Nabla-R2D3 consistently achieves higher rewards and reduced prior forgetting within few finetuning steps.

πŸ“‹ Section-1: Results on Aesthetic Score Reward πŸ“‹

✨ Section-2: Results on HPSv2 Reward ✨

πŸŽ‹Β Β  Section-3: Results on Normal EstimatorΒ Β πŸŽ‹