AI & interpretability researcher
I'm a staff research scientist at Voyage AI, working on training large embedding models. Before that, I spent ~4 lovely years at Microsoft Research (Deep learning group) researching LLMs to build & interpret healthcare data, brain models, and LLMs themselves. I completed my PhD in AI at UC Berkeley (with the brilliant Bin Yu) in 2022, working on interpretability.
| Year | Title | Authors | Tags | Paper | Code | Misc |
|---|---|---|---|---|---|---|
| '26 | Agentic-imodels: evolving agentic interpretability tools via autoresearch | Singh, Tan, Xu, Gero, Yang, Galley, & Gao | ๐๐ค๐ณ | NeurIPS | ||
| '26 | Generative causal testing | Antonello*, Singh*, Jain, Hsu, Gao, Yu, & Huth | ๐๐ค๐ง | Nature Neuroscience | ||
| '26 | StudentSim: Training LLM-based Student Simulators | Yang, Wang, Galley, Singh, Inala, Zhai, & Gao | ๐ค | NeurIPS workshop | ||
| '26 | Learning the electronic health record at the minute-scale | Narang et al. | ๐ค๐ | arXiv | ||
| '26 | RL for EHR foundation models | Xiao, Zhang, Singh, Naumann, Poon, Gao, & Liu | ๐ค๐ | arXiv | ||
| '26 | TokenEconomizer: Experience-Driven Black-Box Cost Control for Coding Agents | Li et al. | ๐ค | arXiv | ||
| '26 | Register tokens for dLLMs | Ge, Singh, Zhuang, Liu, Gao, & Sala | ๐ค | arXiv | ||
| '26 | Sanity checks for agentic data science | Rewolinski*, Zane*, Huang, Singh, Wang, Gao, & Yu | ๐ค | arXiv | ||
| '26 | Test-time learning with an evolving library | Xu, Sordoni, Singh, Gero, Galley, Yuan, & Gao | ๐ค | NeurIPS | ||
| '26 | Test-time recursive thinking | Zhuang, Singh, Liu, Shen, Zhang, Shang, Gao, & Chen | ๐ค | arXiv | ||
| '26 | Selecting feature interactions by distilling foundation models | Jia, Singh, Carauana, & Lengerich | ๐๐ค๐ณ | arXiv | ||
| '26 | Human-AI co-design for clinical prediction models | Feng*, Kothari*, Vossler, Bishara, Zier, Addo, Kornblith, Tan, & Singh | ๐๐ค๐๐ณ | npj Digital Medicine | ||
| '26 | Do explanations generalize across large reasoning models? | Pal, Bau, & Singh | ๐๐ค | arXiv | ||
| '26 | Data-efficient adaptation of LLMs via attention head reweighting | Oikarinen, Chen, Siska, Weng, Singh, & Gao | ๐๐ค | COLM | ||
| '26 | Interpreting and steering state-space models via activation subspace bottlenecks | Mohan, Gupta, Das, & Singh | ๐๐ค | ICML | ||
| '25 | Evaluating scientific theories as predictive models in language neuroscience | Singh*, Antonello*, Guo, Mischler, Gao, Mesgarani, & Huth | ๐๐ค๐ง | bioRxiv | ||
| '25 | Mixture of inputs | Zhuang, Liu, Singh, Shang, & Gao | ๐ค | NeurIPS | ||
| '25 | Interpretable language modeling via induction-head ngram models | Kim*, Mantena*, Yang, Singh, Yoon, & Gao | ๐๐ค๐ง ๐ณ | NeurIPS | ||
| '25 | Bayesian concept bottleneck models with LLM priors | Feng, Kothari, Zier, Singh, & Tan | ๐๐ค๐๐ณ | NeurIPS | ||
| '25 | OmniGuard | Verma, Hines, Bilmes, Siska, Zettlemoyer, Gonen, & Singh | ๐ค | EMNLP | ||
| '25 | Systematic bias in clinical decision instrument development | Obra, Singh, et al. | ๐ | npj Digital Medicine | ||
| '25 | Analyzing patient perspectives with LLMs | Kornblith*, Singh* et al. | ๐ค๐ | Nature Scientific Reports | ||
| '25 | SimDINO | Wu et al. | ICML | |||
| '25 | Vector-ICL: in-context learning with continuous vector representations | Zhuang et al. | ๐ค | ICLR | ||
| '24 | Crafting interpretable embeddings by asking LLMs questions | Benara*, Singh*, Morris, Antonello, Stoica, Huth, & Gao | ๐๐ค๐ง ๐ณ | NeurIPS | ||
| '25 | Towards understanding graphical perception in large multimodal models | Zhang et al. | ๐ค | arXiv | ||
| '24 | Rethinking interpretability in the era of large language models | Singh, Inala, Galley, Caruana, & Gao | ๐๐ค | arXiv | ||
| '24 | Towards consistent natural-language explanations via explanation-consistency finetuning | Chen et al. | ๐๐ค | COLING | ||
| '24 | Learning a decision tree algorithm with Transformers | Zhuang et al. | ๐๐ค๐ณ | TMLR | ||
| '24 | Model tells itself where to attend: faithfulness meets automatic attention steering | Zhang*, Yu*, et al. | ๐๐ค | arXiv | ||
| '24 | Tell your model where to attend: post-hoc attention steering for LLMs | Zhang et al. | ๐๐ค | ICLR | ||
| '24 | Attribute structuring improves LLM-based evaluation of clinical text summaries | Gero et al. | ๐ค๐ | ML4H findings | ||
| '23 | Tree prompting | Morris*, Singh*, Rush, Gao, & Deng | ๐๐ค๐ณ | EMNLP | ||
| '23 | Augmenting interpretable models with LLMs during training | Singh, Askari, Caruana, & Gao | ๐๐ค๐ณ | Nature Communications | ||
| '23 | Explaining black box text modules in natural language with language models | Singh*, Hsu*, Antonello, Jain, Huth, Yu & Gao | ๐๐ค๐ง | NeurIPS workshop | ||
| '23 | Self-verification improves few-shot clinical information extraction | Gero*, Singh*, Cheng, Naumann, Galley, Gao, & Poon | ๐ค๐ | ICML workshop | ||
| '22 | Explaining patterns in data with language models via interpretable autoprompting | Singh*, Morris*, Aneja, Rush, & Gao | ๐๐ค | EMNLP workshop | ||
| '22 | Stress testing a clinical decision instrument performance for intra-abdominal injury | Kornblith*, Singh* et al. | ๐๐ณ | PLOS Digital Health | ||
| '22 | Fast interpretable greedy-tree sums (FIGS) | Tan*, Singh*, Nasseri, Agarwal, & Yu | ๐๐ณ | PNAS | ||
| '22 | Hierarchical shrinkage for trees | Agarwal*, Tan*, Ronen, Singh, & Yu | ๐๐ณ | ICML (spotlight) | ||
| '22 | VeridicalFlow: a python package for building trustworthy data science pipelines with PCS | Duncan*, Kapoor*, Agarwal*, Singh*, & Yu | ๐ | JOSS | ||
| '21 | imodels: a python package for fitting interpretable models | Singh*, Nasseri*, et al. | ๐๐ณ | JOSS | ||
| '21 | Adaptive wavelet distillation from neural networks through interpretations | Ha, Singh, et al. | ๐๐ณ | NeurIPS | ||
| '21 | Matched sample selection with GANs for mitigating attribute confounding | Singh, Balakrishnan, & Perona | CVPR workshop | |||
| '21 | Revisiting complexity and the bias-variance tradeoff | Dwivedi*, Singh*, Yu & Wainwright | JMLR | |||
| '20 | Curating a COVID-19 data repository and forecasting county-level death counts in the United States | Altieri et al. | ๐ | HDSR | ||
| '20 | Transformation importance with applications to cosmology | Singh*, Ha*, Lanusse, Boehm, Liu & Yu | ๐ | ICLR workshop (spotlight) | ||
| '20 | Interpretations are useful: penalizing explanations to align neural networks with prior knowledge | Rieger, Singh, Murdoch & Yu | ๐ | ICML | ||
| '19 | Hierarchical interpretations for neural network predictions | Singh*, Murdoch*, & Yu | ๐ | ICLR | ||
| '19 | Interpretable machine learning: definitions, methods, and applications | Murdoch*, Singh*, et al. | ๐๐ณ | PNAS | ||
| '19 | Disentangled attribution curves for interpreting random forests and boosted trees | Devlin, Singh, Murdoch & Yu | ๐๐ณ | arXiv | ||
| '18 | Large scale image segmentation with structured loss based deep learning for connectome reconstruction | Funke*, Tschopp*, et al. | ๐ง | TPAMI | ||
| '18 | Linearization of excitatory synaptic integration at no extra cost | Morel, Singh, & Levy | ๐ง | J Comp Neuro | ||
| '17 | A consensus layer V pyramidal neuron can sustain interpulse-interval coding | Singh & Levy | ๐ง | PLOS One | ||
| '17 | A constrained, weighted-l1 minimization approach for joint discovery of heterogeneous neural connectivity graphs | Singh, Wang, & Qi | ๐ง | NeurIPS Workshop |
Notes in AI / neuroscience.
Mini personal projects. There's also some dumb stuff here. (Largely obsoleted by coding agents)
Training large embedding models
Generating explanations with LLMs for science & medicine, working in the Deep Learning group
Researched interpretable machine learning, especially for neural networks and in scientific applications, working in the Bin Yu group
Worked on graphical models under Yanjun Qi and on biophysical modeling under William Levy; also spent 3 summers doing research at HHMI Janelia under Srini Turaga
I've been lucky to work with many amazing people & help advise some incredible students.