← Search

Rajat Sen

20 accepted papers

2026

Rethinking Multimodal Time-Series Forecasting Evaluation

ICML 2026poster

We introduce a new context-enriched, multimodal time series forecasting benchmark TimesX. TimesX contains a wide selection of high-quality real-world time series with diverse domains and textual contexts obtained from an automated data generation pipeline, which helps address three main issues of ex…

Cited by 0SourceScholar
2024

A decoder-only foundation model for time-series forecasting

ICML 2024poster

Motivated by recent advances in large language models for Natural Language Processing (NLP), we design a time-series foundation model for forecasting whose out-of-the-box zero-shot performance on a variety of public datasets comes close to the accuracy of state-of-the-art supervised forecasting mode…

2024

Linear Regression using Heterogeneous Data Batches

NeurIPS 2024spotlight

In many learning applications, data are collected from multiple sources, each providing a \emph{batch} of samples that by itself is insufficient to learn its input-output relationship. A common approach assumes that the sources fall in one of several unknown subgroups, each with an unknown input dis…

Cited by 3SourcePDFScholar
2022

On the benefits of maximum likelihood estimation for Regression and Forecasting

ICLR 2022poster

We advocate for a practical Maximum Likelihood Estimation (MLE) approach towards designing loss functions for regression and forecasting, as an alternative to the typical approach of direct empirical risk minimization on a specific target metric. The MLE approach is better suited to capture inductiv…

Cited by 14SourcePDFScholar
2022

Trimmed Maximum Likelihood Estimation for Robust Generalized Linear Model

NeurIPS 2022accept

We study the problem of learning generalized linear models under adversarial corruptions. We analyze a classical heuristic called the \textit{iterative trimmed maximum likelihood estimator} which is known to be effective against \textit{label corruptions} in practice. Under label corruptions, we pro…

Cited by 6SourcePDFScholar
2021

Top-k eXtreme Contextual Bandits with Arm Hierarchy

ICML 2021spotlight

Motivated by modern applications, such as online advertisement and recommender systems, we study the top-$k$ extreme contextual bandits problem, where the total number of arms can be enormous, and the learner is allowed to select $k$ arms and observe all or some of the rewards for the chosen arms. W…

Cited by 31SourcePDFScholar
2020

Mix and Match: An Optimistic Tree-Search Approach for Learning Models from Mixture Distributions

NeurIPS 2020poster

We consider a covariate shift problem where one has access to several different training datasets for the same learning problem and a small validation set which possibly differs from all the individual training distributions. The distribution shift is due, in part, to \emph{unobserved} features in…

2019

Noisy Blackbox Optimization using Multi-fidelity Queries: A Tree Search Approach

AISTATS 2019poster

We study the problem of black-box optimization of a noisy function in the presence of low-cost approximations or fidelities, which is motivated by problems like hyper-parameter tuning. In hyper-parameter tuning evaluating the black-box function at a point involves training a learning algorithm on a…

2019

Think Globally, Act Locally: A Deep Neural Network Approach to High-Dimensional Time Series Forecasting

NeurIPS 2019poster

Forecasting high-dimensional time series plays a crucial role in many applications such as demand forecasting and financial predictions. Modern datasets can have millions of correlated time-series that evolve together, i.e they are extremely high dimensional (one dimension for each individual time-s…

2018

Multi-Fidelity Black-Box Optimization with Hierarchical Partitions

ICML 2018oral

Motivated by settings such as hyper-parameter tuning and physical simulations, we consider the problem of black-box optimization of a function. Multi-fidelity techniques have become popular for applications where exact function evaluations are expensive, but coarse (biased) approximations are availa…

2017

Contextual Bandits with Latent Confounders: An NMF Approach

AISTATS 2017poster

Motivated by online recommendation and advertising systems, we consider a causal model for stochastic contextual bandits with a latent low-dimensional confounder. In our model, there are $L$ observed contexts and $K$ arms of the bandit. The observed context influences the reward obtained through a l…

Cited by 55SourcePDFScholar
2017

Identifying Best Interventions through Online Importance Sampling

ICML 2017poster

Motivated by applications in computational advertising and systems biology, we consider the problem of identifying the best out of several possible soft interventions at a source node $V$ in an acyclic causal directed graph, to maximize the expected value of a target node $Y$ (located downstream of…

Cited by 94SourcePDFScholar
2017

Model-Powered Conditional Independence Test

NeurIPS 2017poster

We consider the problem of non-parametric Conditional Independence testing (CI testing) for continuous random variables. Given i.i.d samples from the joint distribution $f(x,y,z)$ of continuous random vectors $X,Y$ and $Z,$ we determine whether $X \independent Y \vert Z$. We approach this by convert…