Relational Programming with Foundation Models

TL;DR

Vieira is a declarative framework that unifies various mechanisms for programming with foundation models, enabling complex neuro-symbolic and multi-modal applications with improved accuracy and conciseness.

Contribution

It introduces Vieira, a probabilistic relational framework that seamlessly integrates foundation models with logic programming and diverse sub-models, extending the Scallop compiler.

Findings

Vieira achieves comparable or better accuracy than baselines.

Programs in Vieira are concise and incorporate modern foundation models.

The framework supports diverse tasks across language, vision, and databases.

Abstract

Foundation models have vast potential to enable diverse AI applications. The powerful yet incomplete nature of these models has spurred a wide range of mechanisms to augment them with capabilities such as in-context learning, information retrieval, and code interpreting. We propose Vieira, a declarative framework that unifies these mechanisms in a general solution for programming with foundation models. Vieira follows a probabilistic relational paradigm and treats foundation models as stateless functions with relational inputs and outputs. It supports neuro-symbolic applications by enabling the seamless combination of such models with logic programs, as well as complex, multi-modal applications by streamlining the composition of diverse sub-models. We implement Vieira by extending the Scallop compiler with a foreign interface that supports foundation models as plugins. We implement plugins for 12 foundation models including GPT, CLIP, and SAM. We evaluate Vieira on 9 challenging tasks that span language, vision, and structured and vector databases. Our evaluation shows that programs in Vieira are concise, can incorporate modern foundation models, and have comparable or better accuracy than competitive baselines.