← Search

Vineet Jain

6 accepted papers

2026

Long-Horizon Model-Based Offline Reinforcement Learning Without Conservatism

ICML 2026poster

Popular offline reinforcement learning (RL) methods rely on conservatism, penalizing out-of-dataset actions or restricting rollout horizons. We question the universality of this principle and revisit a complementary Bayesian perspective. By modeling a posterior over plausible world models and traini…

Cited by 0SourceScholar
2025

Diffusion Tree Sampling: Scalable inference‑time alignment of diffusion models

NeurIPS 2025poster

Adapting a pretrained diffusion model to new objectives at inference time remains an open problem in generative modeling. Existing steering methods suffer from inaccurate value estimation, especially at high noise levels, which biases guidance. Moreover, information from past runs is not reused to i…

Cited by 32SourceScholar
2025

Scaling Laws and Efficient Inference for Ternary Language Models

ACL 2025long

Large language models (LLMs) are increasingly used across research and industry applications, yet their inference efficiency remains a significant challenge. As the computational power of modern GPU architectures continuously improves, their memory bandwidth and capacity have not scaled proportional…

2024

On Diffusion Modeling for Anomaly Detection

ICLR 2024spotlight

Known for their impressive performance in generative modeling, diffusion models are attractive candidates for density-based anomaly detection. This paper investigates different variations of diffusion modeling for unsupervised and semi-supervised anomaly detection. In particular, we find that Denois…

2022

EqR: Equivariant Representations for Data-Efficient Reinforcement Learning

ICML 2022spotlight

We study a variety of notions of equivariance as an inductive bias in Reinforcement Learning (RL). In particular, we propose new mechanisms for learning representations that are equivariant to both the agent’s action, as well as symmetry transformations of the state-action pairs. Whereas prior work…