Rajkumar Vasudeva Raju

dblp:191/6745 · DBLP profile ↗
← Back
2ranked-venue papers
1as first author
1since 2021 · last 2023
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 2 · 1 first-author · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
2 papers
Probabilistic and Bayesian machine learning · 29% Language models and text generation · 19% Trustworthy machine learning · 19%

Topics — the 8 heaviest of 9, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation
in-context learning
0.712023
Schema-learning and rebinding as mechanisms of in-context learning and emergence · NeurIPS 2023
Machine learning › Trustworthy machine learning › interpretability
mechanistic interpretability
0.712023
Schema-learning and rebinding as mechanisms of in-context learning and emergence · NeurIPS 2023
Machine learning › Probabilistic and Bayesian machine learning › probabilistic inference
approximate inference
0.212016
Inference by Reparameterization in Neural Population Codes · NIPS 2016
Machine learning › Probabilistic and Bayesian machine learning › probabilistic inference › approximate inference
belief propagation
0.212016
Inference by Reparameterization in Neural Population Codes · NIPS 2016
Machine learning › Probabilistic and Bayesian machine learning › structured models
graphical models
0.212016
Inference by Reparameterization in Neural Population Codes · NIPS 2016
Machine learning › Probabilistic and Bayesian machine learning › probabilistic inference › approximate inference
loopy belief propagation
0.212016
Inference by Reparameterization in Neural Population Codes · NIPS 2016
Machine learning › Representation and self-supervised learning › computational neuroscience › neural coding
neural population coding
0.212016
Inference by Reparameterization in Neural Population Codes · NIPS 2016
Machine learning › Learning theory › online learning
sequence prediction
0.212023
Schema-learning and rebinding as mechanisms of in-context learning and emergence · NeurIPS 2023

Methods — techniques the papers use, named apart from their topics

schema learning · 0.7rebinding · 0.7clone-structured causal graphs · 0.7tree-based reparameterization · 0.2nonlinear dynamical systems · 0.2
YearPublicationVenuePosition
2023 Schema-learning and rebinding as mechanisms of in-context learning and emergence
abstract
In-context learning (ICL) is one of the most powerful and most unexpected capabilities to emerge in recent transformer-based large language models (LLMs). Yet the mechanisms that underlie it are poorly understood. In this paper, we demonstrate that comparable ICL capabilities can be acquired by an alternative sequence prediction learning method using clone-structured causal graphs (CSCGs). Moreover, a key property of CSCGs is that, unlike transformer-based LLMs, they are {\em interpretable}, which considerably simplifies the task of explaining how ICL works. Specifically, we show that it uses a combination of (a) learning template (schema) circuits for pattern completion, (b) retrieving relevant templates in a context-sensitive manner, and (c) rebinding of novel tokens to appropriate slots in the templates. We go on to marshall evidence for the hypothesis that similar mechanisms underlie ICL in LLMs. For example, we find that, with CSCGs as with LLMs, different capabilities emerge at different levels of overparameterization, suggesting that overparameterization helps in learning more complex template (schema) circuits. By showing how ICL can be achieved with small models and datasets, we open up a path to novel architectures, and take a vital step towards a more general understanding of the mechanics behind this important capability.
Sivaramakrishnan Swaminathan, Antoine Dedieu, Rajkumar Vasudeva Raju, Murray Shanahan, Miguel Lázaro-Gredilla, Dileep George
NeurIPS3
2016 Inference by Reparameterization in Neural Population Codes
abstract
Behavioral experiments on humans and animals suggest that the brain performs probabilistic inference to interpret its environment. Here we present a new general-purpose, biologically-plausible neural implementation of approximate inference. The neural network represents uncertainty using Probabilistic Population Codes (PPCs), which are distributed neural representations that naturally encode probability distributions, and support marginalization and evidence integration in a biologically-plausible manner. By connecting multiple PPCs together as a probabilistic graphical model, we represent multivariate probability distributions. Approximate inference in graphical models can be accomplished by message-passing algorithms that disseminate local information throughout the graph. An attractive and often accurate example of such an algorithm is Loopy Belief Propagation (LBP), which uses local marginalization and evidence integration operations to perform approximate inference efficiently even for complex models. Unfortunately, a subtle feature of LBP renders it neurally implausible. However, LBP can be elegantly reformulated as a sequence of Tree-based Reparameterizations (TRP) of the graphical model. We re-express the TRP updates as a nonlinear dynamical system with both fast and slow timescales, and show that this produces a neurally plausible solution. By combining all of these ideas, we show that a network of PPCs can represent multivariate probability distributions and implement the TRP updates to perform probabilistic inference. Simulations with Gaussian graphical models demonstrate that the neural network inference quality is comparable to the direct evaluation of LBP and robust to noise, and thus provides a promising mechanism for general probabilistic inference in the population codes of the brain.
Rajkumar Vasudeva Raju, Xaq Pitkow
NIPS1