Skip to main content ->
Ai2

Research - Papers

Explore a selection of our published work on a variety of key research challenges in AI.

Filter papers

Evidence-Informed LLM Beliefs for Continual Scientific Discovery

Dhruv AgarwalReece AdamsonAndrew McCallumBodhisattwa Prasad Majumder
2026
arXiv.org

Open-ended scientific discovery with large language models (LLMs) increasingly operates as a long-horizon loop of hypothesis search and verification, where a reward signal guides which hypotheses to… 

Efficiently Representing Algorithms With Chain-of-Thought Transformers

Yanhong LiAnej SveteAshish SabharwalWilliam Merrill
2026
arXiv.org

The increasing popularity of \emph{reasoning} models -- language models that output a series of reasoning or thought tokens before producing an answer -- is justified, in part, by theoretical… 

Operadic consistency: a label-free signal for compositional reasoning failures in LLMs

Nathaniel BottmanYinhong LiuKyle Richardson
2026
NeurIPS

Detecting LLM reasoning failures at inference time without ground-truth labels has motivated a wide range of confidence baselines, including self-consistency, semantic entropy, and P(True), built on… 

Operads for compositional reasoning in LLMs

Nathaniel BottmanKyle Richardson
2026
Workshop on Combining Theory and Benchmarks (@ICML 2026)

Question decomposition, i.e. breaking a complex query into simpler sub-queries whose answers are composed to produce a final answer, is a widely used strategy for improving LLM reasoning, yet it… 

Scientific reasoning does not reliably translate into scientific forecasting in frontier AI

Sean WuPan LuYupeng ChenJunchi Yu
2026
arXiv

AI systems are increasingly used to support forward-looking scientific judgment, but it remains unclear whether they can form reliable expectations about future scientific advances. Here we show… 

Analytica: Soft Propositional Reasoning for Robust and Scalable LLM-Driven Analysis

Junyan ChengKyle RichardsonPeter Chin
2026
ICLR 2026

Large language model (LLM) agents are increasingly tasked with complex real-world analysis (e.g., in financial forecasting, scientific discovery), yet their reasoning suffers from stochastic… 

AstaBench: Rigorous Benchmarking of AI Agents with a Scientific Research Suite

Jonathan BraggMike D'ArcyNishant BalepurDaniel S. Weld
2026
ICLR

AI agents hold the potential to revolutionize scientific productivity by automating literature reviews, replicating experiments, analyzing data, and even proposing new directions of inquiry; indeed,… 

On the Reasoning Abilities of Masked Diffusion Language Models

Anej SveteAshish Sabharwal
2026
ICLR

Masked diffusion models (MDMs) for text offer a compelling alternative to traditional autoregressive language models. Parallel generation makes them efficient, but their computational capabilities… 

SimpleToM: Exposing the Gap between Explicit ToM Inference and Implicit ToM Application in LLMs

Yuling GuOyvind TafjordHyunwoo KimYejin Choi
2026
ICLR

Large language models (LLMs) are increasingly tested for a"Theory of Mind"(ToM) - the ability to attribute mental states to oneself and others. Yet most evaluations stop at explicit belief… 

Probabilistic Programs of Thought

Poorva GargRenato Lui GehDaniel Mingyi IsraelGuy Van den Broeck
2026
arXiV

LLMs are widely used for code generation and mathematical reasoning tasks where they are required to generate structured output. They either need to reason about code, generate code for a given…