PrT-Flow: Probability-Tractable Flow Matching for Reinforcement Learning

Congzhong Sun, Ning Tan

Diffusion models and Flow Matching can represent flexible distributions over robot actions, but many reinforcement learning algorithms also require the probability of each action.…