← Search

Albert Q. Jiang

6 accepted papers

2024

End-to-End Ontology Learning with Large Language Models

NeurIPS 2024poster

Ontologies are useful for automatic machine processing of domain knowledge as they represent it in a structured format. Yet, constructing ontologies requires substantial manual effort. To automate part of this process, large language models (LLMs) have been applied to solve various subtasks of ontol…

2024

Llemma: An Open Language Model for Mathematics

ICLR 2024poster

We present Llemma, a large language model for mathematics. We continue pretraining Code Llama on the Proof-Pile-2, a mixture of scientific papers, web data containing mathematics, and mathematical code, yielding Llemma. On the MATH benchmark Llemma outperforms all known openly released models, as we…

Cited by 320SourcePDFScholar
2024

Magnushammer: A Transformer-Based Approach to Premise Selection

ICLR 2024poster

This paper presents a novel approach to premise selection, a crucial reasoning task in automated theorem proving. Traditionally, symbolic methods that rely on extensive domain knowledge and engineering effort are applied to this task. In contrast, this work demonstrates that contrastive training wit…

Cited by 44SourcePDFScholar
2024

Repurposing Language Models into Embedding Models: Finding the Compute-Optimal Recipe

NeurIPS 2024poster

Text embeddings are essential for tasks such as document retrieval, clustering, and semantic similarity assessment. In this paper, we study how to contrastively train text embedding models in a compute-optimal fashion, given a suite of pretrained decoder-only language models. Our innovation is an al…