← Search

Nirmit Joshi

5 accepted papers

2026

Learning to Answer from Correct Demonstrations

ICLR 2026poster

We study the problem of learning to generate an answer (or completion) to a question (or prompt), where there could be multiple correct answers, any one of which is acceptable at test time. Learning is based on demonstrations of some correct answer to each training question, as in Supervised Fine Tu…

Cited by 0SourceScholar
2025

Learning single index models via harmonic decomposition

NeurIPS 2025poster

We study the problem of learning single-index models, where the label $y \in \mathbb{R}$ depends on the input $\boldsymbol{x} \in \mathbb{R}^d$ only through an unknown one-dimensional projection $\langle \boldsymbol{w_*}, \boldsymbol{x} \rangle$. Prior work has shown that under Gaussian inputs, the…

Cited by 0SourceScholar
2024

On the Complexity of Learning Sparse Functions with Statistical and Gradient Queries

NeurIPS 2024poster

The goal of this paper is to investigate the complexity of gradient algorithms when learning sparse functions (juntas). We introduce a type of Statistical Queries ($\mathsf{SQ}$), which we call Differentiable Learning Queries ($\mathsf{DLQ}$), to model gradient queries on a specified loss with respe…

Cited by 5SourcePDFScholar