← Search

Kelsey Allen

6 accepted papers

2023

Improving Dual-Encoder Training through Dynamic Indexes for Negative Mining

AISTATS 2023poster

Dual encoder models are ubiquitous in modern classification and retrieval. Crucial for training such dual encoders is an accurate estimation of gradients from the partition function of the softmax over the large output space; this requires finding negative targets that contribute most significantly…

Cited by 6SourcePDFScholar
2019

Learning sparse relational transition models

ICLR 2019poster

We present a representation for describing transition models in complex uncertain domains using relational rules. For any action, a rule selects a set of relevant objects and computes a distribution over properties of just those objects in the resulting state given their properties in the previous…

Cited by 29SourcePDFScholar
2018

Differentiable Physics and Stable Modes for Tool-Use and Manipulation Planning

RSS 2018poster

We consider the problem of sequential manipulation and tool-use planning in domains that include physical interactions such as hitting and throwing. The approach integrates a Task And Motion Planning formulation with primitives that either impose stable kinematic constraints or differentiabl…

2018

End-to-End Differentiable Physics for Learning and Control

NeurIPS 2018spotlight

We present a differentiable physics engine that can be integrated as a module in deep neural networks for end-to-end learning. As a result, structured physics knowledge can be embedded into larger systems, allowing them, for example, to match observations by performing precise simulations, while ac…