← Search

Divya Jyoti Bajpai

8 accepted papers

2026

FastFlow: Accelerating The Generative Flow Matching Models with Bandit Inference

ICLR 2026poster

Flow-matching models deliver state-of-the-art fidelity in image and video generation, but the inherent sequential denoising process renders them slower. Existing acceleration methods like distillation, trajectory truncation, and consistency approaches are static, require retraining, and often fail t…

Cited by 0SourcecodeScholar
2026

FlowCast: Trajectory Forecasting for Scalable Zero-Cost Speculative Flow Matching

ICLR 2026poster

Flow Matching (FM) has recently emerged as a powerful approach for high-quality visual generation. However, their prohibitively slow inference due to a large number of denoising steps limits their potential use in real-time or interactive applications. Existing acceleration methods, like distillatio…

Cited by 0SourceScholar
2025

BEEM: Boosting Performance of Early Exit DNNs using Multi-Exit Classifiers as Experts

ICLR 2025poster

Early Exit (EE) techniques have emerged as a means to reduce inference latency in Deep Neural Networks (DNNs). The latency improvement and accuracy in these techniques crucially depend on the criteria used to make exit decisions. We propose a new decision criterion BEEM where exit classifiers are tr…

2025

Beyond Greedy Exits: Improved Early Exit Decisions for Risk Control and Reliability

NeurIPS 2025poster

Early-Exit Deep Neural Networks enable adaptive inference by allowing prediction at intermediary layers, significantly reducing computational costs and latency. Most of the early exit strategies greedily exit a sample at an intermediary layer if the confidence in class prediction exceeds a predefine…

Cited by 0SourceScholar
2024

CapEEN: Image Captioning with Early Exits and Knowledge Distillation

EMNLP 2024finding

Deep neural networks (DNNs) have made significant progress in recognizing visual elements and generating descriptive text in image-captioning tasks. However, their improved performance comes from increased computational burden and inference latency. Early Exit (EE) strategies can be used to enhance…