output
20122024
most citedCaptum: A unified and generic model interpretability library for PyTorch

649 citations

Showing cs.AIShow all

12 papers · 1 filter

cs.AI20216 cited

Scalable Online Planning via Reinforcement Learning Fine-Tuning

Arnaud Fickinger, Hengyuan Hu, Brandon Amos +2

Lookahead search has been a critical component of recent AI successes, such as in the games of chess, go, and poker. However, the search methods used in these games, and in many ot…

cs.AI202115 cited

MBRL-Lib: A Modular Library for Model-based Reinforcement Learning

Luis Pineda, Brandon Amos, Amy Zhang +2

Model-based reinforcement learning is a compelling framework for data-efficient learning of agents that interact with the world. This family of algorithms has many subcomponents th…

cs.AI2021

A Self-Supervised Auxiliary Loss for Deep RL in Partially Observable Settings

Eltayeb Ahmed, Luisa Zintgraf, Christian A. Schroeder de Witt +1

In this work we explore an auxiliary loss useful for reinforcement learning in environments where strong performing agents are required to be able to navigate a spatial environment…

cs.AI202032 cited

Learning Reasoning Strategies in End-to-End Differentiable Proving

Pasquale Minervini, Sebastian Riedel, Pontus Stenetorp +2

Attempts to render deep learning models interpretable, data-efficient, and robust have seen some success through hybridisation with rule-based systems, for example, in Neural Theor…

cs.AI20201 cited

Fairness-Aware Online Personalization

G Roshan Lal, Sahin Cem Geyik, Krishnaram Kenthapadi

Decision making in crucial applications such as lending, hiring, and college admissions has witnessed increasing use of algorithmic models and techniques as a result of a confluenc…

cs.AI2020

Empirically Verifying Hypotheses Using Reinforcement Learning

Kenneth Marino, Rob Fergus, Arthur Szlam +1

This paper formulates hypothesis verification as an RL problem. Specifically, we aim to build an agent that, given a hypothesis about the dynamics of the world, can take actions to…