question about loss

[HanzhenLu]

Great work on simplifying the training of diffusion models!

I am fairly new to this area and I am curious about the loss computation in your code, where you use the cart weights. Could you kindly point me to any paper or reference that discusses this strategy?

[ZHZisZZ]

Hi, CART is introduced in Section 4.2 of the Dream paper: https://arxiv.org/pdf/2508.15487.