← Search

Adithya M Devraj

4 accepted papers

2025

AutoMixAlign: Adaptive Data Mixing for Multi-Task Preference Optimization in LLMs

ACL 2025long

When aligning large language models (LLMs), their performance across various tasks (such as being helpful, harmless, and honest) is heavily influenced by the composition of the training data. However, it is difficult to determine what mixture of data should be used to produce a model with strong per…

2020

Zap Q-Learning With Nonlinear Function Approximation

NeurIPS 2020poster

Zap Q-learning is a recent class of reinforcement learning algorithms, motivated primarily as a means to accelerate convergence. Stability theory has been absent outside of two restrictive classes: the tabular setting, and optimal stopping. This paper introduces a new framework for analysis of a m…

2019

Stochastic Variance Reduced Primal Dual Algorithms for Empirical Composition Optimization

NeurIPS 2019poster

We consider a generic empirical composition optimization problem, where there are empirical averages present both outside and inside nonlinear loss functions. Such a problem is of interest in various machine learning applications, and cannot be directly solved by standard methods such as stochastic…

2017

Zap Q-Learning

NeurIPS 2017poster

The Zap Q-learning algorithm introduced in this paper is an improvement of Watkins' original algorithm and recent competitors in several respects. It is a matrix-gain algorithm designed so that its asymptotic variance is optimal. Moreover, an ODE analysis suggests that the transient behavior is a c…

Cited by 110SourcePDFScholar