← Search

Ida Momennejad

8 accepted papers

2026

Algorithmic Primitives and Compositional Geometry of Reasoning in Language Models

ICML 2026poster

How do latent and inference time computations enable large language models (LLMs) to solve multi-step reasoning? We introduce a framework for tracing and steering algorithmic primitives that underlie model reasoning. Our approach links reasoning traces to internal activations and evaluates algorithm…

Cited by 0SourceScholar
2025

Position: We Need An Algorithmic Understanding of Generative AI

ICML 2025spotlight

What algorithms do LLMs actually learn and use to solve problems? Studies addressing this question are sparse, as research priorities are focused on improving performance through scale, leaving a theoretical and empirical gap in understanding emergent algorithms. This position paper proposes AlgEval…

Cited by 0SourcePDFScholar
2023

Evaluating Cognitive Maps and Planning in Large Language Models with CogEval

NeurIPS 2023poster

Recently an influx of studies claims emergent cognitive abilities in large language models (LLMs). Yet, most rely on anecdotes, overlook contamination of training sets, or lack systematic Evaluation involving multiple tasks, control conditions, multiple iterations, and statistical robustness tests.…

Cited by 64SourcePDFScholar
2023

Imitating Human Behaviour with Diffusion Models

ICLR 2023poster

Diffusion models have emerged as powerful generative models in the text-to-image domain. This paper studies their application as observation-to-action models for imitating human behaviour in sequential environments. Human behaviour is stochastic and multimodal, with structured correlations between a…

2022

Interaction-Grounded Learning with Action-Inclusive Feedback

NeurIPS 2022accept

Consider the problem setting of Interaction-Grounded Learning (IGL), in which a learner's goal is to optimally interact with the environment with no explicit reward to ground its policies. The agent observes a context vector, takes an action, and receives a feedback vector, using this information to…

Cited by 7SourcePDFScholar
2022

Towards Evaluating Adaptivity of Model-Based Reinforcement Learning Methods

ICML 2022spotlight

In recent years, a growing number of deep model-based reinforcement learning (RL) methods have been introduced. The interest in deep model-based RL is not surprising, given its many potential benefits, such as higher sample efficiency and the potential for fast adaption to changes in the environment…

2021

Navigation Turing Test (NTT): Learning to Evaluate Human-Like Navigation

ICML 2021spotlight

A key challenge on the path to developing agents that learn complex human-like behavior is the need to quickly and accurately quantify human-likeness. While human assessments of such behavior can be highly accurate, speed and scalability are limited. We address these limitations through a novel auto…