← Search

Awni Altabaa

4 accepted papers

2025

CoT Information: Improved Sample Complexity under Chain-of-Thought Supervision

NeurIPS 2025spotlight

Learning complex functions that involve multi-step reasoning poses a significant challenge for standard supervised learning from input-output examples. Chain-of-thought (CoT) supervision, which augments training data with intermediate reasoning steps to provide a richer learning signal, has driven r…

Cited by 0SourceScholar
2025

Disentangling and Integrating Relational and Sensory Information in Transformer Architectures

ICML 2025poster

Relational reasoning is a central component of generally intelligent systems, enabling robust and data-efficient inductive generalization. Recent empirical evidence shows that many existing neural architectures, including Transformers, struggle with tasks requiring relational reasoning. In this work…

2024

Abstractors and relational cross-attention: An inductive bias for explicit relational reasoning in Transformers

ICLR 2024poster

An extension of Transformers is proposed that enables explicit relational reasoning through a novel module called the *Abstractor*. At the core of the Abstractor is a variant of attention called *relational cross-attention*. The approach is motivated by an architectural inductive bias for relational…

2024

On the Role of Information Structure in Reinforcement Learning for Partially-Observable Sequential Teams and Games

NeurIPS 2024poster

In sequential decision-making problems, the *information structure* describes the causal dependencies between system variables, encompassing the dynamics of the environment and the agents' actions. Classical models of reinforcement learning (e.g., MDPs, POMDPs) assume a restricted and highly regular…

Cited by 3SourcePDFScholar