← Search

Amirhossein Kazemnejad

6 accepted papers

2026

The Markovian Thinker

ICLR 2026poster

Reasoning LLMs suffer from quadratic compute growth as their context length increases, making reinforcement learning with verifiable rewards (RLVR) and test-time scaling prohibitively expensive. Prior work has tried to lighten the computational burden by shortening reasoning traces through pruning,…

Cited by 0SourceScholar
2025

The Promise of RL for Autoregressive Image Editing

NeurIPS 2025poster

While image generation techniques are now capable of producing high-quality images that respect prompts which span multiple sentences, the task of text-guided image editing remains a challenge. Even edit requests that consist of only a few words often fail to be executed correctly. We explore three…

Cited by 0SourcecodeScholar
2025

VinePPO: Refining Credit Assignment in RL Training of LLMs

ICML 2025poster

Large language models (LLMs) are increasingly applied to complex reasoning tasks that require executing several complex steps before receiving any reward. Properly assigning credit to these steps is essential for enhancing model performance. Proximal Policy Optimization (PPO), a common reinforcement…

Cited by 0SourcePDFScholar
2023

Measuring the Knowledge Acquisition-Utilization Gap in Pretrained Language Models

EMNLP 2023long findings

While pre-trained language models (PLMs) have shown evidence of acquiring vast amounts of knowledge, it remains unclear how much of this parametric knowledge is actually usable in performing downstream tasks. We propose a systematic framework to measure parametric knowledge utilization in PLMs. Our…

Cited by 0SourceScholar
2023

The Impact of Positional Encoding on Length Generalization in Transformers

NeurIPS 2023poster

Length generalization, the ability to generalize from small training context sizes to larger ones, is a critical challenge in the development of Transformer-based language models. Positional encoding (PE) has been identified as a major factor influencing length generalization, but the exact impact o…

2022

The Curious Case of Absolute Position Embeddings

EMNLP 2022finding

Transformer language models encode the notion of word order using positional information. Most commonly, this positional information is represented by absolute position embeddings (APEs), that are learned from the pretraining data. However, in natural language, it is not absolute position that matte…