ReNFT: Repairing Mode Collapse in Reward Post-Training via Internal Probability-Mass Recalibration

Yuchen Bao, Chao Wen, Haowei Wang +10

Reward post-training of diffusion generators inevitably concentrates probability mass on a few reward-favored modes, a mode collapse that erases within-prompt diversity. Existing m…