← Search

Akash Gokul

8 accepted papers

2025

LaViDa: A Large Diffusion Model for Vision-Language Understanding

NeurIPS 2025spotlight

Modern Vision-Language Models (VLMs) can solve a wide range of tasks requiring visual reasoning. In real-world scenarios, desirable properties for VLMs include fast inference and controllable generation (e.g., constraining outputs to adhere to a desired format). However, existing autoregressive (AR)…

Cited by 0SourcecodeScholar
2025

OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows

CVPR 2025poster

We introduce OminiFlow, a novel generative model designed for any-to-any generation tasks such as text-to-image, text-to-audio, and audio-to-image synthesis. OminiFlow advances the rectified flow (RF) framework used in text-to-image models to handle the joint distribution of multiple modalities. It…

2025

Reflect-DiT: Inference-Time Scaling for Text-to-Image Diffusion Transformers via In-Context Reflection

ICCV 2025poster

The predominant approach to advancing text-to-image generation has been training-time scaling, where larger models are trained on more data using greater computational resources. While effective, this approach is computationally expensive, leading to growing interest in inference-time scaling to imp…

2024

Aligning Diffusion Models by Optimizing Human Utility

NeurIPS 2024poster

We present Diffusion-KTO, a novel approach for aligning text-to-image diffusion models by formulating the alignment objective as the maximization of expected human utility. Unlike previous methods, Diffusion-KTO does not require collecting pairwise preference data nor training a complex reward model…

2024

CodeChain: Towards Modular Code Generation Through Chain of Self-revisions with Representative Sub-modules

ICLR 2024poster

Large Language Models (LLMs) have already become quite proficient at solving simpler programming tasks like those in HumanEval or MBPP benchmarks. However, solving more complex and competitive programming tasks is still quite challenging for these models - possibly due to their tendency to generate…

2023

End-to-End Diffusion Latent Optimization Improves Classifier Guidance

ICCV 2023poster

Classifier guidance---using the gradients of an image classifier to steer the generations of a diffusion model---has the potential to dramatically expand the creative control over image generation and editing. However, currently classifier guidance requires either training new noise-aware models to…

Cited by 71PDFcodeScholar
2021

Remembering for the Right Reasons: Explanations Reduce Catastrophic Forgetting

ICLR 2021poster

The goal of continual learning (CL) is to learn a sequence of tasks without suffering from the phenomenon of catastrophic forgetting. Previous work has shown that leveraging memory in the form of a replay buffer can reduce performance degradation on prior tasks. We hypothesize that forgetting can be…