← Search

Luke Vilnis

6 accepted papers

2024

MEMORY-VQ: Compression for Tractable Internet-Scale Memory

NAACL 2024short

Retrieval augmentation is a powerful but expensive method to make language models more knowledgeable about the world. Memory-based methods like LUMEN (de Jong et al., 2023a) pre-compute token representations for retrieved passages to drastically speed up inference. However, memory also leads to much…

Cited by 0SourcePDFScholar
2023

Arithmetic Sampling: Parallel Diverse Decoding for Large Language Models

ICML 2023oral

Decoding methods for large language models often trade-off between diversity of outputs and parallelism of computation. Methods such as beam search and Gumbel top-k sampling can guarantee a different output for each element of the beam, but are not easy to parallelize. Alternatively, methods such as…

2021

Capacity and Bias of Learned Geometric Embeddings for Directed Graphs

NeurIPS 2021poster

A wide variety of machine learning tasks such as knowledge base completion, ontology alignment, and multi-label classification can benefit from incorporating into learning differentiable representations of graphs or taxonomies. While vectors in Euclidean space can theoretically represent any graph,…

2020

Improving Local Identifiability in Probabilistic Box Embeddings

NeurIPS 2020poster

Geometric embeddings have recently received attention for their natural ability to represent transitive asymmetric relations via containment. Box embeddings, where objects are represented by n-dimensional hyperrectangles, are a particularly promising example of such an embedding as they are closed…

Cited by 71SourcePDFScholar
2019

Smoothing the Geometry of Probabilistic Box Embeddings

ICLR 2019oral

There is growing interest in geometrically-inspired embeddings for learning hierarchies, partial orders, and lattice structures, with natural applications to transitive relational data such as entailment graphs. Recent work has extended these ideas beyond deterministic hierarchies to probabilistical…

Cited by 106SourcePDFScholar
2018

Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learning

ICLR 2018poster

Knowledge bases (KB), both automatically and manually constructed, are often incomplete --- many valid facts can be inferred from the KB by synthesizing existing information. A popular approach to KB completion is to infer new relations by combinatory reasoning over the information found along other…