← Search

Mattia Atzeni

6 accepted papers

2023

Infusing Lattice Symmetry Priors in Attention Mechanisms for Sample-Efficient Abstract Geometric Reasoning

ICML 2023poster

The Abstraction and Reasoning Corpus (ARC) (Chollet, 2019) and its most recent language-complete instantiation (LARC) has been postulated as an important step towards general AI. Yet, even state-of-the-art machine learning models struggle to achieve meaningful performance on these problems, falling…

Cited by 6SourcePDFScholar
2023

Polar Ducks and Where to Find Them: Enhancing Entity Linking with Duck Typing and Polar Box Embeddings

EMNLP 2023long main

Entity linking methods based on dense retrieval are widely adopted in large-scale applications for their efficiency, but they can fall short of generative models, as they are sensitive to the structure of the embedding space. To address this issue, this paper introduces DUCK, an approach to infusing…

Cited by 0SourceScholar
2022

Case-based reasoning for better generalization in textual reinforcement learning

ICLR 2022poster

Text-based games (TBG) have emerged as promising environments for driving research in grounded language understanding and studying problems like generalization and sample efficiency. Several deep reinforcement learning (RL) methods with varying architectures and learning schemes have been proposed f…

Cited by 11SourcePDFScholar
2021

Efficient Text-based Reinforcement Learning by Jointly Leveraging State and Commonsense Graph Representations

ACL 2021short

Text-based games (TBGs) have emerged as useful benchmarks for evaluating progress at the intersection of grounded language understanding and reinforcement learning (RL). Recent work has proposed the use of external knowledge to improve the efficiency of RL agents for TBGs. In this paper, we posit th…

Cited by 17SourcePDFScholar
2021

SQALER: Scaling Question Answering by Decoupling Multi-Hop and Logical Reasoning

NeurIPS 2021poster

State-of-the-art approaches to reasoning and question answering over knowledge graphs (KGs) usually scale with the number of edges and can only be applied effectively on small instance-dependent subgraphs. In this paper, we address this issue by showing that multi-hop and more complex logical reason…

Cited by 17SourcePDFScholar
2021

Text-based RL Agents with Commonsense Knowledge: New Challenges, Environments and Baselines

AAAI 2021technical

Text-based games have emerged as an important test-bed for Reinforcement Learning (RL) research, requiring RL agents to combine grounded language understanding with sequential decision making. In this paper, we examine the problem of infusing RL agents with commonsense knowledge. Such knowledge woul…