Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Joint Flow Matching (JFM), a novel training framework that enhances continuous normalizing flows by enabling consistent forward and reverse conditional inference from a shared joint model. By assigning opposite roles to variables at temporal endpoints, JFM achieves a coherent joint distribution that supports both joint classification and generation tasks. The framework demonstrates competitive accuracy and well-calibrated confidence scores on conditional datasets, highlighting its potential for improving interpretability in discriminative-generative models.
JFM achieves competitive accuracy and reliable confidence scores without the need for post-hoc calibration, transforming how we approach joint classification and generation tasks.
We introduce Joint Flow Matching (JFM), a training framework for continuous normalising flows over multiple variables. Standard flow matching transports variables from noise to data simultaneously, offering no natural mechanism for forward and reverse conditional inference from a shared joint model. JFM resolves this by assigning opposite roles to each variable at the temporal endpoints. We prove that JFM produces a consistent joint distribution where that forward or reverse integration are conditionals of the same joint. We explore this consistency in the context of joint classification and generation as the basis for interpretability in discriminative-generative models. We validate JFM on conditional datasets producing competitive accuracy with inherently well-calibrated confidence scores without post-hoc calibration, and classifier-consistent image generation.