Saved in:
Bibliographic Details
Main Authors: Jha, Om Govind, Bamniya, Manoj, Borthakur, Ayon
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.13928
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915862877831168
author Jha, Om Govind
Bamniya, Manoj
Borthakur, Ayon
author_facet Jha, Om Govind
Bamniya, Manoj
Borthakur, Ayon
contents Traditional discriminative computer vision relies predominantly on static projections, mapping input features to outputs in a single computational step. Although efficient, this paradigm lacks the iterative refinement and robustness inherent in biological vision and modern generative modelling. In this paper, we propose Discriminative Flow Matching, a framework that reformulates classification and object detection as a conditional transport process. By learning a vector field that continuously transports samples from a simple noise distribution toward a task-aligned target manifold -- such as class embeddings or bounding box coordinates -- we are at the interface between generative and discriminative learning. Our method attaches multiple independent flow predictors to a shared backbone. These predictors are trained using local flow matching objectives, where gradients are computed independently for each block. We formulate this approach for standard image classification and extend it to the complex task of object detection, where targets are high-dimensional and spatially distributed. This architecture provides the flexibility to update blocks either sequentially to minimise activation memory or in parallel to suit different hardware constraints. By aggregating the predictions from these independent flow predictors, our framework enables robust, generative-inspired inference across diverse architectures, including CNNs and vision transformers.
format Preprint
id arxiv_https___arxiv_org_abs_2603_13928
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Discriminative Flow Matching Via Local Generative Predictors
Jha, Om Govind
Bamniya, Manoj
Borthakur, Ayon
Computer Vision and Pattern Recognition
Artificial Intelligence
Traditional discriminative computer vision relies predominantly on static projections, mapping input features to outputs in a single computational step. Although efficient, this paradigm lacks the iterative refinement and robustness inherent in biological vision and modern generative modelling. In this paper, we propose Discriminative Flow Matching, a framework that reformulates classification and object detection as a conditional transport process. By learning a vector field that continuously transports samples from a simple noise distribution toward a task-aligned target manifold -- such as class embeddings or bounding box coordinates -- we are at the interface between generative and discriminative learning. Our method attaches multiple independent flow predictors to a shared backbone. These predictors are trained using local flow matching objectives, where gradients are computed independently for each block. We formulate this approach for standard image classification and extend it to the complex task of object detection, where targets are high-dimensional and spatially distributed. This architecture provides the flexibility to update blocks either sequentially to minimise activation memory or in parallel to suit different hardware constraints. By aggregating the predictions from these independent flow predictors, our framework enables robust, generative-inspired inference across diverse architectures, including CNNs and vision transformers.
title Discriminative Flow Matching Via Local Generative Predictors
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2603.13928