geneDRAGNN: Gene Disease Prioritization using Graph Neural Networks. - Awni Altabaa; David Huang; Ciaran Byles-Ho; Hani Khatib; Fabian Sosa; Ting Hu 001 (CIBCB, 2022)

Decentralized Multi-Agent Reinforcement Learning for Continuous-Space Stochastic Games. - Awni Altabaa; Bora Yongacoglu; Serdar Yüksel (ACC, 2023)

Abstractors: Transformer Modules for Symbolic Message Passing and Relational Reasoning. - Awni Altabaa; Taylor W. Webb; Jonathan D. Cohen 003; John Lafferty (CoRR, 2023)

The Relational Bottleneck as an Inductive Bias for Efficient Abstraction. - Taylor W. Webb; Steven M. Frankland; Awni Altabaa; Kamesh Krishnamurthy; Declan Campbell; Jacob L. Russin; Randall C. O'Reilly; John Lafferty; Jonathan D. Cohen 003 (CoRR, 2023)

Relational Convolutional Networks: A framework for learning representations of hierarchical relations. - Awni Altabaa; John Lafferty (CoRR, 2023)

Abstractors and relational cross-attention: An inductive bias for explicit relational reasoning in Transformers. - Awni Altabaa; Taylor Whittington Webb; Jonathan D. Cohen 003; John Lafferty (ICLR, 2024)

On the Role of Information Structure in Reinforcement Learning for Partially-Observable Sequential Teams and Games. - Awni Altabaa; Zhuoran Yang (NeurIPS, 2024)

Approximation of relation functions and attention mechanisms. - Awni Altabaa; John Lafferty (CoRR, 2024)

Disentangling and Integrating Relational and Sensory Information in Transformer Architectures. - Awni Altabaa; John Lafferty (CoRR, 2024)

Learning Hierarchical Relational Representations through Relational Convolutions. - Awni Altabaa; John Lafferty (Trans. Mach. Learn. Res., 2024)

CoT Information: Improved Sample Complexity under Chain-of-Thought Supervision. - Awni Altabaa; Omar Montasser; John Lafferty (CoRR, 2025)